diff --git a/plugins/inputs/haproxy/haproxy.go b/plugins/inputs/haproxy/haproxy.go index bcb0fb17c..25a2a13b5 100644 --- a/plugins/inputs/haproxy/haproxy.go +++ b/plugins/inputs/haproxy/haproxy.go @@ -1,24 +1,24 @@ package haproxy import ( - "encoding/csv" - "fmt" - "github.com/influxdata/telegraf" - "github.com/influxdata/telegraf/plugins/inputs" - "io" - "net" - "net/http" - "net/url" - "strconv" - "strings" - "sync" - "time" + "encoding/csv" + "fmt" + "github.com/influxdata/telegraf" + "github.com/influxdata/telegraf/plugins/inputs" + "io" + "net" + "net/http" + "net/url" + "strconv" + "strings" + "sync" + "time" ) type haproxy struct { - Servers []string + Servers []string - client *http.Client + client *http.Client } var sampleConfig = ` @@ -31,214 +31,214 @@ var sampleConfig = ` ` func (r *haproxy) SampleConfig() string { - return sampleConfig + return sampleConfig } func (r *haproxy) Description() string { - return "Read metrics of haproxy, via socket or csv stats page" + return "Read metrics of haproxy, via socket or csv stats page" } // Reads stats from all configured servers accumulates stats. // Returns one of the errors encountered while gather stats (if any). func (g *haproxy) Gather(acc telegraf.Accumulator) error { - if len(g.Servers) == 0 { - return g.gatherServer("http://127.0.0.1:1936", acc) - } + if len(g.Servers) == 0 { + return g.gatherServer("http://127.0.0.1:1936", acc) + } - var wg sync.WaitGroup + var wg sync.WaitGroup - var outerr error + var outerr error - for _, serv := range g.Servers { - wg.Add(1) - go func(serv string) { - defer wg.Done() - outerr = g.gatherServer(serv, acc) - }(serv) - } + for _, serv := range g.Servers { + wg.Add(1) + go func(serv string) { + defer wg.Done() + outerr = g.gatherServer(serv, acc) + }(serv) + } - wg.Wait() + wg.Wait() - return outerr + return outerr } func (g *haproxy) gatherServerSocket(addr string, acc telegraf.Accumulator) error { - var socketPath string - socketAddr := strings.Split(addr, ":") + var socketPath string + socketAddr := strings.Split(addr, ":") - if len(socketAddr) >= 2 { - socketPath = socketAddr[1] - } else { - socketPath = socketAddr[0] - } + if len(socketAddr) >= 2 { + socketPath = socketAddr[1] + } else { + socketPath = socketAddr[0] + } - c, err := net.Dial("unix", socketPath) + c, err := net.Dial("unix", socketPath) - if err != nil { - return fmt.Errorf("Could not connect to socket '%s': %s", addr, err) - } + if err != nil { + return fmt.Errorf("Could not connect to socket '%s': %s", addr, err) + } - _, errw := c.Write([]byte("show stat\n")) + _, errw := c.Write([]byte("show stat\n")) - if errw != nil { - return fmt.Errorf("Could not write to socket '%s': %s", addr, errw) - } + if errw != nil { + return fmt.Errorf("Could not write to socket '%s': %s", addr, errw) + } - return importCsvResult(c, acc, socketPath) + return importCsvResult(c, acc, socketPath) } func (g *haproxy) gatherServer(addr string, acc telegraf.Accumulator) error { - if !strings.HasPrefix(addr, "http") { - return g.gatherServerSocket(addr, acc) - } + if !strings.HasPrefix(addr, "http") { + return g.gatherServerSocket(addr, acc) + } - if g.client == nil { - tr := &http.Transport{ResponseHeaderTimeout: time.Duration(3 * time.Second)} - client := &http.Client{ - Transport: tr, - Timeout: time.Duration(4 * time.Second), - } - g.client = client - } + if g.client == nil { + tr := &http.Transport{ResponseHeaderTimeout: time.Duration(3 * time.Second)} + client := &http.Client{ + Transport: tr, + Timeout: time.Duration(4 * time.Second), + } + g.client = client + } - u, err := url.Parse(addr) - if err != nil { - return fmt.Errorf("Unable parse server address '%s': %s", addr, err) - } + u, err := url.Parse(addr) + if err != nil { + return fmt.Errorf("Unable parse server address '%s': %s", addr, err) + } - req, err := http.NewRequest("GET", fmt.Sprintf("%s://%s%s/;csv", u.Scheme, u.Host, u.Path), nil) - if u.User != nil { - p, _ := u.User.Password() - req.SetBasicAuth(u.User.Username(), p) - } + req, err := http.NewRequest("GET", fmt.Sprintf("%s://%s%s/;csv", u.Scheme, u.Host, u.Path), nil) + if u.User != nil { + p, _ := u.User.Password() + req.SetBasicAuth(u.User.Username(), p) + } - res, err := g.client.Do(req) - if err != nil { - return fmt.Errorf("Unable to connect to haproxy server '%s': %s", addr, err) - } + res, err := g.client.Do(req) + if err != nil { + return fmt.Errorf("Unable to connect to haproxy server '%s': %s", addr, err) + } - if res.StatusCode != 200 { - return fmt.Errorf("Unable to get valid stat result from '%s': %s", addr, err) - } + if res.StatusCode != 200 { + return fmt.Errorf("Unable to get valid stat result from '%s': %s", addr, err) + } - return importCsvResult(res.Body, acc, u.Host) + return importCsvResult(res.Body, acc, u.Host) } func importCsvResult(r io.Reader, acc telegraf.Accumulator, host string) error { - csv := csv.NewReader(r) - result, err := csv.ReadAll() - now := time.Now() + csv := csv.NewReader(r) + result, err := csv.ReadAll() + now := time.Now() - var keys []string - px_sv_status := make(map[string]map[string]int) - non_stat_fields := map[string]bool{ - "pxname": true, - "svname": true, - "status": true, - "tracked": true, - "check_status": true, - "last_chk": true, - "pid": true, - "iid": true, - "sid": true, - "lastchg": true, - "type": true, - "check_code": true, - } + var keys []string + px_sv_status := make(map[string]map[string]int) + non_stat_fields := map[string]bool{ + "pxname": true, + "svname": true, + "status": true, + "tracked": true, + "check_status": true, + "last_chk": true, + "pid": true, + "iid": true, + "sid": true, + "lastchg": true, + "type": true, + "check_code": true, + } - for i := range result { - if i == 0 { - keys = result[i] - keys[0] = strings.Replace(keys[0], "# ", "", -1) - } else { + for i := range result { + if i == 0 { + keys = result[i] + keys[0] = strings.Replace(keys[0], "# ", "", -1) + } else { - row := make(map[string]string, len(result[i])) + row := make(map[string]string, len(result[i])) - for f, v := range result[i] { - row[keys[f]] = v - } + for f, v := range result[i] { + row[keys[f]] = v + } - tags := map[string]string{ - "server": host, - "proxy": row["pxname"], - "sv": row["svname"], - } + tags := map[string]string{ + "server": host, + "proxy": row["pxname"], + "sv": row["svname"], + } - if row["svname"] != "BACKEND" && row["svname"] != "FRONTEND" { - if len(px_sv_status[row["pxname"]]) == 0 { - px_sv_status[row["pxname"]] = map[string]int{ - "_status_act": 0, - "_status_act_up": 0, - "_status_act_down": 0, - "_status_act_maint": 0, - "_status_act_drain": 0, - "_status_act_other": 0, - "_status_bck": 0, - "_status_bck_up": 0, - "_status_bck_down": 0, - "_status_bck_maint": 0, - "_status_bck_drain": 0, - "_status_bck_other": 0, - } - } + if row["svname"] != "BACKEND" && row["svname"] != "FRONTEND" { + if len(px_sv_status[row["pxname"]]) == 0 { + px_sv_status[row["pxname"]] = map[string]int{ + "_status_act": 0, + "_status_act_up": 0, + "_status_act_down": 0, + "_status_act_maint": 0, + "_status_act_drain": 0, + "_status_act_other": 0, + "_status_bck": 0, + "_status_bck_up": 0, + "_status_bck_down": 0, + "_status_bck_maint": 0, + "_status_bck_drain": 0, + "_status_bck_other": 0, + } + } - if row["act"] == "1" { - px_sv_status[row["pxname"]]["_status_act"]++ - switch row["status"] { - case "UP": - px_sv_status[row["pxname"]]["_status_act_up"]++ - case "DOWN": - px_sv_status[row["pxname"]]["_status_act_down"]++ - case "MAINT": - px_sv_status[row["pxname"]]["_status_act_maint"]++ - case "DRAIN": - px_sv_status[row["pxname"]]["_status_act_drain"]++ - default: - px_sv_status[row["pxname"]]["_status_act_other"]++ - } + if row["act"] == "1" { + px_sv_status[row["pxname"]]["_status_act"]++ + switch row["status"] { + case "UP": + px_sv_status[row["pxname"]]["_status_act_up"]++ + case "DOWN": + px_sv_status[row["pxname"]]["_status_act_down"]++ + case "MAINT": + px_sv_status[row["pxname"]]["_status_act_maint"]++ + case "DRAIN": + px_sv_status[row["pxname"]]["_status_act_drain"]++ + default: + px_sv_status[row["pxname"]]["_status_act_other"]++ + } - } else { - px_sv_status[row["pxname"]]["_status_bck"]++ - switch row["status"] { - case "UP": - px_sv_status[row["pxname"]]["_status_bck_up"]++ - case "DOWN": - px_sv_status[row["pxname"]]["_status_bck_down"]++ - case "MAINT": - px_sv_status[row["pxname"]]["_status_bck_maint"]++ - case "DRAIN": - px_sv_status[row["pxname"]]["_status_bck_drain"]++ - default: - px_sv_status[row["pxname"]]["_status_bck_other"]++ - } - } - } + } else { + px_sv_status[row["pxname"]]["_status_bck"]++ + switch row["status"] { + case "UP": + px_sv_status[row["pxname"]]["_status_bck_up"]++ + case "DOWN": + px_sv_status[row["pxname"]]["_status_bck_down"]++ + case "MAINT": + px_sv_status[row["pxname"]]["_status_bck_maint"]++ + case "DRAIN": + px_sv_status[row["pxname"]]["_status_bck_drain"]++ + default: + px_sv_status[row["pxname"]]["_status_bck_other"]++ + } + } + } - if row["svname"] == "BACKEND" && len(px_sv_status[row["pxname"]]) > 0 { - for s, c := range px_sv_status[row["pxname"]] { - row[s] = strconv.Itoa(c) - } - } + if row["svname"] == "BACKEND" && len(px_sv_status[row["pxname"]]) > 0 { + for s, c := range px_sv_status[row["pxname"]] { + row[s] = strconv.Itoa(c) + } + } - fields := make(map[string]interface{}) - for field, v := range row { - if non_stat_fields[field] { - continue - } + fields := make(map[string]interface{}) + for field, v := range row { + if non_stat_fields[field] { + continue + } - ival, err := strconv.ParseUint(v, 10, 64) - if err == nil { - fields[field] = ival - } - } - acc.AddFields("haproxy", fields, tags, now) - } - } - return err + ival, err := strconv.ParseUint(v, 10, 64) + if err == nil { + fields[field] = ival + } + } + acc.AddFields("haproxy", fields, tags, now) + } + } + return err } func init() { - inputs.Add("haproxy", func() telegraf.Input { - return &haproxy{} - }) + inputs.Add("haproxy", func() telegraf.Input { + return &haproxy{} + }) }