Java 串連遠程Linux 伺服器執行 shell 指令碼查看 CPU、記憶體、硬碟資訊

標籤：inpu   本地   conf   命令   ash   tmpfs   進程資訊   stream   buffer   

pom.xml jar 包支援

　　　　<dependency>            <groupId>com.jcraft</groupId>            <artifactId>jsch</artifactId>            <version>0.1.53</version>        </dependency>

代碼：

package com.spring.bean.annotation;import java.io.BufferedReader;import java.io.IOException;import java.io.InputStream;import java.io.InputStreamReader;import java.util.HashMap;import java.util.Map;import com.jcraft.jsch.Channel;import com.jcraft.jsch.ChannelExec;import com.jcraft.jsch.JSch;import com.jcraft.jsch.JSchException;import com.jcraft.jsch.Session;/** * 遠程調用Linux shell 命令 * * @author wei.Li by 14-9-2. */public class LinuxStateForShell {    public static final String CPU_MEM_SHELL = "top -b -n 1";    public static final String FILES_SHELL = "df -hl";    public static final String[] COMMANDS = {CPU_MEM_SHELL, FILES_SHELL};    public static final String LINE_SEPARATOR = System.getProperty("line.separator");    private static Session session;    /**     * 串連到指定的HOST     *     * @return isConnect     * @throws JSchException JSchException     */    private static boolean connect(String user, String passwd, String host) {        JSch jsch = new JSch();        try {            session = jsch.getSession(user, host, 22);            session.setPassword(passwd);            java.util.Properties config = new java.util.Properties();            config.put("StrictHostKeyChecking", "no");            session.setConfig(config);            session.connect();        } catch (JSchException e) {            e.printStackTrace();            System.out.println("connect error !");            return false;        }        return true;    }    /**     * 遠端連線Linux 伺服器 執行相關的命令     *     * @param commands 執行的指令碼     * @param user     遠端連線的使用者名稱     * @param passwd   遠端連線的密碼     * @param host     遠端連線的主機IP     * @return 最終命令返回資訊     */    public static Map<String, String> runDistanceShell(String[] commands, String user, String passwd, String host) {        if (!connect(user, passwd, host)) {            return null;        }        Map<String, String> map = new HashMap<>();        StringBuilder stringBuffer;        BufferedReader reader = null;        Channel channel = null;        try {            for (String command : commands) {                stringBuffer = new StringBuilder();                channel = session.openChannel("exec");                ((ChannelExec) channel).setCommand(command);                channel.setInputStream(null);                ((ChannelExec) channel).setErrStream(System.err);                channel.connect();                InputStream in = channel.getInputStream();                reader = new BufferedReader(new InputStreamReader(in));                String buf;                while ((buf = reader.readLine()) != null) {                    //捨棄PID 進程資訊                    if (buf.contains("PID")) {                        break;                    }                    stringBuffer.append(buf.trim()).append(LINE_SEPARATOR);                }                //每個命令儲存自己返回資料-用於後續對返回資料進行處理                map.put(command, stringBuffer.toString());            }        } catch (IOException | JSchException e) {            e.printStackTrace();        } finally {            try {                if (reader != null) {                    reader.close();                }            } catch (IOException e) {                e.printStackTrace();            }            if (channel != null) {                channel.disconnect();            }            session.disconnect();        }        return map;    }    /**     * 直接在本地執行 shell     *     * @param commands 執行的指令碼     * @return 執行結果資訊     */    public static Map<String, String> runLocalShell(String[] commands) {        Runtime runtime = Runtime.getRuntime();        Map<String, String> map = new HashMap<>();        StringBuilder stringBuffer;        BufferedReader reader;        Process process;        for (String command : commands) {            stringBuffer = new StringBuilder();            try {                process = runtime.exec(command);                InputStream inputStream = process.getInputStream();                reader = new BufferedReader(new InputStreamReader(inputStream));                String buf;                while ((buf = reader.readLine()) != null) {                    //捨棄PID 進程資訊                    if (buf.contains("PID")) {                        break;                    }                    stringBuffer.append(buf.trim()).append(LINE_SEPARATOR);                }            } catch (IOException e) {                e.printStackTrace();                return null;            }            //每個命令儲存自己返回資料-用於後續對返回資料進行處理            map.put(command, stringBuffer.toString());        }        return map;    }    /**     * 處理 shell 返回的資訊     * <p>     * 具體處理過程以伺服器返回資料格式為準     * 不同的Linux 版本返回資訊格式不同     *     * @param result shell 返回的資訊     * @return 最終處理後的資訊     */    private static String disposeResultMessage(Map<String, String> result) {        StringBuilder buffer = new StringBuilder();        for (String command : COMMANDS) {            String commandResult = result.get(command);            if (null == commandResult) continue;            if (command.equals(CPU_MEM_SHELL)) {                String[] strings = commandResult.split(LINE_SEPARATOR);                //將返回結果按分行符號分割                for (String line : strings) {                    line = line.toUpperCase();//轉大寫處理                    //處理CPU Cpu(s): 10.8%us,  0.9%sy,  0.0%ni, 87.6%id,  0.7%wa,  0.0%hi,  0.0%si,  0.0%st                    if (line.startsWith("CPU(S):")) {                        String cpuStr = "CPU 使用者使用佔有率:";                        try {                            cpuStr += line.split(":")[1].split(",")[0].replace("US", "");                        } catch (Exception e) {                            e.printStackTrace();                            cpuStr += "計算過程出錯";                        }                        buffer.append(cpuStr).append(LINE_SEPARATOR);                        //處理記憶體 Mem:  66100704k total, 65323404k used,   777300k free,    89940k buffers                    } else if (line.startsWith("MEM")) {                        String memStr = "記憶體使用量情況:";                        try {                            memStr += line.split(":")[1]                                    .replace("TOTAL", "總計")                                    .replace("USED", "已使用")                                    .replace("FREE", "空閑")                                    .replace("BUFFERS", "緩衝");                        } catch (Exception e) {                            e.printStackTrace();                            memStr += "計算過程出錯";                            buffer.append(memStr).append(LINE_SEPARATOR);                            continue;                        }                        buffer.append(memStr).append(LINE_SEPARATOR);                    }                }            } else if (command.equals(FILES_SHELL)) {                //處理系統磁碟狀態                buffer.append("系統磁碟狀態:");                try {                    buffer.append(disposeFilesSystem(commandResult)).append(LINE_SEPARATOR);                } catch (Exception e) {                    e.printStackTrace();                    buffer.append("計算過程出錯").append(LINE_SEPARATOR);                }            }        }        return buffer.toString();    }    //處理系統磁碟狀態    /**     * Filesystem            Size  Used Avail Use% Mounted on     * /dev/sda3             442G  327G   93G  78% /     * tmpfs                  32G     0   32G   0% /dev/shm     * /dev/sda1             788M   60M  689M   8% /boot     * /dev/md0              1.9T  483G  1.4T  26% /ezsonar     *     * @param commandResult 處理系統磁碟狀態shell執行結果     * @return 處理後的結果     */    private static String disposeFilesSystem(String commandResult) {        String[] strings = commandResult.split(LINE_SEPARATOR);        // final String PATTERN_TEMPLATE = "([a-zA-Z0-9%_/]*)\\s";        int size = 0;        int used = 0;        for (int i = 0; i < strings.length - 1; i++) {            if (i == 0) continue;            int temp = 0;            for (String s : strings[i].split("\\b")) {                if (temp == 0) {                    temp++;                    continue;                }                if (!s.trim().isEmpty()) {                    if (temp == 1) {                        size += disposeUnit(s);                        temp++;                    } else {                        used += disposeUnit(s);                        temp = 0;                    }                }            }        }        return new StringBuilder().append("大小 ").append(size).append("G , 已使用").append(used).append("G ,空閑")                .append(size - used).append("G").toString();    }    /**     * 處理單位轉換     * K/KB/M/T 最終轉換為G 處理     *     * @param s 帶單位的資料字串     * @return 以G 為單位處理後的數值     */    private static int disposeUnit(String s) {        try {            s = s.toUpperCase();            String lastIndex = s.substring(s.length() - 1);            String num = s.substring(0, s.length() - 1);            int parseInt = Integer.parseInt(num);            if (lastIndex.equals("G")) {                return parseInt;            } else if (lastIndex.equals("T")) {                return parseInt * 1024;            } else if (lastIndex.equals("M")) {                return parseInt / 1024;            } else if (lastIndex.equals("K") || lastIndex.equals("KB")) {                return parseInt / (1024 * 1024);            }        } catch (NumberFormatException e) {            e.printStackTrace();            return 0;        }        return 0;    }    public static void main(String[] args) {        Map<String, String> result = runDistanceShell(COMMANDS, "dell", "1", "192.168.1.122");        System.out.println(disposeResultMessage(result));    }}

Java 串連遠程Linux 伺服器執行 shell 指令碼查看 CPU、記憶體、硬碟資訊