1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
//go:build linux
package resources
import (
"os"
"strconv"
"strings"
"github.com/lxc/incus/v6/shared/api"
)
// GetLoad returns the system load information.
func GetLoad() (*api.ResourcesLoad, error) {
loadAvgs, err := getLoadAvgs()
if err != nil {
return nil, err
}
processes, err := getProcessCount()
if err != nil {
return nil, err
}
loadAverage := api.ResourcesLoad{
Average1Min: loadAvgs[0],
Average5Min: loadAvgs[1],
Average10Min: loadAvgs[2],
Processes: processes,
}
return &loadAverage, nil
}
// getLoadAvgs returns the host's load averages from /proc/loadavg.
func getLoadAvgs() ([]float64, error) {
loadAvgs := make([]float64, 3)
loadAvgsBuf, err := os.ReadFile("/proc/loadavg")
if err != nil {
return nil, err
}
loadAvgFields := strings.Fields(string(loadAvgsBuf))
loadAvgs[0], err = strconv.ParseFloat(loadAvgFields[0], 64)
if err != nil {
return nil, err
}
loadAvgs[1], err = strconv.ParseFloat(loadAvgFields[1], 64)
if err != nil {
return nil, err
}
loadAvgs[2], err = strconv.ParseFloat(loadAvgFields[2], 64)
if err != nil {
return nil, err
}
return loadAvgs, nil
}
// getProcessCount returns the count of all processes on the system.
func getProcessCount() (int, error) {
dirEntries, err := os.ReadDir("/proc")
if err != nil {
return 0, err
}
total := 0
for _, entry := range dirEntries {
if entry.IsDir() {
_, err := strconv.Atoi(entry.Name())
if err != nil {
continue
} else {
total += 1
}
}
}
return total, nil
}
|