komari-agent/monitoring/cpu.go

package monitoring

import (
	"runtime"
	"strconv"
	"strings"
	"time"

	"github.com/shirou/gopsutil/cpu"
)

type CpuInfo struct {
	CPUName         string  `json:"cpu_name"`
	CPUArchitecture string  `json:"cpu_architecture"`
	CPUCores        int     `json:"cpu_cores"`
	CPUUsage        float64 `json:"cpu_usage"`
}

func Cpu() CpuInfo {
	cpuinfo := CpuInfo{}
	info, err := cpu.Info()
	if err != nil {
		cpuinfo.CPUName = "Unknown"
	}
	// multiple CPU
	// 多个 CPU
	if len(info) > 1 {
		cpuCountMap := make(map[string]int)
		for _, cpu := range info {
			cpuCountMap[cpu.ModelName]++
		}
		for modelName, count := range cpuCountMap {
			if count > 1 {
				cpuinfo.CPUName += modelName + " x " + strconv.Itoa(count) + ", "
			} else {
				cpuinfo.CPUName += modelName + ", "
			}
		}
		cpuinfo.CPUName = cpuinfo.CPUName[:len(cpuinfo.CPUName)-2] // Remove trailing comma and space
	} else if len(info) == 1 {
		cpuinfo.CPUName = info[0].ModelName
	}

	cpuinfo.CPUName = strings.TrimSpace(cpuinfo.CPUName)

	cpuinfo.CPUArchitecture = runtime.GOARCH

	cores, err := cpu.Counts(true)
	if err != nil {
		cpuinfo.CPUCores = 1 // Error case
	}
	cpuinfo.CPUCores = cores

	// Get CPU Usage
	percentages, err := cpu.Percent(1*time.Second, false)
	if err != nil {
		cpuinfo.CPUUsage = 0.0 // Error case
	} else {
		cpuinfo.CPUUsage = percentages[0]
	}
	return cpuinfo
}