chore: simplify linux nocgroup logic (#3953)
This commit is contained in:
@@ -27,42 +27,9 @@ var (
|
|||||||
initOnce sync.Once
|
initOnce sync.Once
|
||||||
)
|
)
|
||||||
|
|
||||||
// if /proc not present, ignore the cpu calculation, like wsl linux
|
|
||||||
func initialize() {
|
|
||||||
cpus, err := effectiveCpus()
|
|
||||||
if err != nil {
|
|
||||||
noCgroup = true
|
|
||||||
logx.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
cores = uint64(cpus)
|
|
||||||
limit = float64(cpus)
|
|
||||||
quota, err := cpuQuota()
|
|
||||||
if err == nil && quota > 0 {
|
|
||||||
if quota < limit {
|
|
||||||
limit = quota
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
preSystem, err = systemCpuUsage()
|
|
||||||
if err != nil {
|
|
||||||
noCgroup = true
|
|
||||||
logx.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
preTotal, err = cpuUsage()
|
|
||||||
if err != nil {
|
|
||||||
noCgroup = true
|
|
||||||
logx.Error(err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// RefreshCpu refreshes cpu usage and returns.
|
// RefreshCpu refreshes cpu usage and returns.
|
||||||
func RefreshCpu() uint64 {
|
func RefreshCpu() uint64 {
|
||||||
initOnce.Do(initialize)
|
initializeOnce()
|
||||||
|
|
||||||
if noCgroup {
|
if noCgroup {
|
||||||
return 0
|
return 0
|
||||||
@@ -120,6 +87,47 @@ func effectiveCpus() (int, error) {
|
|||||||
return cg.effectiveCpus()
|
return cg.effectiveCpus()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// if /proc not present, ignore the cpu calculation, like wsl linux
|
||||||
|
func initialize() error {
|
||||||
|
cpus, err := effectiveCpus()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
cores = uint64(cpus)
|
||||||
|
limit = float64(cpus)
|
||||||
|
quota, err := cpuQuota()
|
||||||
|
if err == nil && quota > 0 {
|
||||||
|
if quota < limit {
|
||||||
|
limit = quota
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
preSystem, err = systemCpuUsage()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
preTotal, err = cpuUsage()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func initializeOnce() {
|
||||||
|
initOnce.Do(func() {
|
||||||
|
defer func() {
|
||||||
|
if p := recover(); p != nil {
|
||||||
|
noCgroup = true
|
||||||
|
logx.Error(p)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
if err := initialize(); err != nil {
|
||||||
|
noCgroup = true
|
||||||
|
logx.Error(err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func systemCpuUsage() (uint64, error) {
|
func systemCpuUsage() (uint64, error) {
|
||||||
lines, err := iox.ReadTextLines(statFile, iox.WithoutBlank())
|
lines, err := iox.ReadTextLines(statFile, iox.WithoutBlank())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
Reference in New Issue
Block a user