68b351ca05
* plugins/input/consul: moved check_id from regular fields to tags. When service has more than one check sending data for both would overwrite each other resulting only in one check being written (the last one). Adding check_id as a tag ensures we will get info for all unique checks per service. * plugins/inputs/consul: updated tests
137 lines
2.9 KiB
Go
137 lines
2.9 KiB
Go
package consul
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/internal"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
)
|
|
|
|
type Consul struct {
|
|
Address string
|
|
Scheme string
|
|
Token string
|
|
Username string
|
|
Password string
|
|
Datacentre string
|
|
|
|
// Path to CA file
|
|
SSLCA string `toml:"ssl_ca"`
|
|
// Path to host cert file
|
|
SSLCert string `toml:"ssl_cert"`
|
|
// Path to cert key file
|
|
SSLKey string `toml:"ssl_key"`
|
|
// Use SSL but skip chain & host verification
|
|
InsecureSkipVerify bool
|
|
|
|
// client used to connect to Consul agnet
|
|
client *api.Client
|
|
}
|
|
|
|
var sampleConfig = `
|
|
## Most of these values defaults to the one configured on a Consul's agent level.
|
|
## Optional Consul server address (default: "localhost")
|
|
# address = "localhost"
|
|
## Optional URI scheme for the Consul server (default: "http")
|
|
# scheme = "http"
|
|
## Optional ACL token used in every request (default: "")
|
|
# token = ""
|
|
## Optional username used for request HTTP Basic Authentication (default: "")
|
|
# username = ""
|
|
## Optional password used for HTTP Basic Authentication (default: "")
|
|
# password = ""
|
|
## Optional data centre to query the health checks from (default: "")
|
|
# datacentre = ""
|
|
`
|
|
|
|
func (c *Consul) Description() string {
|
|
return "Gather health check statuses from services registered in Consul"
|
|
}
|
|
|
|
func (c *Consul) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (c *Consul) createAPIClient() (*api.Client, error) {
|
|
config := api.DefaultConfig()
|
|
|
|
if c.Address != "" {
|
|
config.Address = c.Address
|
|
}
|
|
|
|
if c.Scheme != "" {
|
|
config.Scheme = c.Scheme
|
|
}
|
|
|
|
if c.Datacentre != "" {
|
|
config.Datacenter = c.Datacentre
|
|
}
|
|
|
|
if c.Username != "" {
|
|
config.HttpAuth = &api.HttpBasicAuth{
|
|
Username: c.Username,
|
|
Password: c.Password,
|
|
}
|
|
}
|
|
|
|
tlsCfg, err := internal.GetTLSConfig(
|
|
c.SSLCert, c.SSLKey, c.SSLCA, c.InsecureSkipVerify)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
config.HttpClient.Transport = &http.Transport{
|
|
TLSClientConfig: tlsCfg,
|
|
}
|
|
|
|
return api.NewClient(config)
|
|
}
|
|
|
|
func (c *Consul) GatherHealthCheck(acc telegraf.Accumulator, checks []*api.HealthCheck) {
|
|
for _, check := range checks {
|
|
record := make(map[string]interface{})
|
|
tags := make(map[string]string)
|
|
|
|
record["check_name"] = check.Name
|
|
record["service_id"] = check.ServiceID
|
|
record["status"] = check.Status
|
|
|
|
tags["node"] = check.Node
|
|
tags["service_name"] = check.ServiceName
|
|
tags["check_id"] = check.CheckID
|
|
|
|
acc.AddFields("consul_health_checks", record, tags)
|
|
}
|
|
}
|
|
|
|
func (c *Consul) Gather(acc telegraf.Accumulator) error {
|
|
if c.client == nil {
|
|
newClient, err := c.createAPIClient()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.client = newClient
|
|
}
|
|
|
|
checks, _, err := c.client.Health().State("any", nil)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
c.GatherHealthCheck(acc, checks)
|
|
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("consul", func() telegraf.Input {
|
|
return &Consul{}
|
|
})
|
|
}
|