open-nomad/client/stats/cpu.go

package stats

import (
	"runtime"
	"time"
)

// CpuStats calculates cpu usage percentage
type CpuStats struct {
	prevProcessUsage float64
	prevTime         time.Time

	totalCpus int
}

// NewCpuStats returns a cpu stats calculator
func NewCpuStats() *CpuStats {
	numCpus := runtime.NumCPU()
	return &CpuStats{totalCpus: numCpus}
}

// Percent calculates the cpu usage percentage based on the current cpu usage
// and the previous cpu usage
func (c *CpuStats) Percent(currentProcessUsage float64) float64 {
	now := time.Now()

	if c.prevProcessUsage == 0.0 {
		// invoked first time
		c.prevProcessUsage = currentProcessUsage
		c.prevTime = now
		return 0.0
	}

	delta := (now.Sub(c.prevTime).Seconds()) * float64(c.totalCpus)
	ret := c.calculatePercent(c.prevProcessUsage, currentProcessUsage, delta)
	c.prevProcessUsage = currentProcessUsage
	c.prevTime = now
	return ret

}

func (c *CpuStats) calculatePercent(t1, t2 float64, delta float64) float64 {
	if delta == 0 {
		return 0
	}
	delta_proc := t2 - t1
	overall_percent := ((delta_proc / delta) * 100) * float64(c.totalCpus)
	return overall_percent
}