使用springboot對linux進行操控

kill_clalala發表於2020-11-23

1,在pom中匯入

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

2,編寫工具類

package org.jeecg.modules.system.util;

/**
 * @Description:
 * @Author: LGX
 * @Date: 2020/11/19 10:36
 */


import ch.ethz.ssh2.Connection;
import ch.ethz.ssh2.Session;
import ch.ethz.ssh2.StreamGobbler;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;


import java.io.*;


/**
 * 遠端執行linux的shell script
 * @author Ickes
 * @since  V0.1
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
@Slf4j
@Component
public class RemoteExecuteCommandutil {
    //字元編碼預設是utf-8
    private static String  DEFAULTCHART="UTF-8";
    private Connection conn;

    @Value(value = "${jeecg.linux.ip}")
    public String ip;

    @Value(value = "${jeecg.linux.userName}")
    public String userName;

    @Value(value = "${jeecg.linux.userPwd}")
    public String userPwd;


    /**
     * 遠端登入linux的主機
     * @author Ickes
     * @since  V0.1
     * @return
     *      登入成功返回true,否則返回false
     */
    public Boolean login(){
        boolean flg=false;
        try {
            conn = new Connection(ip);
            conn.connect();//連線
            flg=conn.authenticateWithPassword(userName, userPwd);//認證
        } catch (IOException e) {
            e.printStackTrace();
        }
        return flg;
    }
    /**
     * @author Ickes
     * 遠端執行shll指令碼或者命令
     * @param cmd
     *      即將執行的命令
     * @return
     *      命令執行完後返回的結果值
     * @since V0.1
     */
    public String execute(String cmd){
        String result="";
        try {
            if(login()){
                Session session= conn.openSession();//開啟一個會話
                session.execCommand(cmd);//執行命令
                result=processStdout(session.getStdout(),DEFAULTCHART);
                //如果為得到標準輸出為空,說明指令碼執行出錯了
                if(StringUtils.isBlank(result)){
                    result=processStdout(session.getStderr(),DEFAULTCHART);
                }
                conn.close();
                session.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return result;
    }


    /**
     * @author Ickes
     * 遠端執行shll指令碼或者命令
     * @param cmd
     *      即將執行的命令
     * @return
     *      命令執行成功後返回的結果值,如果命令執行失敗,返回空字串,不是null
     * @since V0.1
     */
    public String executeSuccess(String cmd){
        String result="";
        try {
            if(login()){
                Session session= conn.openSession();//開啟一個會話
                session.execCommand(cmd);//執行命令
                result=processStdout(session.getStdout(),DEFAULTCHART);
                conn.close();
                session.close();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
        return result;
    }

    /**
     * 解析指令碼執行返回的結果集
     * @author Ickes
     * @param in 輸入流物件
     * @param charset 編碼
     * @since V0.1
     * @return
     *       以純文字的格式返回
     */
    private 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) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return buffer.toString();
    }


}

3,yml裡編寫配置資訊

jeecg :
    linux:
      ip: 192.168.xxx.xxx
      userName: root
      userPwd: 123456

4## 注入工具類,編寫命令

    @Autowired
	private RemoteExecuteCommandutil Commandutil;
	@GetMapping(value = "/training")
	public String training(@RequestParam(name="cmd") String cmd){
//		String a = "sh /opt/shops/test1.sh 1 3";
		//命令返回的資訊
		String cmdInformation =Commandutil.execute("source /etc/profile;"+cmd);
		return cmdInformation;
	}

由於ssh連線無法自動獲取環境變數的值,得再執行前面加入source /etc/profile;來手動識別,如果還是不行可以在/etc/profile末尾加入export PATH="$PATH:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games"


相關文章