github.com/anth0d/nomad@v0.0.0-20221214183521-ae3a0a2cad06/drivers/docker/util/stats_posix.go (about)

     1  //go:build !windows
     2  
     3  package util
     4  
     5  import (
     6  	"runtime"
     7  
     8  	docker "github.com/fsouza/go-dockerclient"
     9  	cstructs "github.com/hashicorp/nomad/client/structs"
    10  	"github.com/hashicorp/nomad/helper/stats"
    11  )
    12  
    13  var (
    14  	DockerMeasuredCPUStats = []string{"Throttled Periods", "Throttled Time", "Percent"}
    15  
    16  	// cgroup-v2 only exposes a subset of memory stats
    17  	DockerCgroupV1MeasuredMemStats = []string{"RSS", "Cache", "Swap", "Usage", "Max Usage"}
    18  	DockerCgroupV2MeasuredMemStats = []string{"Cache", "Swap", "Usage"}
    19  )
    20  
    21  func DockerStatsToTaskResourceUsage(s *docker.Stats) *cstructs.TaskResourceUsage {
    22  	measuredMems := DockerCgroupV1MeasuredMemStats
    23  
    24  	// use a simple heuristic to check if cgroup-v2 is used.
    25  	// go-dockerclient doesn't distinguish between 0 and not-present value
    26  	if s.MemoryStats.Stats.Rss == 0 && s.MemoryStats.MaxUsage == 0 && s.MemoryStats.Usage != 0 {
    27  		measuredMems = DockerCgroupV2MeasuredMemStats
    28  	}
    29  
    30  	ms := &cstructs.MemoryStats{
    31  		RSS:        s.MemoryStats.Stats.Rss,
    32  		Cache:      s.MemoryStats.Stats.Cache,
    33  		Swap:       s.MemoryStats.Stats.Swap,
    34  		MappedFile: s.MemoryStats.Stats.MappedFile,
    35  		Usage:      s.MemoryStats.Usage,
    36  		MaxUsage:   s.MemoryStats.MaxUsage,
    37  		Measured:   measuredMems,
    38  	}
    39  
    40  	cs := &cstructs.CpuStats{
    41  		ThrottledPeriods: s.CPUStats.ThrottlingData.ThrottledPeriods,
    42  		ThrottledTime:    s.CPUStats.ThrottlingData.ThrottledTime,
    43  		Measured:         DockerMeasuredCPUStats,
    44  	}
    45  
    46  	// Calculate percentage
    47  	cs.Percent = CalculateCPUPercent(
    48  		s.CPUStats.CPUUsage.TotalUsage, s.PreCPUStats.CPUUsage.TotalUsage,
    49  		s.CPUStats.SystemCPUUsage, s.PreCPUStats.SystemCPUUsage, runtime.NumCPU())
    50  	cs.SystemMode = CalculateCPUPercent(
    51  		s.CPUStats.CPUUsage.UsageInKernelmode, s.PreCPUStats.CPUUsage.UsageInKernelmode,
    52  		s.CPUStats.CPUUsage.TotalUsage, s.PreCPUStats.CPUUsage.TotalUsage, runtime.NumCPU())
    53  	cs.UserMode = CalculateCPUPercent(
    54  		s.CPUStats.CPUUsage.UsageInUsermode, s.PreCPUStats.CPUUsage.UsageInUsermode,
    55  		s.CPUStats.CPUUsage.TotalUsage, s.PreCPUStats.CPUUsage.TotalUsage, runtime.NumCPU())
    56  	cs.TotalTicks = (cs.Percent / 100) * stats.TotalTicksAvailable() / float64(runtime.NumCPU())
    57  
    58  	return &cstructs.TaskResourceUsage{
    59  		ResourceUsage: &cstructs.ResourceUsage{
    60  			MemoryStats: ms,
    61  			CpuStats:    cs,
    62  		},
    63  		Timestamp: s.Read.UTC().UnixNano(),
    64  	}
    65  }