xmtop/widgets/cpu.go
2018-02-23 00:42:39 -08:00

46 lines
920 B
Go

package widgets
import (
"strconv"
"time"
ui "github.com/cjbassi/gotop/termui"
cpu "github.com/shirou/gopsutil/cpu"
)
type CPU struct {
*ui.LineGraph
count int // number of CPUs
interval time.Duration
}
func NewCPU() *CPU {
count, _ := cpu.Counts(false)
c := &CPU{ui.NewLineGraph(), count, time.Second}
c.Label = "CPU Usage"
for i := 0; i < c.count; i++ {
key := "CPU" + strconv.Itoa(i+1)
c.Data[key] = []float64{0}
}
go c.update()
ticker := time.NewTicker(c.interval)
go func() {
for range ticker.C {
c.update()
}
}()
return c
}
func (c *CPU) update() {
// psutil calculates the CPU usage over a 1 second interval, therefore it blocks for 1 second
// `true` makes it so psutil doesn't group CPU usage percentages
percent, _ := cpu.Percent(time.Second, true)
for i := 0; i < c.count; i++ {
key := "CPU" + strconv.Itoa(i+1)
c.Data[key] = append(c.Data[key], percent[i])
}
}