【Java】+获取Linux服务器的CPU、内存使用率

我看好你哦 發表於 2020-1-21 16:26:00

【Java】+获取Linux服务器的CPU、内存使用率

依赖
<div class="cnblogs_code">
<pre> <dependency>
 <groupId>ch.ethz.ganymed</groupId>
 <artifactId>ganymed-ssh2</artifactId>
 <version>262</version>
 </dependency></pre>
</div>
 
ZgxFileUtil工具
<div class="cnblogs_code"><img src="https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif" id="code_img_closed_d3b0c782-7f64-490f-ba34-5b470cc7f1ec" class="code_img_closed"><img src="https://images.cnblogs.com/OutliningIndicators/ExpandedBlockStart.gif" id="code_img_opened_d3b0c782-7f64-490f-ba34-5b470cc7f1ec" class="code_img_opened" style="display: none">
<div id="cnblogs_code_open_d3b0c782-7f64-490f-ba34-5b470cc7f1ec" class="cnblogs_code_hide">
<pre>package com.file;

import com.ZgxLoggerUtil;
import com.csvreader.CsvReader;
import com.csvreader.CsvWriter;
import org.apache.log4j.Logger;

import javax.swing.*;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

/**
* @author Jarvis
* @version 1.0
* @time 2019/12/20 9:35
*/
public class ZgxFileUtil {
private static Logger log = ZgxLoggerUtil.getLogger(ZgxFileUtil.class);

/**
* 功能：将数据写入csv文件中
*
* @param filePath 文件路径
* @param header 文件头
* @param contents 文件内容
* @return
*/
static public boolean writeCsv(String filePath, String[] header, List<String[]> contents) {
 CsvWriter csvWriter = new CsvWriter(filePath, ',', Charset.forName("GBK"));

 // 写入文件头
 if (header != null) {
 try {
 csvWriter.writeRecord(header);
 } catch (IOException e) {
 e.printStackTrace();
 }
 }

 // 写入文件行
 for (int i = 0; i < contents.size(); i++) {
 try {
 csvWriter.writeRecord(contents.get(i));
 } catch (IOException e) {
 e.printStackTrace();
 }
 }
 csvWriter.close();

 return true;
}

static public boolean writeCsv(String filePath, String[] header, List<String[]> contents, JProgressBar jProgressBar, List<String> fileType) {
 List<CsvWriter> csvWriterList = new ArrayList<>();
 for (int i = 0; i < fileType.size(); i++) {
 csvWriterList.add(new CsvWriter(String.format("%s.%s", filePath, fileType.get(i)), ',', Charset.forName("GBK")));
 }

 // 写入文件头
 if (header != null) {
 try {
 for (int i = 0; i < csvWriterList.size(); i++) {
 csvWriterList.get(i).writeRecord(header);
 }
 } catch (IOException e) {
 e.printStackTrace();
 }
 }

 // 写入文件行
 try {
 for (int i = 0; i < contents.size(); i++) {
 for (int j = 0; j < csvWriterList.size(); j++) {
 csvWriterList.get(j).writeRecord(contents.get(i));
 System.out.println(String.format("写入“%s”文件第%s行", csvWriterList.get(j), i + 1));
 }
 jProgressBar.setValue(i + 1);
 System.out.println(String.format("总：%s 最小：%s 最大：%s 当前：%s", contents.size(), jProgressBar.getMinimum(), jProgressBar.getMaximum(), i));
 }

 for (int i = 0; i < csvWriterList.size(); i++) {
 csvWriterList.get(i).close();
 }
 } catch (IOException e) {
 e.printStackTrace();
 }
 return true;
}

/**
* 功能：读取CSV文件
*
* @param readHeaders 是否要读文件头
* @param filePath 文件路径
* @return 返回读取的数据
*/
static public List<String[]> readCsv(boolean readHeaders, String filePath) {
 List<String[]> contentHangs = new ArrayList<>();
 try {
 CsvReader csvReader = new CsvReader(filePath, ',', Charset.forName("GBK"));
 if (!readHeaders) {
 csvReader.readHeaders();
 }

 while (csvReader.readRecord()) {
 contentHangs.add(csvReader.getValues());
 }

 csvReader.close();
 } catch (IOException e) {
 e.printStackTrace();
 }

 int hang = 0;
 int lie = 0;
 log.info(String.format("第%s行第%s列的数据=%s", hang, lie, contentHangs.get(hang)));
 return contentHangs;
}

/**
* 功能：数据清洗（读取一个旧文件的数据清洗预期的数据后再写入到一个新的文件中）
* 备注：清洗逻辑需变更
*
* @param oldFilePath 原文件路径
* @param newFilePath 清洗后的新文件路径
*/
static public void readAndWriteCsv(String oldFilePath, String newFilePath) {
 /** 【清洗原理】从idb上导出的数据有双引号引起来的所以要清洗下数据取双引号中间的数据*/
 // step1 读取文件清洗数据
 List<String[]> read = readCsv(false, oldFilePath);
 List<String[]> newData = new ArrayList<>();
 for (int i = 0; i < read.size(); i++) {
 String[] substring = {read.get(i).substring(0, read.get(i).length())};
 newData.add(substring);
 }

 // step2 将清洗后的数据写入文件
 writeCsv(newFilePath, null, newData);
}

}</pre>
</div>
View Code</div>
 
