使用golang获取CPU使用率

时间:2012-07-06 05:30:18

标签: linux go

My Go程序需要知道所有系统和用户进程的当前cpu使用百分比。

我怎样才能获得?

6 个答案:

答案 0 :(得分:27)

查看这个包http://github.com/c9s/goprocinfo,goprocinfo包为你做解析。

stat, err := linuxproc.ReadStat("/proc/stat")
if err != nil {
    t.Fatal("stat read fail")
}

for _, s := range stat.CPUStats {
    // s.User
    // s.Nice
    // s.System
    // s.Idle
    // s.IOWait
}

答案 1 :(得分:24)

我有一个类似的问题,从未找到轻量级的实现。这是我的解决方案的精简版,可以回答您的具体问题。我就像tylerl推荐的那样对/proc/stat文件进行采样。您会注意到我在样本之间等待3秒以匹配顶部的输出,但是我在1或2秒内也获得了良好的结果。我在go例程中的循环中运行类似的代码,然后当我从其他go例程中需要它时访问cpu用法。

你也可以解析top -n1 | grep -i cpu的输出以获得cpu使用率,但它只在我的linux盒子上采样半秒,并且在重负载时它已经关闭了。当我将它与以下程序同步时,常规顶部似乎非常接近:

package main

import (
    "fmt"
    "io/ioutil"
    "strconv"
    "strings"
    "time"
)

func getCPUSample() (idle, total uint64) {
    contents, err := ioutil.ReadFile("/proc/stat")
    if err != nil {
        return
    }
    lines := strings.Split(string(contents), "\n")
    for _, line := range(lines) {
        fields := strings.Fields(line)
        if fields[0] == "cpu" {
            numFields := len(fields)
            for i := 1; i < numFields; i++ {
                val, err := strconv.ParseUint(fields[i], 10, 64)
                if err != nil {
                    fmt.Println("Error: ", i, fields[i], err)
                }
                total += val // tally up all the numbers to get total ticks
                if i == 4 {  // idle is the 5th field in the cpu line
                    idle = val
                }
            }
            return
        }
    }
    return
}

func main() {
    idle0, total0 := getCPUSample()
    time.Sleep(3 * time.Second)
    idle1, total1 := getCPUSample()

    idleTicks := float64(idle1 - idle0)
    totalTicks := float64(total1 - total0)
    cpuUsage := 100 * (totalTicks - idleTicks) / totalTicks

    fmt.Printf("CPU usage is %f%% [busy: %f, total: %f]\n", cpuUsage, totalTicks-idleTicks, totalTicks)
}

似乎我被允许链接到我在bitbucket上写的完整实现;如果不是,请随意删除。到目前为止,它仅适用于Linux:systemstat.go

答案 2 :(得分:17)

获取CPU使用的机制取决于操作系统,因为这些数字对于不同的操作系统内核意味着略有不同。

在Linux上,您可以通过读取/proc/文件系统中的伪文件来查询内核以获取最新的统计信息。当您读取它们以反映机器的当前状态时,它们会即时生成。

具体而言,每个进程的/proc/<pid>/stat文件包含关联的进程记帐信息。它记录在proc(5)中。您特别感谢字段utimestimecutimecstime(从第14个字段开始)。

您可以轻松地计算百分比:只需读取数字,等待一段时间间隔,然后再次阅读。取差价,除以你等待的时间,这是你的平均值。这正是top程序所做的(以及执行相同服务的所有其他程序)。请记住,如果CPU数超过1,则可以使用超过100%的cpu。

如果您只想要系统范围的摘要(在/proc/stat中报告 - 使用相同的技术计算平均值,但您只需要读取一个文件。

答案 3 :(得分:13)

您可以使用os.exec包执行ps命令并获取结果。

这是一个发出ps aux命令的程序,解析结果并在linux上打印所有进程的CPU使用情况:

package main

import (
    "bytes"
    "log"
    "os/exec"
    "strconv"
    "strings"
)

type Process struct {
    pid int
    cpu float64
}

func main() {
    cmd := exec.Command("ps", "aux")
    var out bytes.Buffer
    cmd.Stdout = &out
    err := cmd.Run()
    if err != nil {
        log.Fatal(err)
    }
    processes := make([]*Process, 0)
    for {
        line, err := out.ReadString('\n')
        if err!=nil {
            break;
        }
        tokens := strings.Split(line, " ")
        ft := make([]string, 0)
        for _, t := range(tokens) {
            if t!="" && t!="\t" {
                ft = append(ft, t)
            }
        }
        log.Println(len(ft), ft)
        pid, err := strconv.Atoi(ft[1])
        if err!=nil {
            continue
        }
        cpu, err := strconv.ParseFloat(ft[2], 64)
        if err!=nil {
            log.Fatal(err)
        }
        processes = append(processes, &Process{pid, cpu})
    }
    for _, p := range(processes) {
        log.Println("Process ", p.pid, " takes ", p.cpu, " % of the CPU")
    }
}

答案 4 :(得分:4)

以下是一个独立于操作系统的解决方案,使用Cgo来利用clock()提供的C standard library功能:

//#include <time.h>
import "C"
import "time"

var startTime = time.Now()
var startTicks = C.clock()

func CpuUsagePercent() float64 {
    clockSeconds := float64(C.clock()-startTicks) / float64(C.CLOCKS_PER_SEC)
    realSeconds := time.Since(startTime).Seconds()
    return clockSeconds / realSeconds * 100
}

答案 5 :(得分:0)

我最近不得不从Raspberry Pi(Raspbian OS)中获取CPU使用率,并使用github.com/c9s/goprocinfo结合此处提出的内容:

这个想法来自htop源代码,并且要进行两次测量(上一次/当前)以计算CPU使用率:

func calcSingleCoreUsage(curr, prev linuxproc.CPUStat) float32 {

  PrevIdle := prev.Idle + prev.IOWait
  Idle := curr.Idle + curr.IOWait

  PrevNonIdle := prev.User + prev.Nice + prev.System + prev.IRQ + prev.SoftIRQ + prev.Steal
  NonIdle := curr.User + curr.Nice + curr.System + curr.IRQ + curr.SoftIRQ + curr.Steal

  PrevTotal := PrevIdle + PrevNonIdle
  Total := Idle + NonIdle
  // fmt.Println(PrevIdle, Idle, PrevNonIdle, NonIdle, PrevTotal, Total)

  //  differentiate: actual value minus the previous one
  totald := Total - PrevTotal
  idled := Idle - PrevIdle

  CPU_Percentage := (float32(totald) - float32(idled)) / float32(totald)

  return CPU_Percentage
}

您还可以查看https://github.com/tgogos/rpi_cpu_memory