chore: simplify linux nocgroup logic (#3949)

This commit is contained in:
Kevin Wan
2024-03-02 00:23:52 +08:00
committed by GitHub
parent a7363f0c21
commit 5263805b3b
5 changed files with 13 additions and 8 deletions

View File

@@ -1,5 +1,3 @@
//go:build !nocgroup
package internal
import (
@@ -25,6 +23,7 @@ var (
preTotal uint64
limit float64
cores uint64
noCgroup bool
initOnce sync.Once
)
@@ -32,6 +31,7 @@ var (
func initialize() {
cpus, err := effectiveCpus()
if err != nil {
noCgroup = true
logx.Error(err)
return
}
@@ -47,12 +47,14 @@ func initialize() {
preSystem, err = systemCpuUsage()
if err != nil {
noCgroup = true
logx.Error(err)
return
}
preTotal, err = cpuUsage()
if err != nil {
noCgroup = true
logx.Error(err)
return
}
@@ -62,6 +64,10 @@ func initialize() {
func RefreshCpu() uint64 {
initOnce.Do(initialize)
if noCgroup {
return 0
}
total, err := cpuUsage()
if err != nil {
return 0