go读取Linux下cpu占用率、cpu温度、内存占用率及磁盘占用率

“携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第5天，点击查看活动详情”

1. 前言

类似于电脑管家可以获取到设备的pu占用率、cpu温度、内存占用率及磁盘占用率，一些设备上也可以用，可以将该数据展示到web页面上，比如家里的路由器可以做一个这样的web来展示。

后续会利用相关界面库来开发一个跨平台的小工具，这里的内容是该小工具的一部分，但是也挺常用的。

2. 代码

接口获取到的数据注意Go的数据类型转换，Go还是比较严格的。

package main

import (
	"bytes"
	"fmt"
	"os/exec"
	"strconv"
	"strings"
	"time"

	"github.com/shirou/gopsutil/cpu"
	"github.com/shirou/gopsutil/disk"
	"github.com/shirou/gopsutil/mem"

	"github.com/wonderivan/logger"
)

//获取CPU使用率
func GetCpuPercent() (float64, error) {
	percent, err := cpu.Percent(time.Second, false)
	if err != nil {
		logger.Error(err)
		return 0, err
	}
	return percent[0], nil
}

//获取CPU温度
func GetCpuTemp() (string, error) {
	cmd := exec.Command("cat", "/sys/class/thermal/thermal_zone0/temp")
	var out bytes.Buffer
	cmd.Stdout = &out
	err := cmd.Run()
	if err != nil {
		logger.Error(err)
		return "", err
	}
	logger.Debug(out.String())
	tempStr := strings.Replace(out.String(), "\n", "", -1)
	temp, err := strconv.Atoi(tempStr)
	if err != nil {
		logger.Error(err)
		return "", err
	}
	temp = temp / 1000
	logger.Debug(temp)
	return strconv.Itoa(temp), nil
}

//获取内存使用率
func GetMemPercent() (float64, error) {
	memInfo, err := mem.VirtualMemory()
	if err != nil {
		logger.Error(err)
		return 0, err
	}
	return memInfo.UsedPercent, nil
}

//获取硬盘使用率
func GetDiskPercent() (float64, error) {
	parts, err := disk.Partitions(true)
	if err != nil {
		logger.Error(err)
		return 0, err
	}
	diskInfo, err := disk.Usage(parts[0].Mountpoint)
	if err != nil {
		logger.Error(err)
		return 0, err
	}
	return diskInfo.UsedPercent, nil
}

func main() {
	cpuPer, err := GetCpuPercent()
	if err != nil {
		logger.Error(err)
	} else {
		cpuPer, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", cpuPer), 64)
		cpuPerStr := strconv.FormatFloat(cpuPer, 'f', -1, 64)
		logger.Debug("cpu:", cpuPerStr)
	}

	cpuTemp, err := GetCpuTemp()
	if err != nil {
		logger.Error(err)
	} else {
		logger.Debug("cpuTemp:", cpuTemp)
	}

	diskPer, err := GetDiskPercent()
	if err != nil {
		logger.Error(err)
	} else {
		diskPer, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", diskPer), 64)
		diskPerStr := strconv.FormatFloat(diskPer, 'f', -1, 64)
		logger.Debug("diskPer:", diskPerStr)
	}
	memPer, err := GetMemPercent()
	if err != nil {
		logger.Error(err)
	} else {
		memPer, _ = strconv.ParseFloat(fmt.Sprintf("%.2f", memPer), 64)
		memPerStr := strconv.FormatFloat(memPer, 'f', -1, 64)
		logger.Debug("memPer:", memPerStr)
	}
}

3. 测试结果

2021-08-27 09:48:48 [DEBG] [test.go:82] cpu: 1.01
2021-08-27 09:48:49 [DEBG] [test.go:38] 58000

2021-08-27 09:48:49 [DEBG] [test.go:46] 58
2021-08-27 09:48:49 [DEBG] [test.go:89] cpuTemp: 58
2021-08-27 09:48:49 [DEBG] [test.go:98] diskPer: 23.34
2021-08-27 09:48:49 [DEBG] [test.go:106] memPer: 5.95

4. 编译

$ ./compile.sh "arm"