122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
package widgets
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/metrics"
|
|
"github.com/VividCortex/ewma"
|
|
"github.com/xxxserxxx/gotop/v4/devices"
|
|
|
|
"github.com/gizak/termui/v3"
|
|
ui "github.com/xxxserxxx/gotop/v4/termui"
|
|
)
|
|
|
|
// TODO Maybe group CPUs in columns if space permits
|
|
type CPUWidget struct {
|
|
*ui.LineGraph
|
|
CPUCount int
|
|
ShowAverageLoad bool
|
|
ShowPerCPULoad bool
|
|
updateInterval time.Duration
|
|
cpuLoads map[string]float64
|
|
average ewma.MovingAverage
|
|
}
|
|
|
|
var cpuLabels []string
|
|
|
|
func NewCPUWidget(updateInterval time.Duration, horizontalScale int, showAverageLoad bool, showPerCPULoad bool) *CPUWidget {
|
|
self := &CPUWidget{
|
|
LineGraph: ui.NewLineGraph(),
|
|
CPUCount: len(cpuLabels),
|
|
updateInterval: updateInterval,
|
|
ShowAverageLoad: showAverageLoad,
|
|
ShowPerCPULoad: showPerCPULoad,
|
|
cpuLoads: make(map[string]float64),
|
|
average: ewma.NewMovingAverage(),
|
|
}
|
|
self.LabelStyles[AVRG] = termui.ModifierBold
|
|
self.Title = tr.Value("widget.label.cpu")
|
|
self.HorizontalScale = horizontalScale
|
|
|
|
if !(self.ShowAverageLoad || self.ShowPerCPULoad) {
|
|
if self.CPUCount <= 8 {
|
|
self.ShowPerCPULoad = true
|
|
} else {
|
|
self.ShowAverageLoad = true
|
|
}
|
|
}
|
|
|
|
if self.ShowAverageLoad {
|
|
self.Data[AVRG] = []float64{0}
|
|
}
|
|
|
|
if self.ShowPerCPULoad {
|
|
cpus := make(map[string]int)
|
|
devices.UpdateCPU(cpus, self.updateInterval, self.ShowPerCPULoad)
|
|
for k, v := range cpus {
|
|
self.Data[k] = []float64{float64(v)}
|
|
}
|
|
}
|
|
|
|
self.update()
|
|
|
|
go func() {
|
|
for range time.NewTicker(self.updateInterval).C {
|
|
self.update()
|
|
}
|
|
}()
|
|
|
|
return self
|
|
}
|
|
|
|
const AVRG = "AVRG"
|
|
|
|
func (cpu *CPUWidget) EnableMetric() {
|
|
if cpu.ShowAverageLoad {
|
|
metrics.NewGauge(makeName("cpu", " avg"), func() float64 {
|
|
return cpu.cpuLoads[AVRG]
|
|
})
|
|
} else {
|
|
cpus := make(map[string]int)
|
|
devices.UpdateCPU(cpus, cpu.updateInterval, cpu.ShowPerCPULoad)
|
|
for key, perc := range cpus {
|
|
kc := key
|
|
cpu.cpuLoads[key] = float64(perc)
|
|
metrics.NewGauge(makeName("cpu", key), func() float64 {
|
|
return cpu.cpuLoads[kc]
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func (cpu *CPUWidget) Scale(i int) {
|
|
cpu.LineGraph.HorizontalScale = i
|
|
}
|
|
|
|
func (cpu *CPUWidget) update() {
|
|
go func() {
|
|
cpus := make(map[string]int)
|
|
devices.UpdateCPU(cpus, cpu.updateInterval, true)
|
|
cpu.Lock()
|
|
defer cpu.Unlock()
|
|
// AVG = ((AVG*i)+n)/(i+1)
|
|
var sum int
|
|
for key, percent := range cpus {
|
|
sum += percent
|
|
if cpu.ShowPerCPULoad {
|
|
cpu.Data[key] = append(cpu.Data[key], float64(percent))
|
|
cpu.Labels[key] = fmt.Sprintf("%3d%%", percent)
|
|
cpu.cpuLoads[key] = float64(percent)
|
|
}
|
|
}
|
|
if cpu.ShowAverageLoad {
|
|
cpu.average.Add(float64(sum) / float64(len(cpus)))
|
|
avg := cpu.average.Value()
|
|
cpu.Data[AVRG] = append(cpu.Data[AVRG], avg)
|
|
cpu.Labels[AVRG] = fmt.Sprintf("%3.0f%%", avg)
|
|
cpu.cpuLoads[AVRG] = avg
|
|
}
|
|
}()
|
|
}
|