mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-11-21 14:44:00 +00:00
ef7e2af8f5
Update GOMAXPROCS to limits set via cgroups. This should reduce CPU trashing and reduce memory usage for cases when VictoriaMetrics components run in containers with CPU limits. See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/685
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package cgroup
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/logger"
|
|
)
|
|
|
|
// UpdateGOMAXPROCSToCPUQuota updates GOMAXPROCS to cgroup CPU quota if GOMAXPROCS isn't set in environment var.
|
|
//
|
|
// This function must be called after logger.Init().
|
|
func UpdateGOMAXPROCSToCPUQuota() {
|
|
if v := os.Getenv("GOMAXPROCS"); v != "" {
|
|
return
|
|
}
|
|
q := getCPUQuota()
|
|
if q <= 0 {
|
|
// Do not change GOMAXPROCS
|
|
return
|
|
}
|
|
gomaxprocs := int(q + 0.5)
|
|
if gomaxprocs <= 0 {
|
|
gomaxprocs = 1
|
|
}
|
|
logger.Infof("updating GOMAXPROCS to %d according to cgroup CPU quota", gomaxprocs)
|
|
runtime.GOMAXPROCS(gomaxprocs)
|
|
}
|
|
|
|
func getCPUQuota() float64 {
|
|
quotaUS, err := readInt64("/sys/fs/cgroup/cpu/cpu.cfs_quota_us", "cat /sys/fs/cgroup/cpu$(cat /proc/self/cgroup | grep cpu, | cut -d: -f3)/cpu.cfs_quota_us")
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
periodUS, err := readInt64("/sys/fs/cgroup/cpu/cpu.cfs_period_us", "cat /sys/fs/cgroup/cpu$(cat /proc/self/cgroup | grep cpu, | cut -d: -f3)/cpu.cfs_period_us")
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
return float64(quotaUS) / float64(periodUS)
|
|
}
|