50 lines
1.1 KiB
Go
50 lines
1.1 KiB
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
|
|
}
|