java调用远程服务器shell脚本

绝地灬酷狼 2022-01-12 03:35 364阅读 0赞

# 1.代码 #

代码比较简单

package com.springboot.information.utils;
    
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.UnsupportedEncodingException;
    import java.nio.charset.Charset;
    
    import org.apache.commons.io.IOUtils;
    
    import ch.ethz.ssh2.ChannelCondition;
    import ch.ethz.ssh2.Connection;
    import ch.ethz.ssh2.Session;
    import ch.ethz.ssh2.StreamGobbler;
    
    public class RemoteShellExecutor {
    
        private Connection conn;
        /** 远程机器IP */
        private String ip;
        /** 用户名 */
        private String osUsername;
        /** 密码 */
        private String password;
        private String charset = Charset.defaultCharset().toString();
    
        private static final int TIME_OUT = 1000 * 5 * 60;
    
        /**
         * 构造函数
         * @param ip
         * @param usr
         * @param pasword
         */
        public RemoteShellExecutor(String ip, String usr, String pasword) {
            this.ip = ip;
            this.osUsername = usr;
            this.password = pasword;
        }
    
    
        /**
         * 登录
         * @return
         * @throws IOException
         */
        private boolean login() throws IOException {
            conn = new Connection(ip);
            conn.connect();
            return conn.authenticateWithPassword(osUsername, password);
        }
    
        /**
         * 执行脚本
         *
         * @param cmds
         * @return
         * @throws Exception
         */
        public int exec(String cmds) throws Exception {
            InputStream stdOut = null;
            InputStream stdErr = null;
            String outStr = "";
            String outErr = "";
            int ret = -1;
            try {
                if (login()) {
                    System.out.println("login success");
                    // Open a new {@link Session} on this connection
                    Session session = conn.openSession();
                    // Execute a command on the remote machine.
                    System.out.println("session success");
                    session.execCommand(cmds);
                    System.out.println("exec:"+cmds);
                    System.out.println("session:"+session);
                    stdOut = new StreamGobbler(session.getStdout());
                    System.out.println("stdOut:"+stdOut.toString());
                    outStr = processStream(stdOut, charset);
                    System.out.println("stdOut");
    
    
                    stdErr = new StreamGobbler(session.getStderr());
                    outErr = processStream(stdErr, charset);
                    System.out.println("stdErr");
    
                    session.waitForCondition(ChannelCondition.EXIT_STATUS, TIME_OUT);
    
                    System.out.println("outStr=" + outStr);
                    System.out.println("outErr=" + outErr);
    
                    ret = session.getExitStatus();
                } else {
                    throw new Exception("登录远程机器失败" + ip); // 自定义异常类 实现略
                }
            } finally {
                if (conn != null) {
                    conn.close();
                }
                IOUtils.closeQuietly(stdOut);
                IOUtils.closeQuietly(stdErr);
            }
            return ret;
        }
    
        /**
         * @param in
         * @param charset
         * @return
         * @throws IOException
         * @throws UnsupportedEncodingException
         */
        private String processStream(InputStream in, String charset) throws Exception {
            System.out.println("ininin");
            byte[] buf = new byte[1024];
            StringBuilder sb = new StringBuilder();
            int num =0;
            while (in.read(buf) != -1) {
                num++;
                System.out.println("number:"+num);
                sb.append(new String(buf, charset));
                System.out.println("sb:"+sb.toString());
                System.out.println("buf:"+in.read(buf));
    
            }
            System.out.println("enene22");
            return sb.toString();
        }
    
        public static void main(String args[]) throws Exception {
            RemoteShellExecutor executor = new RemoteShellExecutor("XXXX", "XX", "XXX");
            // 执行myTest.sh 参数为java Know dummy
            System.out.println(executor.exec("/home/zh/Documents/start/note.sh"));
        }
    }
    复制代码

转载于:https://juejin.im/post/5d072f08518825092c7171d5