2

使用 Golang，我正在尝试查找/编写一个函数，该函数将返回 Linux 系统的 CPU 套接字数、每个套接字的内核数和每个内核的线程数。例如，一个服务器可能有 2 个 CPU，每个 CPU 有 4 个内核，使用超线程，每个 CPU 可以处理 8 个线程。

样本输出：

{
  "CPUSockets": "2",
  "CoresPerSocket": "4",
  "ThreadsPerCore": "2"
}

问题： 您知道任何可以提供此信息的 Go 包或伪代码吗？

注意：我查看了 psutil 的各种 go 实现，但找不到一个返回套接字数量或区分核心和线程的实现。我想要的数据很容易通过运行访问，lscpu但我不知道如何使用 Go 访问它。

4

1 回答 1

3

我终于弄明白了，见下文：

package main

import (
    "encoding/json"
    "fmt"
    "os/exec"
    "strconv"
    "strings"
)

type CPUInfo struct {
    Sockets        int32 `json:"sockets"`
    CoresPerSocket int32 `json:"cores_per_socket"`
    ThreadsPerCore int32 `json:"threads_per_core"`
}

func main() {
    out, _ := exec.Command("lscpu").Output()
    outstring := strings.TrimSpace(string(out))
    lines := strings.Split(outstring, "\n")
    c := CPUInfo{}

    for _, line := range lines {
        fields := strings.Split(line, ":")
        if len(fields) < 2 {
            continue
        }
        key := strings.TrimSpace(fields[0])
        value := strings.TrimSpace(fields[1])

        switch key {
        case "Socket(s)":
            t, _ := strconv.Atoi(value)
            c.Sockets = int32(t)
        case "Core(s) per socket":
            t, _ := strconv.Atoi(value)
            c.CoresPerSocket = int32(t)
        case "Thread(s) per core":
            t, _ := strconv.Atoi(value)
            c.ThreadsPerCore = int32(t)
        }
    }

    CPUInfoJSON, _ := json.MarshalIndent(c, "", "  ")
    fmt.Println(string(CPUInfoJSON))
}

输出：

tbenz9@ubuntu-dev: go run socket.go 
{
  "sockets": 1,
  "cores_per_socket": 2,
  "threads_per_core": 2
}
于 2016-02-24T04:11:09.863 回答