代码 
<div class="cnblogs_code">
<pre>package com.alipay.ipay.gn.commontool;

import ch.ethz.ssh2.Connection;
import ch.ethz.ssh2.Session;
import ch.ethz.ssh2.StreamGobbler;
import com.alipay.ipay.gn.commontool.file.ZgxFileUtil;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.log4j.Logger;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.List;

/**
* @author
* @version 1.0
* @time 2019/7/15 20:28
* 
* 类功能说明：
* 1、连接服务器
* 2、执行Linux日志查询命令，返回查询后的日志字符串（以行为单位）
*/
public class LogAuto {
private static Logger log;
private Session ssh;
private String hostName;
private String userName;
private String password;
private int port;

/**
* 连接服务器
*
* @param hostname 服务器IP
* @param port 端口
* @param username 账号
* @param password 密码
*/
public LogAuto(String hostname, int port, String username, String password) {
 this.hostName = hostname;
 this.port = port;
 this.userName = username;
 this.password = password;
 this.log = ZgxLoggerUtil.getLoger(LogAuto.class);
}

/**
* 通过Linux命令查询日志内容
*
* @param command Linux日志查询命令
* @return 返回根据命令查出的日志内容
*/
public List<String> execCom(String command) {
 Connection conn = new Connection(this.hostName, this.port);
 createConnection(conn);
 List<String> logContent = new ArrayList<String>();
 try {
 ssh.execCommand(command);
 } catch (IOException e) {
 e.printStackTrace();
 }

 //将Terminal屏幕上的文字全部打印出来
 InputStream is = new StreamGobbler(ssh.getStdout());
 BufferedReader brs = new BufferedReader(new InputStreamReader(is));
 while (true) {
 String line = null;
 try {
 line = brs.readLine();
 } catch (IOException e) {
 e.printStackTrace();
 }
 if (line == null) {
 break;
 }
// System.out.println(line);
 logContent.add(line);
 }

 return logContent;
}

private void createConnection(Connection connection) {
 //创建连接
 try {
 connection.connect();
 } catch (IOException e) {
 e.printStackTrace();
 }
 try {
 connection.authenticateWithPassword(this.userName, this.password);
 } catch (IOException e) {
 e.printStackTrace();
 }

 //创建与服务器的会话节点
 try {
 setSsh(connection.openSession());
 } catch (IOException e) {
 e.printStackTrace();
 }
}

private void setSsh(Session ssh) {
 this.ssh = ssh;
}

private Session getSsh() {
 return ssh;
}

/**
* 功能：获取指定服务器在当前时间的内存使用率
*
* @param logAuto
* @return 返回使用率（百分比）
*/
public static double getMemoryRate(LogAuto logAuto) {
 String shell = String.format("%s", "free"); // 多个命令用“;”隔开
 List<String> listResult = logAuto.execCom(shell);
 // 打印shell命令执行后的结果
 for (int i = 0; i < listResult.size(); i++) {
 System.out.println(listResult.get(i));
 }

 // 提取内存数据（第二行）
 List<String> mem = ZgxStringUtil.splitString(listResult.get(1), " ");
 Integer usedMemory = Integer.valueOf(mem.get(2));
 Integer allMemory = Integer.valueOf(mem.get(1));
 log.info(String.format("usedMemory=%s;allMemory=%s", usedMemory, allMemory));

 // 计算内存使用率（已使用内存/总内存）
 double f1 = new BigDecimal((float) usedMemory / allMemory).setScale(2, BigDecimal.ROUND_HALF_UP).doubleValue() * 100;
 log.info(String.format("内存使用率=%s%s", f1, "%"));
 return f1;
}

private static Pair getCpuData(LogAuto logAuto) {
 String shell = String.format("%s", "cat /proc/stat"); // 多个命令用“;”隔开
 List<String> listResult = logAuto.execCom(shell);
 // 打印shell命令执行后的结果
 for (int i = 0; i < listResult.size(); i++) {
 System.out.println(listResult.get(i));
 }

 // 提取CPU数据（第一行）
 List<String> mem = ZgxStringUtil.splitString(listResult.get(0), " ");
 mem.remove(0);
 Long allTime1 = 0L;
 for (int i = 0; i < mem.size(); i++) {
 allTime1 += Long.valueOf(mem.get(i));
 }
 System.out.println(String.format("CPU使用总时间=%s;idle=%s", allTime1, mem.get(3)));

 return Pair.of(allTime1, mem.get(3));
}

/**
* 功能：获取指定服务器在当前时间的CPU使用率
*
* @param logAuto
* @return 返回使用率（百分比）
*/
public static double getCpuRate(LogAuto logAuto) {
 Pair cpuData1 = getCpuData(logAuto);
 try {
 Thread.sleep(500);
 } catch (InterruptedException e) {
 e.printStackTrace();
 }
 Pair cpuData2 = getCpuData(logAuto);

 // step1 计算两次的cpu总时间
 Long allTime = Long.valueOf(cpuData2.getLeft().toString()) - Long.valueOf(cpuData1.getLeft().toString());
 // step2 计算两次的cpu剩余时间
 Long idle = Long.valueOf(cpuData2.getRight().toString()) - Long.valueOf(cpuData1.getRight().toString());
 // step3 计算两次的cpu使用时间
 Long used = allTime - idle;
 // step4 计算cpu使用率（cpu使用率 = 使用时间 / 总时间 * 100%）
 double f1 = new BigDecimal((float) used / allTime).setScale(2, BigDecimal.ROUND_HALF_UP).doubleValue() * 100;
 log.info(String.format("CPU使用率=%s%s", f1, "%"));

 return f1;
}

public static void main(String[] args) {
 LogAuto logAuto = new LogAuto("服务器IP", 端口号, "账号", "密码");
 getMemoryRate(logAuto);
 getCpuRate(logAuto);

 String[] header = {"时间,CPU使用率,内存使用率"};
 List<String[]> content = new ArrayList<>();
 // 每分钟获取一次服务器CPU、内存使用率（获取30次）
 for (int i = 0; i < 30; i++) {
 String[] data = {ZgxDateUtil.getNowDate("yyyy-MM-dd HH:mm:ss")
 , String.valueOf(getCpuRate(logAuto))
 , String.valueOf(getMemoryRate(logAuto))};
 content.add(data);
 ZgxFileUtil.writeCsv("服务器性能数据.csv", header, content);
 try {
 Thread.sleep(1000 * 60);
 } catch (InterruptedException e) {
 e.printStackTrace();
 }
 }
}
}</pre>
</div>

</div>
<div id="MySignature" role="contentinfo">
<div style="color: red"> 如果忍耐算是坚强我选择抵抗如果妥协算是努力我选择争取 </div> 
来源：https://www.cnblogs.com/danhuai/p/12222208.html

頁: [1]

琼殿技术论坛's Archiver

【Java】+获取Linux服务器的CPU、内存使用率