renaming plugins -> inputs

plugins/inputs/system/cpu.go

@@ -0,0 +1,117 @@
+package system
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/influxdb/telegraf/plugins/inputs"
+	"github.com/shirou/gopsutil/cpu"
+)
+
+type CPUStats struct {
+	ps        PS
+	lastStats []cpu.CPUTimesStat
+
+	PerCPU   bool `toml:"percpu"`
+	TotalCPU bool `toml:"totalcpu"`
+}
+
+func NewCPUStats(ps PS) *CPUStats {
+	return &CPUStats{
+		ps: ps,
+	}
+}
+
+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
+  # Comment this line if you want the raw CPU time metrics
+  drop = ["time_*"]
+`
+
+func (_ *CPUStats) SampleConfig() string {
+	return sampleConfig
+}
+
+func (s *CPUStats) Gather(acc inputs.Accumulator) error {
+	times, err := s.ps.CPUTimes(s.PerCPU, s.TotalCPU)
+	if err != nil {
+		return fmt.Errorf("error getting CPU info: %s", err)
+	}
+	now := time.Now()
+
+	for i, cts := range times {
+		tags := map[string]string{
+			"cpu": cts.CPU,
+		}
+
+		total := totalCpuTime(cts)
+
+		// Add cpu time metrics
+		fields := 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,
+		}
+
+		// Add in percentage
+		if len(s.lastStats) == 0 {
+			acc.AddFields("cpu", fields, tags, now)
+			// If it's the 1st gather, can't get CPU Usage stats yet
+			continue
+		}
+		lastCts := s.lastStats[i]
+		lastTotal := totalCpuTime(lastCts)
+		totalDelta := total - lastTotal
+
+		if totalDelta < 0 {
+			s.lastStats = times
+			return fmt.Errorf("Error: current total CPU time is less than previous total CPU time")
+		}
+
+		if totalDelta == 0 {
+			continue
+		}
+
+		fields["usage_user"] = 100 * (cts.User - lastCts.User) / totalDelta
+		fields["usage_system"] = 100 * (cts.System - lastCts.System) / totalDelta
+		fields["usage_idle"] = 100 * (cts.Idle - lastCts.Idle) / totalDelta
+		fields["usage_nice"] = 100 * (cts.Nice - lastCts.Nice) / totalDelta
+		fields["usage_iowait"] = 100 * (cts.Iowait - lastCts.Iowait) / totalDelta
+		fields["usage_irq"] = 100 * (cts.Irq - lastCts.Irq) / totalDelta
+		fields["usage_softirq"] = 100 * (cts.Softirq - lastCts.Softirq) / totalDelta
+		fields["usage_steal"] = 100 * (cts.Steal - lastCts.Steal) / totalDelta
+		fields["usage_guest"] = 100 * (cts.Guest - lastCts.Guest) / totalDelta
+		fields["usage_guest_nice"] = 100 * (cts.GuestNice - lastCts.GuestNice) / totalDelta
+		acc.AddFields("cpu", fields, tags, now)
+	}
+
+	s.lastStats = times
+
+	return nil
+}
+
+func totalCpuTime(t cpu.CPUTimesStat) float64 {
+	total := t.User + t.System + t.Nice + t.Iowait + t.Irq + t.Softirq + t.Steal +
+		t.Guest + t.GuestNice + t.Idle
+	return total
+}
+
+func init() {
+	inputs.Add("cpu", func() inputs.Input {
+		return &CPUStats{ps: &systemPS{}}
+	})
+}