cc-metric-collector/collectors/loadavgMetric.go

64 lines
1.3 KiB
Go
Raw Normal View History

package collectors
import (
2021-10-04 15:23:43 +02:00
lp "github.com/influxdata/line-protocol"
"io/ioutil"
"strconv"
"strings"
2021-03-25 17:47:08 +01:00
"time"
)
const LOADAVGFILE = `/proc/loadavg`
type LoadavgCollector struct {
MetricCollector
2021-10-04 15:23:43 +02:00
tags map[string]string
load_matches []string
proc_matches []string
}
func (m *LoadavgCollector) Init() error {
2021-03-25 17:47:08 +01:00
m.name = "LoadavgCollector"
m.setup()
2021-10-04 15:23:43 +02:00
m.tags = map[string]string{"type": "node"}
m.load_matches = []string{"load_one", "load_five", "load_fifteen"}
m.proc_matches = []string{"proc_run", "proc_total"}
m.init = true
return nil
}
2021-10-04 15:23:43 +02:00
func (m *LoadavgCollector) Read(interval time.Duration, out *[]lp.MutableMetric) {
buffer, err := ioutil.ReadFile(string(LOADAVGFILE))
if err != nil {
return
}
ls := strings.Split(string(buffer), ` `)
2021-10-04 15:23:43 +02:00
for i, name := range m.load_matches {
x, err := strconv.ParseFloat(ls[i], 64)
if err == nil {
y, err := lp.New(name, m.tags, map[string]interface{}{"value": float64(x)}, time.Now())
if err == nil {
*out = append(*out, y)
}
}
}
lv := strings.Split(ls[3], `/`)
2021-10-04 15:23:43 +02:00
for i, name := range m.proc_matches {
x, err := strconv.ParseFloat(lv[i], 64)
if err == nil {
y, err := lp.New(name, m.tags, map[string]interface{}{"value": float64(x)}, time.Now())
if err == nil {
*out = append(*out, y)
}
}
}
}
func (m *LoadavgCollector) Close() {
2021-10-04 15:47:03 +02:00
m.init = false
2021-03-25 17:47:08 +01:00
return
}