af41e4d843
* kill process tree using syscall on windows & cleanup (#80) * kill process tree using syscall on windows & cleanup * use job api * add error check for cmd.Start * modularize monitor, reduce init usage * replace slices with sort * update gopsutil & other dependencies
51 lines
908 B
Go
51 lines
908 B
Go
package cpu
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
psCpu "github.com/shirou/gopsutil/v4/cpu"
|
|
)
|
|
|
|
type CPUHostType string
|
|
|
|
const CPUHostKey CPUHostType = "cpu"
|
|
|
|
func GetHost(ctx context.Context) ([]string, error) {
|
|
ci, err := psCpu.InfoWithContext(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cpuModelCount := make(map[string]int)
|
|
for _, c := range ci {
|
|
cpuModelCount[c.ModelName]++
|
|
}
|
|
|
|
var cpuType string
|
|
if t, ok := ctx.Value(CPUHostKey).(string); ok {
|
|
cpuType = t
|
|
}
|
|
|
|
var ch []string
|
|
u := len(ci) > 1
|
|
for model, count := range cpuModelCount {
|
|
if u {
|
|
ch = append(ch, fmt.Sprintf("%s %d %s Core", model, count, cpuType))
|
|
} else {
|
|
ch = append(ch, fmt.Sprintf("%s %d %s Core", model, ci[0].Cores, cpuType))
|
|
}
|
|
}
|
|
|
|
return ch, nil
|
|
}
|
|
|
|
func GetState(ctx context.Context) ([]float64, error) {
|
|
cp, err := psCpu.PercentWithContext(ctx, 0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return cp, nil
|
|
}
|