package nginx_upstream_check import ( "encoding/json" "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/internal" "github.com/influxdata/telegraf/internal/tls" "github.com/influxdata/telegraf/plugins/inputs" "net/http" "net/url" "strconv" "time" ) const sampleConfig = ` ## An URL where Nginx Upstream check module is enabled ## It should be set to return a JSON formatted response url = "http://127.0.0.1/status?format=json" ## HTTP method # method = "GET" ## Optional HTTP headers # headers = {"X-Special-Header" = "Special-Value"} ## Override HTTP "Host" header # host_header = "check.example.com" ## Timeout for HTTP requests timeout = "5s" ## Optional HTTP Basic Auth credentials # username = "username" # password = "pa$$word" ## Optional TLS Config # tls_ca = "/etc/telegraf/ca.pem" # tls_cert = "/etc/telegraf/cert.pem" # tls_key = "/etc/telegraf/key.pem" ## Use TLS but skip chain & host verification # insecure_skip_verify = false ` const description = "Read nginx_upstream_check module status information (https://github.com/yaoweibin/nginx_upstream_check_module)" type NginxUpstreamCheck struct { URL string `toml:"url"` Username string `toml:"username"` Password string `toml:"password"` Method string `toml:"method"` Headers map[string]string `toml:"headers"` HostHeader string `toml:"host_header"` Timeout internal.Duration `toml:"timeout"` tls.ClientConfig client *http.Client } func NewNginxUpstreamCheck() *NginxUpstreamCheck { return &NginxUpstreamCheck{ URL: "http://127.0.0.1/status?format=json", Method: "GET", Headers: make(map[string]string), HostHeader: "", Timeout: internal.Duration{Duration: time.Second * 5}, } } func init() { inputs.Add("nginx_upstream_check", func() telegraf.Input { return NewNginxUpstreamCheck() }) } func (check *NginxUpstreamCheck) SampleConfig() string { return sampleConfig } func (check *NginxUpstreamCheck) Description() string { return description } type NginxUpstreamCheckData struct { Servers struct { Total uint64 `json:"total"` Generation uint64 `json:"generation"` Server []NginxUpstreamCheckServer `json:"server"` } `json:"servers"` } type NginxUpstreamCheckServer struct { Index uint64 `json:"index"` Upstream string `json:"upstream"` Name string `json:"name"` Status string `json:"status"` Rise uint64 `json:"rise"` Fall uint64 `json:"fall"` Type string `json:"type"` Port uint16 `json:"port"` } // createHttpClient create a clients to access API func (check *NginxUpstreamCheck) createHttpClient() (*http.Client, error) { tlsConfig, err := check.ClientConfig.TLSConfig() if err != nil { return nil, err } client := &http.Client{ Transport: &http.Transport{ TLSClientConfig: tlsConfig, }, Timeout: check.Timeout.Duration, } return client, nil } // gatherJsonData query the data source and parse the response JSON func (check *NginxUpstreamCheck) gatherJsonData(url string, value interface{}) error { var method string if check.Method != "" { method = check.Method } else { method = "GET" } request, err := http.NewRequest(method, url, nil) if err != nil { return err } if (check.Username != "") || (check.Password != "") { request.SetBasicAuth(check.Username, check.Password) } for header, value := range check.Headers { request.Header.Add(header, value) } if check.HostHeader != "" { request.Host = check.HostHeader } response, err := check.client.Do(request) if err != nil { return err } defer response.Body.Close() err = json.NewDecoder(response.Body).Decode(value) if err != nil { return err } return nil } func (check *NginxUpstreamCheck) Gather(accumulator telegraf.Accumulator) error { if check.client == nil { client, err := check.createHttpClient() if err != nil { return err } check.client = client } statusURL, err := url.Parse(check.URL) if err != nil { return err } err = check.gatherStatusData(statusURL.String(), accumulator) if err != nil { return err } return nil } func (check *NginxUpstreamCheck) gatherStatusData(url string, accumulator telegraf.Accumulator) error { checkData := &NginxUpstreamCheckData{} err := check.gatherJsonData(url, checkData) if err != nil { return err } for _, server := range checkData.Servers.Server { tags := map[string]string{ "upstream": server.Upstream, "type": server.Type, "name": server.Name, "port": strconv.Itoa(int(server.Port)), "url": url, } fields := map[string]interface{}{ "status": server.Status, "status_code": check.getStatusCode(server.Status), "rise": server.Rise, "fall": server.Fall, } accumulator.AddFields("nginx_upstream_check", fields, tags) } return nil } func (check *NginxUpstreamCheck) getStatusCode(status string) uint8 { switch status { case "up": return 1 case "down": return 2 default: return 0 } }