package collectors import ( "errors" lp "github.com/influxdata/line-protocol" "log" "strconv" "strings" "time" "os" "os/exec" "encoding/json" ) const IPMITOOL_PATH = `/usr/bin/ipmitool` const IPMISENSORS_PATH = `/usr/sbin/ipmi-sensors` type IpmiCollectorConfig struct { ExcludeDevices []string `json:"exclude_devices"` IpmitoolPath string `json:"ipmitool_path"` IpmisensorsPath string `json:"ipmisensors_path"` } type IpmiCollector struct { MetricCollector tags map[string]string matches map[string]string config IpmiCollectorConfig } func (m *IpmiCollector) Init(config []byte) error { m.name = "IpmiCollector" m.setup() if len(config) > 0 { err := json.Unmarshal(config, &m.config) if err != nil { return err } } _, err1 := os.Stat(m.config.IpmitoolPath) _, err2 := os.Stat(m.config.IpmisensorsPath) if err1 != nil { m.config.IpmitoolPath = "" } if err2 != nil { m.config.IpmisensorsPath = "" } if err1 != nil && err2 != nil { return errors.New("No IPMI reader found") } m.init = true return nil } func ReadIpmiTool(cmd string, out *[]lp.MutableMetric) { command := exec.Command(cmd, "sensor") command.Wait() stdout, err := command.Output() if err != nil { log.Print(err) return } ll := strings.Split(string(stdout), "\n") for _, line := range ll { lv := strings.Split(line, "|") if len(lv) < 3 { continue } v, err := strconv.ParseFloat(strings.Trim(lv[1], " "), 64) if err == nil { name := strings.ToLower(strings.Replace(strings.Trim(lv[0], " "), " ", "_", -1)) unit := strings.Trim(lv[2], " ") if unit == "Volts" { unit = "V" } else if unit == "degrees C" { unit = "C" } else if unit == "degrees F" { unit = "F" } else if unit == "Watts" { unit = "W" } y, err := lp.New(name, map[string]string{"unit": unit, "type" : "node"}, map[string]interface{}{"value": v}, time.Now()) if err == nil { *out = append(*out, y) } } } } func ReadIpmiSensors(cmd string, out *[]lp.MutableMetric) { command := exec.Command(cmd, "--comma-separated-output", "--sdr-cache-recreate") command.Wait() stdout, err := command.Output() if err != nil { log.Print(err) return } ll := strings.Split(string(stdout), "\n") for _, line := range ll { lv := strings.Split(line, ",") if len(lv) > 3 { v, err := strconv.ParseFloat(lv[3], 64) if err == nil { name := strings.ToLower(strings.Replace(lv[1], " ", "_", -1)) y, err := lp.New(name, map[string]string{"unit": lv[4], "type" : "node"}, map[string]interface{}{"value": v}, time.Now()) if err == nil { *out = append(*out, y) } } } } } func (m *IpmiCollector) Read(interval time.Duration, out *[]lp.MutableMetric) { if len(m.config.IpmitoolPath) > 0 { _, err := os.Stat(m.config.IpmitoolPath) if err == nil { ReadIpmiTool(m.config.IpmitoolPath, out) } } else if len(m.config.IpmisensorsPath) > 0 { _, err := os.Stat(m.config.IpmisensorsPath) if err == nil { ReadIpmiSensors(m.config.IpmisensorsPath, out) } } } func (m *IpmiCollector) Close() { m.init = false return }