Breakup the system plugin

plugins/system/disk.go

@@ -0,0 +1,83 @@
+package system
+
+import (
+	"fmt"
+
+	"github.com/influxdb/tivan/plugins"
+)
+
+type DiskStats struct {
+	ps PS
+}
+
+func (_ *DiskStats) Description() string {
+	return "Read metrics about disk usage by mount point"
+}
+
+func (_ *DiskStats) SampleConfig() string { return "" }
+
+func (s *DiskStats) Gather(acc plugins.Accumulator) error {
+	disks, err := s.ps.DiskUsage()
+	if err != nil {
+		return fmt.Errorf("error getting disk usage info: %s", err)
+	}
+
+	for _, du := range disks {
+		tags := map[string]string{
+			"path": du.Path,
+		}
+
+		acc.Add("total", du.Total, tags)
+		acc.Add("free", du.Free, tags)
+		acc.Add("used", du.Total-du.Free, tags)
+		acc.Add("inodes_total", du.InodesTotal, tags)
+		acc.Add("inodes_free", du.InodesFree, tags)
+		acc.Add("inodes_used", du.InodesTotal-du.InodesFree, tags)
+	}
+
+	return nil
+}
+
+type DiskIOStats struct {
+	ps PS
+}
+
+func (_ *DiskIOStats) Description() string {
+	return "Read metrics about disk IO by device"
+}
+
+func (_ *DiskIOStats) SampleConfig() string { return "" }
+
+func (s *DiskIOStats) Gather(acc plugins.Accumulator) error {
+	diskio, err := s.ps.DiskIO()
+	if err != nil {
+		return fmt.Errorf("error getting disk io info: %s", err)
+	}
+
+	for _, io := range diskio {
+		tags := map[string]string{
+			"name":   io.Name,
+			"serial": io.SerialNumber,
+		}
+
+		acc.Add("reads", io.ReadCount, tags)
+		acc.Add("writes", io.WriteCount, tags)
+		acc.Add("read_bytes", io.ReadBytes, tags)
+		acc.Add("write_bytes", io.WriteBytes, tags)
+		acc.Add("read_time", io.ReadTime, tags)
+		acc.Add("write_time", io.WriteTime, tags)
+		acc.Add("io_time", io.IoTime, tags)
+	}
+
+	return nil
+}
+
+func init() {
+	plugins.Add("disk", func() plugins.Plugin {
+		return &DiskStats{ps: &systemPS{}}
+	})
+
+	plugins.Add("io", func() plugins.Plugin {
+		return &DiskIOStats{ps: &systemPS{}}
+	})
+}