telegraf/plugins/inputs/system/cpu.go

153 lines
3.8 KiB
Go
Raw Normal View History

2015-05-18 23:01:42 +00:00
package system
import (
"fmt"
2015-12-11 20:07:32 +00:00
"time"
2015-05-18 23:01:42 +00:00
"github.com/influxdata/telegraf"
2016-01-20 18:57:35 +00:00
"github.com/influxdata/telegraf/plugins/inputs"
2015-09-17 19:07:15 +00:00
"github.com/shirou/gopsutil/cpu"
2015-05-18 23:01:42 +00:00
)
type CPUStats struct {
2015-08-12 18:09:20 +00:00
ps PS
lastStats map[string]cpu.TimesStat
PerCPU bool `toml:"percpu"`
TotalCPU bool `toml:"totalcpu"`
CollectCPUTime bool `toml:"collect_cpu_time"`
ReportActive bool `toml:"report_active"`
2015-08-12 18:09:20 +00:00
}
func NewCPUStats(ps PS) *CPUStats {
return &CPUStats{
ps: ps,
CollectCPUTime: true,
ReportActive: true,
2015-08-12 18:09:20 +00:00
}
2015-05-18 23:01:42 +00:00
}
func (_ *CPUStats) Description() string {
return "Read metrics about cpu usage"
}
var sampleConfig = `
## Whether to report per-cpu stats or not
percpu = true
## Whether to report total system cpu stats or not
totalcpu = true
## If true, collect raw CPU time metrics.
collect_cpu_time = false
## If true, compute and report the sum of all non-idle CPU states.
report_active = false
2015-08-26 15:21:39 +00:00
`
func (_ *CPUStats) SampleConfig() string {
return sampleConfig
}
2015-05-18 23:01:42 +00:00
func (s *CPUStats) Gather(acc telegraf.Accumulator) error {
times, err := s.ps.CPUTimes(s.PerCPU, s.TotalCPU)
2015-05-18 23:01:42 +00:00
if err != nil {
return fmt.Errorf("error getting CPU info: %s", err)
}
2015-12-11 20:07:32 +00:00
now := time.Now()
2015-05-18 23:01:42 +00:00
for _, cts := range times {
2015-05-18 23:01:42 +00:00
tags := map[string]string{
"cpu": cts.CPU,
}
total := totalCpuTime(cts)
active := activeCpuTime(cts)
2015-08-12 18:09:20 +00:00
if s.CollectCPUTime {
// Add cpu time metrics
fieldsC := map[string]interface{}{
"time_user": cts.User,
"time_system": cts.System,
"time_idle": cts.Idle,
"time_nice": cts.Nice,
"time_iowait": cts.Iowait,
"time_irq": cts.Irq,
"time_softirq": cts.Softirq,
"time_steal": cts.Steal,
"time_guest": cts.Guest,
"time_guest_nice": cts.GuestNice,
}
if s.ReportActive {
fieldsC["time_active"] = activeCpuTime(cts)
}
acc.AddCounter("cpu", fieldsC, tags, now)
2015-12-11 20:07:32 +00:00
}
2015-08-12 18:09:20 +00:00
// Add in percentage
if len(s.lastStats) == 0 {
2015-12-11 20:07:32 +00:00
// If it's the 1st gather, can't get CPU Usage stats yet
continue
}
lastCts, ok := s.lastStats[cts.CPU]
if !ok {
continue
}
lastTotal := totalCpuTime(lastCts)
lastActive := activeCpuTime(lastCts)
2015-08-12 18:09:20 +00:00
totalDelta := total - lastTotal
if totalDelta < 0 {
err = fmt.Errorf("Error: current total CPU time is less than previous total CPU time")
break
2015-08-12 18:09:20 +00:00
}
if totalDelta == 0 {
continue
2015-08-12 18:09:20 +00:00
}
fieldsG := map[string]interface{}{
"usage_user": 100 * (cts.User - lastCts.User - (cts.Guest - lastCts.Guest)) / totalDelta,
"usage_system": 100 * (cts.System - lastCts.System) / totalDelta,
"usage_idle": 100 * (cts.Idle - lastCts.Idle) / totalDelta,
"usage_nice": 100 * (cts.Nice - lastCts.Nice - (cts.GuestNice - lastCts.GuestNice)) / totalDelta,
"usage_iowait": 100 * (cts.Iowait - lastCts.Iowait) / totalDelta,
"usage_irq": 100 * (cts.Irq - lastCts.Irq) / totalDelta,
"usage_softirq": 100 * (cts.Softirq - lastCts.Softirq) / totalDelta,
"usage_steal": 100 * (cts.Steal - lastCts.Steal) / totalDelta,
"usage_guest": 100 * (cts.Guest - lastCts.Guest) / totalDelta,
"usage_guest_nice": 100 * (cts.GuestNice - lastCts.GuestNice) / totalDelta,
}
if s.ReportActive {
fieldsG["usage_active"] = 100 * (active - lastActive) / totalDelta
}
acc.AddGauge("cpu", fieldsG, tags, now)
2015-05-18 23:01:42 +00:00
}
s.lastStats = make(map[string]cpu.TimesStat)
for _, cts := range times {
s.lastStats[cts.CPU] = cts
}
2015-08-12 18:09:20 +00:00
return err
2015-05-18 23:01:42 +00:00
}
func totalCpuTime(t cpu.TimesStat) float64 {
total := t.User + t.System + t.Nice + t.Iowait + t.Irq + t.Softirq + t.Steal +
t.Idle
return total
2015-08-12 18:09:20 +00:00
}
func activeCpuTime(t cpu.TimesStat) float64 {
active := totalCpuTime(t) - t.Idle
return active
}
2015-05-18 23:01:42 +00:00
func init() {
inputs.Add("cpu", func() telegraf.Input {
return &CPUStats{
PerCPU: true,
TotalCPU: true,
2017-04-18 18:42:58 +00:00
ps: newSystemPS(),
}
2015-05-18 23:01:42 +00:00
})
}