java通过ssh指令调用shell，上传文件，下载文件

男娘i 2024-02-19 19:36 5阅读 0赞

#### 导入maven依赖 ####

<dependency>
                <groupId>ch.ethz.ganymed</groupId>
                <artifactId>ganymed-ssh2</artifactId>
                <version>262</version>
            </dependency>

#### 编写的公共类 ####

package com.lys.utils;
    
    import ch.ethz.ssh2.*;
    import org.apache.commons.io.IOUtils;
    
    import java.io.*;
    import java.nio.charset.Charset;
    
    /**
     * @ClassName RemoteShellExecutor
     * @Description 远程登陆的脚本
     * @Author liuysh
     * @Date 2019/3/30 14:01
     **/
    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);
        }
    
    
        /**
         * 执行shell命令
         * @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()) {
                    // Open a new {@link Session} on this connection
                    Session session = conn.openSession();
                    // Execute a command on the remote machine.
                    session.execCommand(cmds);
    
                    stdOut = new StreamGobbler(session.getStdout());
                    outStr = processStream(stdOut, charset);
    
                    stdErr = new StreamGobbler(session.getStderr());
                    outErr = processStream(stdErr, charset);
    
                    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 Exception
         */
        private String processStream(InputStream in, String charset) throws Exception {
            byte[] buf = new byte[1024];
            StringBuilder sb = new StringBuilder();
            while (in.read(buf) != -1) {
                sb.append(new String(buf, charset));
            }
            return sb.toString();
        }
    
    
    
        /**
         * 将远程主机的文件下载到指定本机目录
         * @param remoteFile
         * @param localTargetDirectory
         * @throws Exception
         */
        public void getFile(String remoteFile, String localTargetDirectory) throws Exception{
            try {
    
                if (login()) {
                    SCPClient client = new SCPClient(conn);
                    SCPInputStream fis=client.get(remoteFile);
                    FileOutputStream fos = new FileOutputStream(localTargetDirectory);
                    // 构造一个长度为1024的字节数组
                    byte[] buffer = new byte[1024];
                    int len = 0;
                    //读取
                    while ((len = fis.read(buffer)) != -1) {
                        //写入另一个文件
                        fos.write(buffer);
                    }
                } else {
                    // ("登录远程机器失败" + ip);
                    // 自定义异常类 实现略
                }
            } catch (IOException ex) {
                ex.getStackTrace();
            }
        }
    
        /**
         * 将本地文件上传到远程主机指定目录
         * @param localFile
         * @param remoteTargetDirectory
         * @param mode
         */
        public void uploadFile(String localFile, String remoteTargetDirectory, String mode){
            try {
    
    
                if (login()) {
                    File  file=new File(localFile);
                    SCPClient client = new SCPClient(conn);
                    SCPOutputStream os=client.put(file.getName(), file.length(), remoteTargetDirectory,  mode);
    
                    byte[] b = new byte[4096];
                    FileInputStream fis = new FileInputStream(file);
                    int i;
                    while ((i = fis.read(b)) != -1) {
                        os.write(b, 0, i);
                    }
                    os.flush();
                    fis.close();
                    os.close();
    
                } else {
                    // ("登录远程机器失败" + ip);
                    // 自定义异常类 实现略
                }
            } catch (IOException ex) {
                ex.getStackTrace();
            }
        }
    
        public static void main(String args[]) throws Exception {
            RemoteShellExecutor executor = new RemoteShellExecutor("XXXXX", "root", "liuyunsheng");
            System.out.println(executor.exec("kubectl get ns --all-namespaces -o wide "));
    //        executor.getFile("/root/lys/my-pvc.yaml","D:/test/my-pvc.yaml");
    //        executor.uploadFile("D:/test/a.js","/root/lys/","0222");
        }
    
    }