You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
go-zero/core/stat/internal/cpu_linux.go

162 lines
2.6 KiB
Go

4 years ago
package internal
import (
"errors"
"fmt"
"strings"
"sync"
4 years ago
"time"
"github.com/zeromicro/go-zero/core/iox"
"github.com/zeromicro/go-zero/core/logx"
4 years ago
)
const (
cpuTicks = 100
cpuFields = 8
)
var (
preSystem uint64
preTotal uint64
quota float64
cores uint64
)
var initonce sync.Once
// if /proc not present, ignore the cpu calculation, like wsl linux
func initialize() {
cpus, err := cpuSets()
if err != nil {
logx.Error(err)
return
}
4 years ago
cores = uint64(len(cpus))
4 years ago
sets, err := cpuSets()
if err != nil {
logx.Error(err)
return
}
4 years ago
quota = float64(len(sets))
cq, err := cpuQuota()
if err == nil {
if cq != -1 {
period, err := cpuPeriod()
if err != nil {
logx.Error(err)
return
}
4 years ago
limit := float64(cq) / float64(period)
if limit < quota {
quota = limit
}
}
}
preSystem, err = systemCpuUsage()
if err != nil {
logx.Error(err)
return
}
4 years ago
preTotal, err = totalCpuUsage()
if err != nil {
logx.Error(err)
return
}
4 years ago
}
// RefreshCpu refreshes cpu usage and returns.
4 years ago
func RefreshCpu() uint64 {
initonce.Do(initialize)
4 years ago
total, err := totalCpuUsage()
if err != nil {
return 0
}
system, err := systemCpuUsage()
if err != nil {
return 0
}
var usage uint64
cpuDelta := total - preTotal
systemDelta := system - preSystem
if cpuDelta > 0 && systemDelta > 0 {
usage = uint64(float64(cpuDelta*cores*1e3) / (float64(systemDelta) * quota))
}
preSystem = system
preTotal = total
return usage
}
func cpuQuota() (int64, error) {
cg, err := currentCgroup()
if err != nil {
return 0, err
}
return cg.cpuQuotaUs()
}
func cpuPeriod() (uint64, error) {
cg, err := currentCgroup()
if err != nil {
return 0, err
}
return cg.cpuPeriodUs()
}
func cpuSets() ([]uint64, error) {
cg, err := currentCgroup()
if err != nil {
return nil, err
}
return cg.cpus()
}
func systemCpuUsage() (uint64, error) {
lines, err := iox.ReadTextLines("/proc/stat", iox.WithoutBlank())
if err != nil {
return 0, err
}
for _, line := range lines {
fields := strings.Fields(line)
if fields[0] == "cpu" {
if len(fields) < cpuFields {
return 0, fmt.Errorf("bad format of cpu stats")
}
var totalClockTicks uint64
for _, i := range fields[1:cpuFields] {
v, err := parseUint(i)
if err != nil {
return 0, err
}
totalClockTicks += v
}
return (totalClockTicks * uint64(time.Second)) / cpuTicks, nil
}
}
return 0, errors.New("bad stats format")
}
func totalCpuUsage() (usage uint64, err error) {
var cg cgroup
4 years ago
if cg, err = currentCgroup(); err != nil {
return
}
return cg.usageAllCpus()
4 years ago
}