112 lines
3.4 KiB
Go
112 lines
3.4 KiB
Go
package lib
|
|
|
|
import (
|
|
"path/filepath"
|
|
"syscall"
|
|
"time"
|
|
|
|
"strings"
|
|
|
|
"github.com/kubernetes-incubator/cri-o/oci"
|
|
"github.com/opencontainers/runc/libcontainer"
|
|
)
|
|
|
|
// ContainerStats contains the statistics information for a running container
|
|
type ContainerStats struct {
|
|
Container string
|
|
CPU float64
|
|
cpuNano uint64
|
|
systemNano uint64
|
|
MemUsage uint64
|
|
MemLimit uint64
|
|
MemPerc float64
|
|
NetInput uint64
|
|
NetOutput uint64
|
|
BlockInput uint64
|
|
BlockOutput uint64
|
|
PIDs uint64
|
|
}
|
|
|
|
// GetContainerStats gets the running stats for a given container
|
|
func (c *ContainerServer) GetContainerStats(ctr *oci.Container, previousStats *ContainerStats) (*ContainerStats, error) {
|
|
previousCPU := previousStats.cpuNano
|
|
previousSystem := previousStats.systemNano
|
|
libcontainerStats, err := c.LibcontainerStats(ctr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cgroupStats := libcontainerStats.CgroupStats
|
|
stats := new(ContainerStats)
|
|
stats.Container = ctr.ID()
|
|
stats.CPU = calculateCPUPercent(libcontainerStats, previousCPU, previousSystem)
|
|
stats.MemUsage = cgroupStats.MemoryStats.Usage.Usage
|
|
stats.MemLimit = getMemLimit(cgroupStats.MemoryStats.Usage.Limit)
|
|
stats.MemPerc = float64(stats.MemUsage) / float64(stats.MemLimit)
|
|
stats.PIDs = cgroupStats.PidsStats.Current
|
|
stats.BlockInput, stats.BlockOutput = calculateBlockIO(libcontainerStats)
|
|
stats.NetInput, stats.NetOutput = getContainerNetIO(libcontainerStats)
|
|
|
|
return stats, nil
|
|
}
|
|
|
|
func loadFactory(root string) (libcontainer.Factory, error) {
|
|
abs, err := filepath.Abs(root)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cgroupManager := libcontainer.Cgroupfs
|
|
return libcontainer.New(abs, cgroupManager, libcontainer.CriuPath(""))
|
|
}
|
|
|
|
// getMemory limit returns the memory limit for a given cgroup
|
|
// If the configured memory limit is larger than the total memory on the sys, the
|
|
// physical system memory size is returned
|
|
func getMemLimit(cgroupLimit uint64) uint64 {
|
|
si := &syscall.Sysinfo_t{}
|
|
err := syscall.Sysinfo(si)
|
|
if err != nil {
|
|
return cgroupLimit
|
|
}
|
|
|
|
physicalLimit := uint64(si.Totalram)
|
|
if cgroupLimit > physicalLimit {
|
|
return physicalLimit
|
|
}
|
|
return cgroupLimit
|
|
}
|
|
|
|
// Returns the total number of bytes transmitted and received for the given container stats
|
|
func getContainerNetIO(stats *libcontainer.Stats) (received uint64, transmitted uint64) {
|
|
for _, iface := range stats.Interfaces {
|
|
received += iface.RxBytes
|
|
transmitted += iface.TxBytes
|
|
}
|
|
return
|
|
}
|
|
|
|
func calculateCPUPercent(stats *libcontainer.Stats, previousCPU, previousSystem uint64) float64 {
|
|
var (
|
|
cpuPercent = 0.0
|
|
cpuDelta = float64(stats.CgroupStats.CpuStats.CpuUsage.TotalUsage - previousCPU)
|
|
systemDelta = float64(uint64(time.Now().UnixNano()) - previousSystem)
|
|
)
|
|
if systemDelta > 0.0 && cpuDelta > 0.0 {
|
|
// gets a ratio of container cpu usage total, multiplies it by the number of cores (4 cores running
|
|
// at 100% utilization should be 400% utilization), and multiplies that by 100 to get a percentage
|
|
cpuPercent = (cpuDelta / systemDelta) * float64(len(stats.CgroupStats.CpuStats.CpuUsage.PercpuUsage)) * 100
|
|
}
|
|
return cpuPercent
|
|
}
|
|
|
|
func calculateBlockIO(stats *libcontainer.Stats) (read uint64, write uint64) {
|
|
for _, blkIOEntry := range stats.CgroupStats.BlkioStats.IoServiceBytesRecursive {
|
|
switch strings.ToLower(blkIOEntry.Op) {
|
|
case "read":
|
|
read += blkIOEntry.Value
|
|
case "write":
|
|
write += blkIOEntry.Value
|
|
}
|
|
}
|
|
return
|
|
}
|