145 lines
2.7 KiB
Go
145 lines
2.7 KiB
Go
package consul
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/hashicorp/consul/api"
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/internal/tls"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
)
|
|
|
|
type Consul struct {
|
|
Address string
|
|
Scheme string
|
|
Token string
|
|
Username string
|
|
Password string
|
|
Datacentre string
|
|
tls.ClientConfig
|
|
|
|
// client used to connect to Consul agnet
|
|
client *api.Client
|
|
}
|
|
|
|
var sampleConfig = `
|
|
## Consul server address
|
|
# address = "localhost"
|
|
|
|
## URI scheme for the Consul server, one of "http", "https"
|
|
# scheme = "http"
|
|
|
|
## ACL token used in every request
|
|
# token = ""
|
|
|
|
## HTTP Basic Authentication username and password.
|
|
# username = ""
|
|
# password = ""
|
|
|
|
## Data centre to query the health checks from
|
|
# datacentre = ""
|
|
|
|
## 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 = true
|
|
`
|
|
|
|
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.Token != "" {
|
|
config.Token = c.Token
|
|
}
|
|
|
|
if c.Username != "" {
|
|
config.HttpAuth = &api.HttpBasicAuth{
|
|
Username: c.Username,
|
|
Password: c.Password,
|
|
}
|
|
}
|
|
|
|
tlsCfg, err := c.ClientConfig.TLSConfig()
|
|
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
|
|
record["passing"] = 0
|
|
record["critical"] = 0
|
|
record["warning"] = 0
|
|
record[check.Status] = 1
|
|
|
|
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{}
|
|
})
|
|
}
|