com.didiglobal.knowframework.system.metrcis.service.linux.LinuxCpuTime Maven / Gradle / Ivy

Go to download
package com.didiglobal.knowframework.system.metrcis.service.linux;

import com.didiglobal.knowframework.system.metrcis.util.FileUtils;
import com.didiglobal.knowframework.system.metrcis.util.MathUtil;
import org.apache.commons.lang3.StringUtils;

public class LinuxCpuTime {

    public long          all;                    // 整个系统从启动到现在的cpu耗时
    public long          allSubIdle;             // 整个系统从启动到现在除了空闲状态的cpu耗时
    public long          process;                // 当前进程从启动到现在的cpu耗时

    // 获得当前进程从启动到现在的cpu耗时
    private final String PROCESS_PATH;

    // 获得整个系统从启动到现在的cpu耗时
    private final String ALL_PATH = "/proc/stat";

    //cpu 核 数
    private int          cpuNum;

    public LinuxCpuTime(int cpuNum) throws Exception {
        this.cpuNum = cpuNum;
        long[] allTime = getAllTime();
        this.all = allTime[0];
        this.allSubIdle = allTime[1];
        PROCESS_PATH = "";
    }

    // 获得当前的耗时
    public LinuxCpuTime(long pid, int cpuNum) throws Exception {
        this.cpuNum = cpuNum;
        PROCESS_PATH = "/proc/" + pid + "/stat";
        long[] allTime = getAllTime();
        this.all = allTime[0];
        this.allSubIdle = allTime[1];
        this.process = getProcessTime();
    }

    // 根据计算cpu耗时的差值，计算这段时间中的cpu耗时
    public float getUsage(LinuxCpuTime before) {
        if (all - before.all == 0) {
            return 0.0f;
        }
        long cpuUsage = all - before.all;
        long proUsage = process - before.process;
        return MathUtil.divideWith2Digit(proUsage * 100, cpuUsage).floatValue();
    }

    // 获取系统CPU使用率
    public float getSystemUsage(LinuxCpuTime before) {
        if (all - before.all == 0) {
            return 0.0f;
        }
        float allCpuUsage = all - before.all;
        float allSubIdleCpuUsage = allSubIdle - before.allSubIdle;
        return MathUtil.divideWith2Digit(allSubIdleCpuUsage * 100, allCpuUsage).floatValue();
    }

    private long[] getAllTime() throws Exception {
        String[] array = FileUtils.readFirstLine(ALL_PATH);
        if (array[0] == null || !array[0].trim().equals("cpu")) {
            throw new Exception(ALL_PATH + " first line is not init with 'cpu'");
        }

        if (array.length < 8) {
            throw new Exception(ALL_PATH + " first line is too short");
        }

        long count = 0;
        for (int i = 1; i < array.length; i++) {
            if (StringUtils.isNotBlank(array[i])) {
                count += Long.parseLong(array[i]);
            }
        }
        long allSubIdleTime = count - Long.parseLong(array[4]);
        return new long[] { count, allSubIdleTime };
    }

    private long getProcessTime() throws Exception {
        String[] array = FileUtils.readFirstLine(PROCESS_PATH);
        if (array.length < 17) {
            throw new Exception(PROCESS_PATH + " first line is too short");
        }

        long count = 0;
        count += Long.parseLong(array[13]);
        count += Long.parseLong(array[14]);
        count += Long.parseLong(array[15]);
        count += Long.parseLong(array[16]);
        return count;
    }

}