mirror of
https://github.com/datarhei/core.git
synced 2025-10-09 09:50:15 +08:00
69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package monitor
|
|
|
|
import (
|
|
"github.com/datarhei/core/monitor/metric"
|
|
"github.com/datarhei/core/psutil"
|
|
)
|
|
|
|
type cpuCollector struct {
|
|
ncpuDescr *metric.Description
|
|
systemDescr *metric.Description
|
|
userDescr *metric.Description
|
|
idleDescr *metric.Description
|
|
otherDescr *metric.Description
|
|
|
|
ncpu float64
|
|
}
|
|
|
|
func NewCPUCollector() metric.Collector {
|
|
c := &cpuCollector{
|
|
ncpu: 1,
|
|
}
|
|
|
|
c.ncpuDescr = metric.NewDesc("cpu_ncpu", "", nil)
|
|
c.systemDescr = metric.NewDesc("cpu_system", "", nil)
|
|
c.userDescr = metric.NewDesc("cpu_user", "", nil)
|
|
c.idleDescr = metric.NewDesc("cpu_idle", "", nil)
|
|
c.otherDescr = metric.NewDesc("cpu_other", "", nil)
|
|
|
|
if ncpu, err := psutil.CPUCounts(true); err == nil {
|
|
c.ncpu = ncpu
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
func (c *cpuCollector) Stop() {}
|
|
|
|
func (c *cpuCollector) Prefix() string {
|
|
return "cpu"
|
|
}
|
|
|
|
func (c *cpuCollector) Describe() []*metric.Description {
|
|
return []*metric.Description{
|
|
c.ncpuDescr,
|
|
c.systemDescr,
|
|
c.userDescr,
|
|
c.idleDescr,
|
|
c.otherDescr,
|
|
}
|
|
}
|
|
|
|
func (c *cpuCollector) Collect() metric.Metrics {
|
|
metrics := metric.NewMetrics()
|
|
|
|
metrics.Add(metric.NewValue(c.ncpuDescr, c.ncpu))
|
|
|
|
stat, err := psutil.CPUPercent()
|
|
if err != nil {
|
|
return metrics
|
|
}
|
|
|
|
metrics.Add(metric.NewValue(c.systemDescr, stat.System))
|
|
metrics.Add(metric.NewValue(c.userDescr, stat.User))
|
|
metrics.Add(metric.NewValue(c.idleDescr, stat.Idle))
|
|
metrics.Add(metric.NewValue(c.otherDescr, stat.Other))
|
|
|
|
return metrics
|
|
}
|