远程连接linux或者本地执行shell

比眉伴天荒 2021-11-24 01:52 339阅读 0赞

# 远程连接linux或者本地执行shell，单例模式 #

import ch.ethz.ssh2.Connection;
    import ch.ethz.ssh2.Session;
    import ch.ethz.ssh2.StreamGobbler;
    import com.alibaba.druid.util.StringUtils;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    
    import java.io.BufferedReader;
    import java.io.IOException;
    import java.io.InputStream;
    import java.io.InputStreamReader;
    import java.io.UnsupportedEncodingException;
    
    /**
     * @ClassName: RemoteConnection
     * @Description: TODO
     * @Author Tan
     * @Date 2019/7/27
     */
    public class RemoteConnection {
    
        private final static Logger log = LoggerFactory.getLogger(RemoteConnection.class);
    
        private static RemoteConnection instance;
    
        private static String DEFAULTCHART = "UTF-8";
    
        private static Connection conn = null;
    
        private static String SERVER = "linux";
        private static String IP = "192.168.1.2";
        private static String HOST_NAME = "root";
        private static String HOST_PWD = "123456";
    
        public static synchronized RemoteConnection getInstance() {
            if (instance == null) {
                instance = new RemoteConnection();
            }
            return instance;
        }
    
        public static synchronized Connection getCon(String ip, String userName, String userPwd) {
            if (conn == null) {
                conn = login(ip, userName, userPwd);
            }
            return conn;
        }
    
        /**
         * 登录主机
         *
         * @return 登录成功返回true，否则返回false
         */
        public static Connection login(String ip, String userName, String userPwd) {
            boolean flg = false;
            Connection conn = null;
            try {
                conn = new Connection(ip);
                conn.connect();
                flg = conn.authenticateWithPassword(userName, userPwd);//认证
                if (flg) {
                    log.debug("登录成功" + conn);
                    return conn;
                }
            } catch (IOException e) {
                log.debug("登录失败" + e.getMessage());
                e.printStackTrace();
            }
            return conn;
        }
    
        /**
         * 远程执行shll脚本或者命令  使用默认的cmd文件
         *
         * @param cmd 即将执行的命令
         * @return 命令执行完后返回的结果值
         */
        public static String execute(String cmd, String ip, String host, String pwd) {
            String result = "";
            Connection conn = getCon(ip, host, pwd);
            try {
                if (conn != null) {
                    Session session = conn.openSession();//打开一个会话
                    session.execCommand(cmd);//执行命令
                    result = processStdout(session.getStdout(), DEFAULTCHART);
                    //如果为得到标准输出为空，说明脚本执行出错了
                    if (StringUtils.isEmpty(result)) {
                        log.debug("得到标准输出为空, 链接conn:" + conn + ",执行的命令：" + cmd);
                        result = processStdout(session.getStderr(), DEFAULTCHART);
                    } else {
                        log.debug("执行命令成功, 链接conn:" + conn + ",执行的命令：" + cmd);
                    }
                    session.close();
                }
            } catch (IOException e) {
                log.debug("执行命令失败,链接conn:" + conn + ",执行的命令：" + cmd + "  " + e.getMessage());
                e.printStackTrace();
            }
            return result;
        }
    
        /**
         * 解析脚本执行返回的结果集
         *
         * @param in      输入流对象
         * @param charset 编码
         * @return 以纯文本的格式返回
         */
        private static String processStdout(InputStream in, String charset) {
            InputStream stdout = new StreamGobbler(in);
            StringBuffer buffer = new StringBuffer();
            ;
            try {
                BufferedReader br = new BufferedReader(new InputStreamReader(stdout, charset));
                String line = null;
                while ((line = br.readLine()) != null) {
                    buffer.append(line + "\n");
                }
            } catch (UnsupportedEncodingException e) {
                log.error("解析脚本出错：" + e.getMessage());
                e.printStackTrace();
            } catch (IOException e) {
                log.error("解析脚本出错：" + e.getMessage());
                e.printStackTrace();
            }
            return buffer.toString();
        }
    
        /**
         * 执行本地linux命令
         * 部署到linuxs时不需要登录，直接调此方法执行
         * @param cmd
         * @return
         */
        public static String execCommand(String cmd) {
            String result = "";
            String[] cmds = new String[]{"sh", "-c", cmd};
            try {
                Runtime rt = Runtime.getRuntime();
                Process proc = rt.exec(cmds);
                result = processStdout(proc.getInputStream(), DEFAULTCHART);
                //如果为得到标准输出为空，说明脚本执行出错了
                if (StringUtils.isEmpty(result)) {
                    log.debug("执行的命令无结果：" + cmd);
    
                } else {
                    log.debug("执行命令成功, 执行的命令：" + cmd);
                }
    
            } catch (Exception e) {
                log.debug("执行命令失败,执行的命令：" + cmd + "  " + e.getMessage());
                e.printStackTrace();
            }
    
            return result;
    
        }
    
        public static void main(String[] args) {
    
            
            String cmd = "ps  aux | grep tomcat | grep -v grep";
            String execute = null;
            if ("Linux".equalsIgnoreCase(SERVER)) {
                execute = RemoteConnection.getInstance().execCommand(cmd); //linux
            } else {
                execute = RemoteConnection.getInstance().execute(cmd, IP, HOST_NAME, HOST_PWD);//Windows远程调用执行
            }
    
            log.info(execute);
            
        }
    
    }

### 其中有个坑 ###

就是执行复杂命令时如：ps -ef|grep tomcat  时

Process proc = rt.exec(cmd); 执行无结果返回

必须将命令转成数组形式

String[] cmds = new String[]{"sh", "-c", cmd};
    Process proc = rt.exec(cmds);