Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
GPG key ID:
FBB4F0E80A80222F
4963 changed files with 677545 additions and 0 deletions
419
plugins/outputs/sensu/sensu.go
Normal file
419
plugins/outputs/sensu/sensu.go
Normal file
|
|
@ -0,0 +1,419 @@
|
|||
//go:generate ../../../tools/readme_config_includer/generator
|
||||
package sensu
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
_ "embed"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"math"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"path"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/config"
|
||||
"github.com/influxdata/telegraf/internal"
|
||||
"github.com/influxdata/telegraf/internal/choice"
|
||||
"github.com/influxdata/telegraf/plugins/common/tls"
|
||||
"github.com/influxdata/telegraf/plugins/outputs"
|
||||
)
|
||||
|
||||
//go:embed sample.conf
|
||||
var sampleConfig string
|
||||
|
||||
const (
|
||||
defaultURL = "http://127.0.0.1:3031"
|
||||
defaultClientTimeout = 5 * time.Second
|
||||
defaultContentType = "application/json; charset=utf-8"
|
||||
)
|
||||
|
||||
type outputMetadata struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type outputEntity struct {
|
||||
Metadata *outputMetadata `json:"metadata"`
|
||||
}
|
||||
|
||||
type outputCheck struct {
|
||||
Metadata *outputMetadata `json:"metadata"`
|
||||
Status int `json:"status"`
|
||||
Output string `json:"output"`
|
||||
Issued int64 `json:"issued"`
|
||||
OutputMetricHandlers []string `json:"output_metric_handlers"`
|
||||
}
|
||||
|
||||
type outputMetrics struct {
|
||||
Handlers []string `json:"handlers"`
|
||||
Metrics []*outputMetric `json:"points"`
|
||||
}
|
||||
|
||||
type outputMetric struct {
|
||||
Name string `json:"name"`
|
||||
Tags []*outputTag `json:"tags"`
|
||||
Value interface{} `json:"value"`
|
||||
Timestamp int64 `json:"timestamp"`
|
||||
}
|
||||
|
||||
type outputTag struct {
|
||||
Name string `json:"name"`
|
||||
Value string `json:"value"`
|
||||
}
|
||||
|
||||
type outputEvent struct {
|
||||
Entity *outputEntity `json:"entity,omitempty"`
|
||||
Check *outputCheck `json:"check"`
|
||||
Metrics *outputMetrics `json:"metrics"`
|
||||
Timestamp int64 `json:"timestamp"`
|
||||
}
|
||||
|
||||
type sensuEntity struct {
|
||||
Name *string `toml:"name"`
|
||||
Namespace *string `toml:"namespace"`
|
||||
}
|
||||
|
||||
type sensuCheck struct {
|
||||
Name *string `toml:"name"`
|
||||
}
|
||||
|
||||
type sensuMetrics struct {
|
||||
Handlers []string `toml:"handlers"`
|
||||
}
|
||||
|
||||
type Sensu struct {
|
||||
APIKey *string `toml:"api_key"`
|
||||
AgentAPIURL *string `toml:"agent_api_url"`
|
||||
BackendAPIURL *string `toml:"backend_api_url"`
|
||||
Entity *sensuEntity `toml:"entity"`
|
||||
Tags map[string]string `toml:"tags"`
|
||||
Metrics *sensuMetrics `toml:"metrics"`
|
||||
Check *sensuCheck `toml:"check"`
|
||||
|
||||
Timeout config.Duration `toml:"timeout"`
|
||||
ContentEncoding string `toml:"content_encoding"`
|
||||
|
||||
EndpointURL string
|
||||
OutEntity *outputEntity
|
||||
|
||||
Log telegraf.Logger `toml:"-"`
|
||||
|
||||
tls.ClientConfig
|
||||
client *http.Client
|
||||
}
|
||||
|
||||
func (s *Sensu) createClient() (*http.Client, error) {
|
||||
tlsCfg, err := s.ClientConfig.TLSConfig()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
client := &http.Client{
|
||||
Transport: &http.Transport{
|
||||
TLSClientConfig: tlsCfg,
|
||||
},
|
||||
Timeout: time.Duration(s.Timeout),
|
||||
}
|
||||
|
||||
return client, nil
|
||||
}
|
||||
|
||||
func (*Sensu) SampleConfig() string {
|
||||
return sampleConfig
|
||||
}
|
||||
|
||||
func (s *Sensu) Connect() error {
|
||||
err := s.setEndpointURL()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
err = s.setEntity()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
client, err := s.createClient()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
s.client = client
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Sensu) Close() error {
|
||||
s.client.CloseIdleConnections()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Sensu) Write(metrics []telegraf.Metric) error {
|
||||
var points []*outputMetric
|
||||
for _, metric := range metrics {
|
||||
// Add tags from config to each metric point
|
||||
tagList := make([]*outputTag, 0, len(s.Tags)+len(metric.TagList()))
|
||||
for name, value := range s.Tags {
|
||||
tag := &outputTag{
|
||||
Name: name,
|
||||
Value: value,
|
||||
}
|
||||
tagList = append(tagList, tag)
|
||||
}
|
||||
for _, tagSet := range metric.TagList() {
|
||||
tag := &outputTag{
|
||||
Name: tagSet.Key,
|
||||
Value: tagSet.Value,
|
||||
}
|
||||
tagList = append(tagList, tag)
|
||||
}
|
||||
|
||||
// Get all valid numeric values, convert to float64
|
||||
for _, fieldSet := range metric.FieldList() {
|
||||
key := fieldSet.Key
|
||||
value := getFloat(fieldSet.Value)
|
||||
// JSON does not support these special values
|
||||
if math.IsInf(value, 1) {
|
||||
s.Log.Debugf("metric %s returned positive infinity, setting value to %f", key, math.MaxFloat64)
|
||||
value = math.MaxFloat64
|
||||
}
|
||||
if math.IsInf(value, -1) {
|
||||
s.Log.Debugf("metric %s returned negative infinity, setting value to %f", key, -math.MaxFloat64)
|
||||
value = -math.MaxFloat64
|
||||
}
|
||||
if math.IsNaN(value) {
|
||||
s.Log.Debugf("metric %s returned as non a number, skipping", key)
|
||||
continue
|
||||
}
|
||||
|
||||
point := &outputMetric{
|
||||
Name: metric.Name() + "." + key,
|
||||
Tags: tagList,
|
||||
Timestamp: metric.Time().Unix(),
|
||||
Value: value,
|
||||
}
|
||||
points = append(points, point)
|
||||
}
|
||||
}
|
||||
|
||||
reqBody, err := s.encodeToJSON(points)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return s.writeMetrics(reqBody)
|
||||
}
|
||||
|
||||
func (s *Sensu) writeMetrics(reqBody []byte) error {
|
||||
var reqBodyBuffer io.Reader = bytes.NewBuffer(reqBody)
|
||||
method := http.MethodPost
|
||||
|
||||
if s.ContentEncoding == "gzip" {
|
||||
rc := internal.CompressWithGzip(reqBodyBuffer)
|
||||
defer rc.Close()
|
||||
reqBodyBuffer = rc
|
||||
}
|
||||
|
||||
req, err := http.NewRequest(method, s.EndpointURL, reqBodyBuffer)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
req.Header.Set("User-Agent", internal.ProductToken())
|
||||
|
||||
req.Header.Set("Content-Type", defaultContentType)
|
||||
if s.ContentEncoding == "gzip" {
|
||||
req.Header.Set("Content-Encoding", "gzip")
|
||||
}
|
||||
|
||||
if s.APIKey != nil {
|
||||
req.Header.Set("Authorization", "Key "+*s.APIKey)
|
||||
}
|
||||
|
||||
resp, err := s.client.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusCreated {
|
||||
bodyData, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
s.Log.Debugf("Couldn't read response body: %v", err)
|
||||
}
|
||||
s.Log.Debugf("Failed to write, response: %v", string(bodyData))
|
||||
if resp.StatusCode < 400 || resp.StatusCode > 499 {
|
||||
return fmt.Errorf("when writing to [%s] received status code: %d", s.EndpointURL, resp.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Resolves the event write endpoint
|
||||
func (s *Sensu) setEndpointURL() error {
|
||||
var (
|
||||
endpointURL string
|
||||
pathSuffix string
|
||||
)
|
||||
|
||||
if s.BackendAPIURL != nil {
|
||||
endpointURL = *s.BackendAPIURL
|
||||
namespace := "default"
|
||||
if s.Entity != nil && s.Entity.Namespace != nil {
|
||||
namespace = *s.Entity.Namespace
|
||||
}
|
||||
pathSuffix = "/api/core/v2/namespaces/" + namespace + "/events"
|
||||
} else if s.AgentAPIURL != nil {
|
||||
endpointURL = *s.AgentAPIURL
|
||||
pathSuffix = "/events"
|
||||
}
|
||||
|
||||
if len(endpointURL) == 0 {
|
||||
s.Log.Debugf("no backend or agent API URL provided, falling back to default agent API URL %s", defaultURL)
|
||||
endpointURL = defaultURL
|
||||
pathSuffix = "/events"
|
||||
}
|
||||
|
||||
u, err := url.Parse(endpointURL)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
u.Path = path.Join(u.Path, pathSuffix)
|
||||
s.EndpointURL = u.String()
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Sensu) Init() error {
|
||||
if len(s.ContentEncoding) != 0 {
|
||||
validEncoding := []string{"identity", "gzip"}
|
||||
if !choice.Contains(s.ContentEncoding, validEncoding) {
|
||||
return fmt.Errorf("unsupported content_encoding [%q] specified", s.ContentEncoding)
|
||||
}
|
||||
}
|
||||
|
||||
if s.BackendAPIURL != nil && s.APIKey == nil {
|
||||
return fmt.Errorf("backend_api_url [%q] specified, but no API Key provided", *s.BackendAPIURL)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
outputs.Add("sensu", func() telegraf.Output {
|
||||
// Default configuration values
|
||||
|
||||
// make a string from the defaultURL const
|
||||
agentAPIURL := defaultURL
|
||||
|
||||
return &Sensu{
|
||||
AgentAPIURL: &agentAPIURL,
|
||||
Timeout: config.Duration(defaultClientTimeout),
|
||||
ContentEncoding: "identity",
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func (s *Sensu) encodeToJSON(metricPoints []*outputMetric) ([]byte, error) {
|
||||
timestamp := time.Now().Unix()
|
||||
|
||||
check, err := s.getCheck(metricPoints)
|
||||
if err != nil {
|
||||
return make([]byte, 0), err
|
||||
}
|
||||
|
||||
output, err := json.Marshal(&outputEvent{
|
||||
Entity: s.OutEntity,
|
||||
Check: check,
|
||||
Metrics: &outputMetrics{
|
||||
Handlers: s.getHandlers(),
|
||||
Metrics: metricPoints,
|
||||
},
|
||||
Timestamp: timestamp,
|
||||
})
|
||||
|
||||
return output, err
|
||||
}
|
||||
|
||||
// Constructs the entity payload
|
||||
// Throws when no entity name is provided and fails resolve to hostname
|
||||
func (s *Sensu) setEntity() error {
|
||||
if s.BackendAPIURL != nil {
|
||||
var entityName string
|
||||
if s.Entity != nil && s.Entity.Name != nil {
|
||||
entityName = *s.Entity.Name
|
||||
} else {
|
||||
defaultHostname, err := os.Hostname()
|
||||
if err != nil {
|
||||
return fmt.Errorf("resolving hostname failed: %w", err)
|
||||
}
|
||||
entityName = defaultHostname
|
||||
}
|
||||
|
||||
s.OutEntity = &outputEntity{
|
||||
Metadata: &outputMetadata{
|
||||
Name: entityName,
|
||||
},
|
||||
}
|
||||
return nil
|
||||
}
|
||||
s.OutEntity = &outputEntity{}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Constructs the check payload
|
||||
// Throws if check name is not provided
|
||||
func (s *Sensu) getCheck(metricPoints []*outputMetric) (*outputCheck, error) {
|
||||
count := len(metricPoints)
|
||||
|
||||
if s.Check == nil || s.Check.Name == nil {
|
||||
return &outputCheck{}, errors.New("missing check name")
|
||||
}
|
||||
|
||||
return &outputCheck{
|
||||
Metadata: &outputMetadata{
|
||||
Name: *s.Check.Name,
|
||||
},
|
||||
Status: 0, // Always OK
|
||||
Issued: time.Now().Unix(),
|
||||
Output: "Telegraf agent processed " + strconv.Itoa(count) + " metrics",
|
||||
OutputMetricHandlers: s.getHandlers(),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *Sensu) getHandlers() []string {
|
||||
if s.Metrics == nil || s.Metrics.Handlers == nil {
|
||||
return make([]string, 0)
|
||||
}
|
||||
return s.Metrics.Handlers
|
||||
}
|
||||
|
||||
func getFloat(unk interface{}) float64 {
|
||||
switch i := unk.(type) {
|
||||
case float64:
|
||||
return i
|
||||
case float32:
|
||||
return float64(i)
|
||||
case int64:
|
||||
return float64(i)
|
||||
case int32:
|
||||
return float64(i)
|
||||
case int:
|
||||
return float64(i)
|
||||
case uint64:
|
||||
return float64(i)
|
||||
case uint32:
|
||||
return float64(i)
|
||||
case uint:
|
||||
return float64(i)
|
||||
default:
|
||||
return math.NaN()
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue