Add Cpustat collector for /proc/stat

collectors/cpustatMetric.go

@@ -0,0 +1,70 @@
+package collectors
+
+import (
+	"io/ioutil"
+	"strconv"
+	"strings"
+	"time"
+)
+
+const CPUSTATFILE = `/proc/stat`
+
+type CpustatCollector struct {
+	MetricCollector
+}
+
+func (m *CpustatCollector) Init() {
+	m.name = "CpustatCollector"
+	m.setup()
+}
+
+func ParseStatLine(line string, out map[string]interface{}) {
+    ls := strings.Fields(line)
+    user, _ := strconv.ParseInt(ls[1], 0, 64)
+    out["cpu_user"] = float64(user)
+    nice, _ := strconv.ParseInt(ls[2], 0, 64)
+    out["cpu_nice"] = float64(nice)
+    system, _ := strconv.ParseInt(ls[3], 0, 64)
+    out["cpu_system"] = float64(system)
+    idle, _ := strconv.ParseInt(ls[4], 0, 64)
+    out["cpu_idle"] = float64(idle)
+    iowait, _ := strconv.ParseInt(ls[5], 0, 64)
+    out["cpu_iowait"] = float64(iowait)
+    irq, _ := strconv.ParseInt(ls[6], 0, 64)
+    out["cpu_irq"] = float64(irq)
+    softirq, _ := strconv.ParseInt(ls[7], 0, 64)
+    out["cpu_softirq"] = float64(softirq)
+    steal, _ := strconv.ParseInt(ls[8], 0, 64)
+    out["cpu_steal"] = float64(steal)
+    guest, _ := strconv.ParseInt(ls[9], 0, 64)
+    out["cpu_guest"] = float64(guest)
+    guest_nice, _ := strconv.ParseInt(ls[10], 0, 64)
+    out["cpu_guest_nice"] = float64(guest_nice)
+}
+
+func (m *CpustatCollector) Read(interval time.Duration) {
+	buffer, err := ioutil.ReadFile(string(CPUSTATFILE))
+
+	if err != nil {
+		return
+	}
+
+    ll := strings.Split(string(buffer), "\n")
+    for _, line := range ll {
+        if (len(line) == 0) {
+            continue
+        }
+        ls := strings.Fields(line)
+        if (strings.Compare(ls[0], "cpu") == 0) {
+            ParseStatLine(line, m.node)
+        } else if (strings.HasPrefix(ls[0], "cpu")) {
+            cpustr := strings.TrimLeft(ls[0], "cpu")
+            cpu, _ := strconv.Atoi(cpustr)
+            ParseStatLine(line, m.cpus[cpu])
+        }
+    }
+}
+
+func (m *CpustatCollector) Close() {
+	return
+}

config.json

@@ -15,6 +15,7 @@
         "loadavg",
         "netstat",
         "ibstat",
-        "lustrestat"
+        "lustrestat",
+        "cpustat"
     ]
 }

metric-collector.go

@@ -23,6 +23,7 @@ var Collectors = map[string]collectors.MetricGetter{
 	"netstat":    &collectors.NetstatCollector{},
 	"ibstat":     &collectors.InfinibandCollector{},
 	"lustrestat": &collectors.LustreCollector{},
+	"cpustat": &collectors.CpustatCollector{},
 }
 
 var Sinks = map[string]sinks.SinkFuncs{