telegraf/plugins/outputs/http/http.go

165 lines
3.6 KiB
Go
Raw Normal View History

2018-05-15 00:15:40 +00:00
package http
import (
"bytes"
"fmt"
"io/ioutil"
"net/http"
"strings"
"time"
"github.com/influxdata/telegraf"
"github.com/influxdata/telegraf/internal"
"github.com/influxdata/telegraf/internal/tls"
"github.com/influxdata/telegraf/plugins/outputs"
"github.com/influxdata/telegraf/plugins/serializers"
2018-05-15 00:15:40 +00:00
)
var sampleConfig = `
## URL is the address to send metrics to
2018-05-15 00:15:40 +00:00
url = "http://127.0.0.1:8080/metric"
## Timeout for HTTP message
# timeout = "5s"
## HTTP method, one of: "POST" or "PUT"
# method = "POST"
## 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
2018-05-15 00:15:40 +00:00
## Data format to output.
## Each data format has it's own unique set of configuration options, read
## more about them here:
## https://github.com/influxdata/telegraf/blob/master/docs/DATA_FORMATS_OUTPUT.md
# data_format = "influx"
## Additional HTTP headers
# [outputs.http.headers]
# # Should be set manually to "application/json" for json data_format
# Content-Type = "text/plain; charset=utf-8"
2018-05-15 00:15:40 +00:00
`
const (
defaultClientTimeout = 5 * time.Second
defaultContentType = "text/plain; charset=utf-8"
defaultMethod = http.MethodPost
2018-05-15 00:15:40 +00:00
)
type HTTP struct {
URL string `toml:"url"`
Timeout internal.Duration `toml:"timeout"`
Method string `toml:"method"`
Username string `toml:"username"`
Password string `toml:"password"`
Headers map[string]string `toml:"headers"`
tls.ClientConfig
2018-05-15 00:15:40 +00:00
client *http.Client
2018-05-15 00:15:40 +00:00
serializer serializers.Serializer
}
func (h *HTTP) SetSerializer(serializer serializers.Serializer) {
2018-05-15 00:15:40 +00:00
h.serializer = serializer
}
func (h *HTTP) Connect() error {
if h.Method == "" {
h.Method = http.MethodPost
}
h.Method = strings.ToUpper(h.Method)
if h.Method != http.MethodPost && h.Method != http.MethodPut {
return fmt.Errorf("invalid method [%s] %s", h.URL, h.Method)
2018-05-15 00:15:40 +00:00
}
if h.Timeout.Duration == 0 {
h.Timeout.Duration = defaultClientTimeout
}
2018-05-15 00:15:40 +00:00
tlsCfg, err := h.ClientConfig.TLSConfig()
if err != nil {
return err
2018-05-15 00:15:40 +00:00
}
h.client = &http.Client{
Transport: &http.Transport{
TLSClientConfig: tlsCfg,
Proxy: http.ProxyFromEnvironment,
},
Timeout: h.Timeout.Duration,
2018-05-15 00:15:40 +00:00
}
return nil
}
func (h *HTTP) Close() error {
2018-05-15 00:15:40 +00:00
return nil
}
func (h *HTTP) Description() string {
2018-05-15 00:15:40 +00:00
return "A plugin that can transmit metrics over HTTP"
}
func (h *HTTP) SampleConfig() string {
2018-05-15 00:15:40 +00:00
return sampleConfig
}
func (h *HTTP) Write(metrics []telegraf.Metric) error {
2018-05-15 00:15:40 +00:00
reqBody, err := h.serializer.SerializeBatch(metrics)
if err != nil {
return err
2018-05-15 00:15:40 +00:00
}
if err := h.write(reqBody); err != nil {
return err
}
return nil
}
func (h *HTTP) write(reqBody []byte) error {
req, err := http.NewRequest(h.Method, h.URL, bytes.NewBuffer(reqBody))
if err != nil {
return err
}
2018-05-15 00:15:40 +00:00
if h.Username != "" || h.Password != "" {
req.SetBasicAuth(h.Username, h.Password)
}
req.Header.Set("Content-Type", defaultContentType)
2018-05-15 00:15:40 +00:00
for k, v := range h.Headers {
req.Header.Set(k, v)
}
resp, err := h.client.Do(req)
if err != nil {
return err
}
defer resp.Body.Close()
_, err = ioutil.ReadAll(resp.Body)
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
return fmt.Errorf("when writing to [%s] received status code: %d", h.URL, resp.StatusCode)
2018-05-15 00:15:40 +00:00
}
return nil
}
func init() {
outputs.Add("http", func() telegraf.Output {
return &HTTP{
Timeout: internal.Duration{Duration: defaultClientTimeout},
Method: defaultMethod,
2018-05-15 00:15:40 +00:00
}
})
}