mirror of
https://github.com/shirou/gopsutil.git
synced 2025-04-26 13:48:59 +08:00
cded1805c1
The `ProcsTotal` in the `MiscStat` structure was very inaccurate. It was reading a value which is the total number of kernel scheduling entities. This includes both processes and threads significantly overcounting. This instead uses an existing method already in common to count the number of processes via the /proc filesystem where any directory is a number. This should still be a single syscall to read that directory entry. This fixes #1606.
129 lines
2.5 KiB
Go
129 lines
2.5 KiB
Go
//go:build linux
|
|
// +build linux
|
|
|
|
package load
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"github.com/shirou/gopsutil/v3/internal/common"
|
|
)
|
|
|
|
func Avg() (*AvgStat, error) {
|
|
return AvgWithContext(context.Background())
|
|
}
|
|
|
|
func AvgWithContext(ctx context.Context) (*AvgStat, error) {
|
|
stat, err := fileAvgWithContext(ctx)
|
|
if err != nil {
|
|
stat, err = sysinfoAvgWithContext()
|
|
}
|
|
return stat, err
|
|
}
|
|
|
|
func sysinfoAvgWithContext() (*AvgStat, error) {
|
|
var info syscall.Sysinfo_t
|
|
err := syscall.Sysinfo(&info)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
const si_load_shift = 16
|
|
return &AvgStat{
|
|
Load1: float64(info.Loads[0]) / float64(1<<si_load_shift),
|
|
Load5: float64(info.Loads[1]) / float64(1<<si_load_shift),
|
|
Load15: float64(info.Loads[2]) / float64(1<<si_load_shift),
|
|
}, nil
|
|
}
|
|
|
|
func fileAvgWithContext(ctx context.Context) (*AvgStat, error) {
|
|
values, err := readLoadAvgFromFile(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
load1, err := strconv.ParseFloat(values[0], 64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
load5, err := strconv.ParseFloat(values[1], 64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
load15, err := strconv.ParseFloat(values[2], 64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ret := &AvgStat{
|
|
Load1: load1,
|
|
Load5: load5,
|
|
Load15: load15,
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
// Misc returns miscellaneous host-wide statistics.
|
|
// Note: the name should be changed near future.
|
|
func Misc() (*MiscStat, error) {
|
|
return MiscWithContext(context.Background())
|
|
}
|
|
|
|
func MiscWithContext(ctx context.Context) (*MiscStat, error) {
|
|
filename := common.HostProcWithContext(ctx, "stat")
|
|
out, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ret := &MiscStat{}
|
|
lines := strings.Split(string(out), "\n")
|
|
for _, line := range lines {
|
|
fields := strings.Fields(line)
|
|
if len(fields) != 2 {
|
|
continue
|
|
}
|
|
v, err := strconv.ParseInt(fields[1], 10, 64)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
switch fields[0] {
|
|
case "processes":
|
|
ret.ProcsCreated = int(v)
|
|
case "procs_running":
|
|
ret.ProcsRunning = int(v)
|
|
case "procs_blocked":
|
|
ret.ProcsBlocked = int(v)
|
|
case "ctxt":
|
|
ret.Ctxt = int(v)
|
|
default:
|
|
continue
|
|
}
|
|
|
|
}
|
|
|
|
procsTotal, err := common.NumProcsWithContext(ctx)
|
|
if err != nil {
|
|
return ret, err
|
|
}
|
|
ret.ProcsTotal = int(procsTotal)
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func readLoadAvgFromFile(ctx context.Context) ([]string, error) {
|
|
loadavgFilename := common.HostProcWithContext(ctx, "loadavg")
|
|
line, err := os.ReadFile(loadavgFilename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
values := strings.Fields(string(line))
|
|
return values, nil
|
|
}
|