java通过ssh命令远程操作Linux

注意：通过远程操作Linux系统方式，如果涉及到护网行动 可能失效

1.导入java依赖

    <dependency>
        <groupId>com.jcraft</groupId>
        <artifactId>jsch</artifactId>
    <version>0.1.53</version>
　　
    <dependency>
        <groupId>commons-io</groupId>
        <artifactId>commons-io</artifactId>
        <version>2.6</version>
    </dependency>        

 

2.编写工具类

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.InputStream;
import java.util.Iterator;
import java.util.Vector;

import org.apache.commons.io.IOUtils;

import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import com.jcraft.jsch.SftpException;


public class SSHLinuxUtils {

    protected SSHLinuxUtils() {
    }

    /**
     * 创建ssh session
     * 
     * @param host     ip
     * @param port     port
     * @param userName 用户名
     * @param password 密码
     * @return
     * @throws Exception
     */
    public static Session createSession(final String host, final int port, final String userName, final String password)
            throws Exception {
        JSch jsch = new JSch();
        Session session = jsch.getSession(userName, host, port);
        session.setConfig("StrictHostKeyChecking", "no");
        session.setPassword(password);
        session.connect();
        return session;
    }

    /**
     * 获取Java进程id
     * 
     * @param session  session
     * @param command  sh命令
     * @param language 进程类型
     * @return 进程id
     * @throws Exception
     */
    public static String getPid(Session session, final String command, int port) throws Exception {
        ChannelExec channel = (ChannelExec) session.openChannel("exec");
        String pid = null;
        try {
            InputStream inputStream = channel.getInputStream();
            channel.setCommand(command);
            channel.setErrStream(System.err);
            channel.connect();
            pid = IOUtils.toString(inputStream, "UTF-8");
            String[] split = pid.split("\n");
            for (String s : split) {
                if(s.contains("--"+port+"--")){
                    System.out.println(s);
                    pid=s.substring(0,16).split("root")[1].trim();
                    break;
                }
            }
            
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }
        return pid;

    }
    
    
    /**
     * 获取Java进程id
     * 
     * @param session  session
     * @param command  sh命令
     * @param language 进程类型
     * @return 进程id
     * @throws Exception
     */
    public static String getPid(Session session, final String command, String topicGroup) throws Exception {
        ChannelExec channel = (ChannelExec) session.openChannel("exec");
        String pid = null;
        try {
            InputStream inputStream = channel.getInputStream();
            channel.setCommand(command);
            channel.setErrStream(System.err);
            channel.connect();
            String text = IOUtils.toString(inputStream, "UTF-8");
            String[] split = text.split("\n");
            for (String sh : split) {
                if(sh.contains("-jar")) {
                    sh=sh.substring(0,sh.indexOf("-jar")).trim();
                    if(sh.endsWith(topicGroup.trim())){
                        System.out.println(sh);
                        pid=sh.substring(0,16).split("root")[1].trim();
                        break;
                    }
                }
                
            }
            
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }
        return pid;

    }

    /**
     * 执行sh命令
     * 
     * @param session session
     * @param command sh命令
     * @throws JSchException
     * @throws IOException
     */
    public static void exeCommand(Session session, final String command) throws Exception {
        ChannelExec channel = (ChannelExec) session.openChannel("exec");
        try {
            // channel.setPty(true);
            channel.setCommand(command);
            channel.setErrStream(System.err);
            channel.connect();
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }

    }

    /**
     * 上传文件
     * 
     * @param session  session
     * @param fileDest 文件地址
     * @throws Exception
     */
    public static void upload(Session session, final String fileDest) throws Exception {
        ChannelSftp channel = ((ChannelSftp) session.openChannel("sftp"));
        try {
            channel.connect();
            channel.put(new FileInputStream("/start.sh"), fileDest);
            // 添加权限
            channel.chmod(777, fileDest);
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }
    }

    public static void deleteDir(Session session, final String fileDest) throws Exception {
        ChannelSftp channel = ((ChannelSftp) session.openChannel("sftp"));
        try {
            channel.connect();
            channel.rmdir(fileDest);
            // 添加权限
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }
    }

    /**
     * 写入命令到start.sh
     *
     * @param sh 获取当前进程运行以来的配置文件地址
     */
    public static void createTempFile(String createFileDir, String sh) throws Exception {
        File file = new File(createFileDir);
        file.createNewFile();
        BufferedWriter br = new BufferedWriter(new FileWriter(file));
        try {
            br.write(sh);
        } finally {
            br.flush();
            br.close();
        }
    }

    /**
     * 根据路径回去文件名称
     * 
     * @param session
     * @param dir     获取文件名称的路径
     * @return
     * @throws JSchException
     */
    @SuppressWarnings("unchecked")
    public static String getFileNameByDir(Session session, String dir) throws JSchException {
        ChannelSftp channel = ((ChannelSftp) session.openChannel("sftp"));
        try {
            channel.connect();
            Vector<ChannelSftp.LsEntry> vector = channel.ls(dir);
            Iterator<ChannelSftp.LsEntry> iterator = vector.iterator();
            while (iterator.hasNext()) {
                ChannelSftp.LsEntry next = iterator.next();
                String filename = next.getFilename();
                if (filename.endsWith(".jar")) {
                    return filename;
                }
            }
        } catch (SftpException e) {
            e.printStackTrace();
        } finally {
            if (channel != null) {
                channel.disconnect();
            }
        }
        return null;
    }
}

调用实例

public void test(){
    Session session =SSHLinuxUtils.createSession(host,port,userName,password)
    try{    
        SSHLinuxUtils.exeCommand(session, "rm -rf /home/test" );
    }finally {
    　　if (session != null) {
       　　session.disconnect();
    　　}
    }
        
}