4566704
30a0889ffa
- Start 后台每秒采集(CPU/内存/磁盘/网络/进程/连接/主机), GetInfo 取快照 - 基于 gopsutil, 附测试、examples/monitor 例程与包 README
94 lines
2.1 KiB
Go
94 lines
2.1 KiB
Go
package monitor
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
|
||
"github.com/shirou/gopsutil/v3/cpu"
|
||
)
|
||
|
||
// CPU CPU 采集信息
|
||
type CPU struct {
|
||
CPU int32 `json:"cpu"`
|
||
VendorID string `json:"vendorId"`
|
||
ModelName string `json:"modelName"`
|
||
Mhz float64 `json:"mhz"`
|
||
Cores int32 `json:"cores"`
|
||
}
|
||
|
||
// CpuBaseInfo CPU 基本信息（型号/核心数/频率等，gopsutil cpu.Info）
|
||
type CpuBaseInfo struct {
|
||
Core int `json:"core"`
|
||
Logic int `json:"logic"`
|
||
Info []CPU `json:"info"`
|
||
}
|
||
|
||
// CpuInfo 每逻辑核心使用率
|
||
type CpuInfo struct {
|
||
CpuBaseInfo
|
||
CpuUsedInfo
|
||
}
|
||
|
||
// CpuUsedInfo CPU 使用率明细（含用户/系统/空闲等占比）
|
||
type CpuUsedInfo struct {
|
||
UsedPercent float64 `json:"usedPercent"`
|
||
Used []float64 `json:"used"`
|
||
}
|
||
|
||
// CpuInfoLite 精简版 CPU 信息
|
||
type CpuInfoLite struct {
|
||
Core int `json:"core"`
|
||
Logic int `json:"logic"`
|
||
Name []string `json:"name"`
|
||
UsedPercent float64 `json:"usedPercent"`
|
||
Used []float64 `json:"used"`
|
||
}
|
||
|
||
// getCpuInfo 采集 CPU 基础信息与每核心使用率
|
||
func getCpuInfo() (CpuInfo, error) {
|
||
info := CpuInfo{}
|
||
cpuInfo, err := cpu.Info()
|
||
if err != nil {
|
||
return info, err
|
||
}
|
||
info.Info = make([]CPU, len(cpuInfo))
|
||
for i, v := range cpuInfo {
|
||
info.Info[i].CPU = v.CPU
|
||
info.Info[i].VendorID = v.VendorID
|
||
info.Info[i].ModelName = v.ModelName
|
||
info.Info[i].Mhz = v.Mhz
|
||
info.Info[i].Cores = v.Cores
|
||
}
|
||
|
||
core, err := cpu.Counts(false)
|
||
if err != nil {
|
||
return info, err
|
||
}
|
||
logic, err := cpu.Counts(true)
|
||
if err != nil {
|
||
return info, err
|
||
}
|
||
|
||
info.Core = core
|
||
info.Logic = logic
|
||
return info, nil
|
||
}
|
||
|
||
// getCpuUsed 采集 CPU 使用率明细
|
||
func getCpuUsed() (CpuUsedInfo, error) {
|
||
info := CpuUsedInfo{}
|
||
percent, err := cpu.Percent(0, true)
|
||
if err != nil {
|
||
return info, err
|
||
}
|
||
var usedPercent float64 = 0.0
|
||
for i, v := range percent {
|
||
percent[i], _ = strconv.ParseFloat(fmt.Sprintf("%.1f", v), 64)
|
||
usedPercent += v
|
||
}
|
||
usedPercent, _ = strconv.ParseFloat(fmt.Sprintf("%.1f", usedPercent/float64(len(percent))), 64)
|
||
info.Used = percent
|
||
info.UsedPercent = usedPercent
|
||
return info, nil
|
||
}
|