Reputation: 446
Using Golang I'm trying to find/write a function that would return the number of CPU Sockets, Cores per socket, and Threads per core a Linux system. For example, a server might have 2 CPUs each with 4 cores, with hyperthreading, it could handle 8 threads per CPU.
Sample Output:
{
"CPUSockets": "2",
"CoresPerSocket": "4",
"ThreadsPerCore": "2"
}
Question: Do you know of any Go Package or pseudo code that could provide this information?
Note: I've looked at various go implementations of psutil but I can't find one that returns the number of sockets or distinguishes between Cores and Threads. The data I want is very easily accessible by running lscpu
but I don't know how to access that using Go.
Upvotes: 2
Views: 2843
Reputation: 446
I ended up figuring it out, see below:
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))
}
Output:
tbenz9@ubuntu-dev: go run socket.go
{
"sockets": 1,
"cores_per_socket": 2,
"threads_per_core": 2
}
Upvotes: 3