cpu.go

package limit

import (
	"linkfog.com/public/lib/cgroup"
	"linkfog.com/public/lib/cgroup/fs"
	"linkfog.com/public/lib/cgroup/fs2"
	"linkfog.com/public/lib/l"
)

var (
	DefaultCPULimit = float64(2)
)

func GetSelfCPULimit(defaultLimit float64) float64 {
	return GetCPULimit(-1, defaultLimit)
}

func GetCPULimit(pid int, defaultLimit float64) float64 {
	cgroupDir, err := cgroup.GetContainerCPUCgroupPath(pid)
	if err != nil {
		l.Errorf("read self cgroup cpuacct path error:%v, use default value", err)
		return defaultLimit
	}

	if cgroup.IsCgroup2() {
		return fs2.GetCPULimit(cgroupDir, defaultLimit)
	} else {
		return fs.GetCPULimit(cgroupDir, defaultLimit)
	}
}