linux下通过go语言获得系统进程cpu使用情况的方法

时间:2022-05-05
本文章向大家介绍linux下通过go语言获得系统进程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")

    }

}