在Go中监控Linux系统CPU使用率：goprocinfo实战指南

go get github.com/c9s/goprocinfo/linux

package main

import (
    "fmt"
    "log"

    "github.com/c9s/goprocinfo/linux"
)

func main() {
    stat, err := linux.ReadStat("/proc/stat")
    if err != nil {
        log.Fatalf("无法读取 /proc/stat: %v", err)
    }

    // stat.CPUStats[0] 包含了所有CPU核心的总统计信息
    // stat.CPUStats[1] 及以后是每个独立核心的统计信息 (cpu0, cpu1...)
    if len(stat.CPUStats) > 0 {
        totalCPU := stat.CPUStats[0]
        fmt.Printf("总CPU统计 (jiffies):\n")
        fmt.Printf("  User: %d, Nice: %d, System: %d, Idle: %d, IOWait: %d\n",
            totalCPU.User, totalCPU.Nice, totalCPU.System, totalCPU.Idle, totalCPU.IOWait)
    }

    // 遍历并打印每个核心的统计信息
    for i, cpu := range stat.CPUStats {
        if i == 0 { // 索引0是总CPU，已打印
            continue
        }
        fmt.Printf("CPU%d 统计 (jiffies):\n", i-1) // i-1 对应 cpu0, cpu1...
        fmt.Printf("  User: %d, Nice: %d, System: %d, Idle: %d, IOWait: %d\n",
            cpu.User, cpu.Nice, cpu.System, cpu.Idle, cpu.IOWait)
    }
}

package main

import (
    "fmt"
    "log"
    "time"

    "github.com/c9s/goprocinfo/linux"
)

// CPUStatsSnapshot 结构体用于保存CPU时间快照
type CPUStatsSnapshot struct {
    Total uint64 // 所有CPU时间之和
    Idle  uint64 // 空闲时间 + I/O等待时间
}

// GetCPUTimes 从 /proc/stat 读取CPU时间，并返回总CPU和空闲时间的快照
// 对于系统总CPU，我们使用 stat.CPUStats[0]
func GetCPUTimes(cpuInfo linux.CPUStat) CPUStatsSnapshot {
    // 计算所有CPU时间之和
    total := cpuInfo.User + cpuInfo.Nice + cpuInfo.System + cpuInfo.Idle +
        cpuInfo.IOWait + cpuInfo.IRQ + cpuInfo.SoftIRQ + cpuInfo.Steal +
        cpuInfo.Guest + cpuInfo.GuestNice

    // 计算空闲时间（通常包括 Idle 和 IOWait）
    idle := cpuInfo.Idle + cpuInfo.IOWait

    return CPUStatsSnapshot{Total: total, Idle: idle}
}

// CalculateCPUUsage 计算两个快照之间的CPU使用率百分比
func CalculateCPUUsage(prev, curr CPUStatsSnapshot) float64 {
    deltaTotal := curr.Total - prev.Total
    deltaIdle := curr.Idle - prev.Idle

    if deltaTotal == 0 {
        return 0.0 // 避免除以零
    }

    // CPU使用率 = (总时间增量 - 空闲时间增量) / 总时间增量 * 100%
    usage := float64(deltaTotal-deltaIdle) / float64(deltaTotal) * 100.0
    return usage
}

func main() {
    // --- 计算系统总CPU使用率 ---
    fmt.Println("--- 计算系统总CPU使用率 ---")
    // 获取第一个CPU统计快照
    prevStatRaw, err := linux.ReadStat("/proc/stat")
    if err != nil {
        log.Fatalf("错误：无法读取初始 /proc/stat: %v", err)
    }
    if len(prevStatRaw.CPUStats) == 0 {
        log.Fatalf("错误：/proc/stat 中没有CPU统计数据")
    }
    prevTotalCPUStats := GetCPUTimes(prevStatRaw.CPUStats[0])

    // 等待一段时间（例如1秒），以便获取有意义的增量
    time.Sleep(1 * time.Second)

    // 获取第二个CPU统计快照
    currStatRaw, err := linux.ReadStat("/proc/stat")
    if err != nil {
        log.Fatalf("错误：无法读取当前 /proc/stat: %v", err)
    }
    if len(currStatRaw.CPUStats) == 0 {
        log.Fatalf("错误：/proc/stat 中没有CPU统计数据")
    }
    currTotalCPUStats := GetCPUTimes(currStatRaw.CPUStats[0])

    // 计算并打印系统总CPU使用率
    systemCPUUsage := CalculateCPUUsage(prevTotalCPUStats, currTotalCPUStats)
    fmt.Printf("系统总CPU使用率: %.2f%%\n", systemCPUUsage)

    // --- 计算各个核心的CPU使用率 ---
    fmt.Println("\n--- 计算各个核心的CPU使用率 ---")
    // 遍历并计算每个核心的使用率
    for i := 1; i < len(currStatRaw.CPUStats); i++ { // 从索引1开始，因为0是总CPU
        prevCoreStats := GetCPUTimes(prevStatRaw.CPUStats[i])
        currCoreStats := GetCPUTimes(currStatRaw.CPUStats[i])

        coreUsage := CalculateCPUUsage(prevCoreStats, currCoreStats)
        fmt.Printf("CPU%d 使用率: %.2f%%\n", i-1, coreUsage) // i-1 对应 cpu0, cpu1...
    }
}