36 lines
612 B
Go
36 lines
612 B
Go
package devices
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
psCpu "github.com/shirou/gopsutil/cpu"
|
|
)
|
|
|
|
func init() {
|
|
f := func(cpus map[string]int, iv time.Duration, l bool) map[string]error {
|
|
cpuCount, err := psCpu.Counts(l)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
formatString := "CPU%1d"
|
|
if cpuCount > 10 {
|
|
formatString = "CPU%02d"
|
|
}
|
|
vals, err := psCpu.Percent(iv, l)
|
|
if err != nil {
|
|
return map[string]error{"gopsutil": err}
|
|
}
|
|
for i := 0; i < len(vals); i++ {
|
|
key := fmt.Sprintf(formatString, i)
|
|
v := vals[i]
|
|
if v > 100 {
|
|
v = 100
|
|
}
|
|
cpus[key] = int(v)
|
|
}
|
|
return nil
|
|
}
|
|
RegisterCPU(f)
|
|
}
|