diff --git a/collectors/loadavgMetric.go b/collectors/loadavgMetric.go new file mode 100644 index 0000000..63f7c78 --- /dev/null +++ b/collectors/loadavgMetric.go @@ -0,0 +1,36 @@ +package collectors + +import ( + "io/ioutil" + "strconv" + "time" + "strings" +) + +const LOADAVGFILE = `/proc/loadavg` + +type LoadavgCollector struct { + MetricCollector +} + + +func (m *LoadavgCollector) Init() { + m.name = "LoadavgCollector" + m.setup() +} + +func (m *LoadavgCollector) Read(interval time.Duration){ + buffer, err := ioutil.ReadFile(string(LOADAVGFILE)) + + if err != nil { + return + } + + ls := strings.Split(string(buffer), ` `) + loadOne, _ := strconv.ParseFloat(ls[0], 64) + m.node["load_one"] = float64(loadOne) +} + +func (m *LoadavgCollector) Close() { + return +} diff --git a/collectors/memstatMetric.go b/collectors/memstatMetric.go new file mode 100644 index 0000000..43cc885 --- /dev/null +++ b/collectors/memstatMetric.go @@ -0,0 +1,57 @@ +package collectors + +import ( + "errors" + "io/ioutil" + "log" + "strconv" + "strings" + "time" + + //protocol "github.com/influxdata/line-protocol" +) + +const MEMSTATFILE = `/proc/meminfo` + +type MemstatCollector struct { + MetricCollector +} + + +func (m *MemstatCollector) Init() { + m.name = "MemstatCollector" + m.setup() +} + +func (m *MemstatCollector) Read(interval time.Duration){ + buffer, err := ioutil.ReadFile(string(MEMSTATFILE)) + + if err != nil { + log.Print(err) + return + } + + ll := strings.Split(string(buffer), "\n") + memstats := make(map[string]int64) + + for _, line := range ll { + ls := strings.Split(line, `:`) + if len(ls) > 1 { + lv := strings.Fields(ls[1]) + memstats[ls[0]], err = strconv.ParseInt(lv[0], 0, 64) + } + } + + if _, exists := memstats[`MemTotal`]; !exists { + err = errors.New("Parse error") + log.Print(err) + return + } + + memUsed := memstats[`MemTotal`] - (memstats[`MemFree`] + memstats[`Buffers`] + memstats[`Cached`]) + m.node["mem_used"] = float64(memUsed) * 1.0e-3 +} + +func (m *MemstatCollector) Close() { + return +} diff --git a/collectors/netstatMetric.go b/collectors/netstatMetric.go new file mode 100644 index 0000000..e25d578 --- /dev/null +++ b/collectors/netstatMetric.go @@ -0,0 +1,59 @@ +package collectors + +import ( + "io/ioutil" + "strconv" + "strings" + "time" + "log" +) + +const NETSTATFILE = `/proc/net/dev` + +type NetstatCollector struct { + MetricCollector +} + + +func (m *NetstatCollector) Init() { + m.name = "NetstatCollector" + m.setup() +} + +func (m *NetstatCollector) Read(interval time.Duration){ + data, err := ioutil.ReadFile(string(NETSTATFILE)) + if err != nil { + log.Print(err.Error()) + return + } + var matches = map[int]string{ + 1: "bytes_in", + 9: "bytes_out", + 2: "pkts_in", + 10: "pkts_out", + } + + lines := strings.Split(string(data), "\n") + for _, l := range lines { + if !strings.Contains(l, ":") { + continue + } + f := strings.Fields(l) + dev := f[0][0 : len(f[0])-1] + if dev == "lo" { + continue + } + for i, name := range matches { + v, err := strconv.ParseInt(f[i], 10, 0) + if err == nil { + + m.node[name] = float64(v) * 1.0e-3 + } + } + } + +} + +func (m *NetstatCollector) Close() { + return +}