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
179
plugins/inputs/statsd/datadog.go
Normal file
179
plugins/inputs/statsd/datadog.go
Normal file
|
|
@ -0,0 +1,179 @@
|
|||
package statsd
|
||||
|
||||
// this is adapted from datadog's apache licensed version at
|
||||
// https://github.com/DataDog/datadog-agent/blob/fcfc74f106ab1bd6991dfc6a7061c558d934158a/pkg/dogstatsd/parser.go#L173
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
priorityNormal = "normal"
|
||||
priorityLow = "low"
|
||||
|
||||
eventInfo = "info"
|
||||
eventWarning = "warning"
|
||||
eventError = "error"
|
||||
eventSuccess = "success"
|
||||
)
|
||||
|
||||
var uncommenter = strings.NewReplacer("\\n", "\n")
|
||||
|
||||
func (s *Statsd) parseEventMessage(now time.Time, message, defaultHostname string) error {
|
||||
// _e{title.length,text.length}:title|text
|
||||
// [
|
||||
// |d:date_happened
|
||||
// |p:priority
|
||||
// |h:hostname
|
||||
// |t:alert_type
|
||||
// |s:source_type_name
|
||||
// |#tag1,tag2
|
||||
// ]
|
||||
//
|
||||
//
|
||||
// tag is key:value
|
||||
messageRaw := strings.SplitN(message, ":", 2)
|
||||
if len(messageRaw) < 2 || len(messageRaw[0]) < 7 || len(messageRaw[1]) < 3 {
|
||||
return errors.New("invalid message format")
|
||||
}
|
||||
header := messageRaw[0]
|
||||
message = messageRaw[1]
|
||||
|
||||
rawLen := strings.SplitN(header[3:], ",", 2)
|
||||
if len(rawLen) != 2 {
|
||||
return errors.New("invalid message format")
|
||||
}
|
||||
|
||||
titleLen, err := strconv.ParseInt(rawLen[0], 10, 64)
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid message format, could not parse title.length: %q", rawLen[0])
|
||||
}
|
||||
if len(rawLen[1]) < 1 {
|
||||
return fmt.Errorf("invalid message format, could not parse text.length: %q", rawLen[0])
|
||||
}
|
||||
textLen, err := strconv.ParseInt(rawLen[1][:len(rawLen[1])-1], 10, 64)
|
||||
if err != nil {
|
||||
return fmt.Errorf("invalid message format, could not parse text.length: %q", rawLen[0])
|
||||
}
|
||||
if titleLen+textLen+1 > int64(len(message)) {
|
||||
return errors.New("invalid message format, title.length and text.length exceed total message length")
|
||||
}
|
||||
|
||||
rawTitle := message[:titleLen]
|
||||
rawText := message[titleLen+1 : titleLen+1+textLen]
|
||||
message = message[titleLen+1+textLen:]
|
||||
|
||||
if len(rawTitle) == 0 || len(rawText) == 0 {
|
||||
return errors.New("invalid event message format: empty 'title' or 'text' field")
|
||||
}
|
||||
|
||||
name := rawTitle
|
||||
tags := make(map[string]string, strings.Count(message, ",")+2) // allocate for the approximate number of tags
|
||||
fields := make(map[string]interface{}, 9)
|
||||
fields["alert_type"] = eventInfo // default event type
|
||||
fields["text"] = uncommenter.Replace(rawText)
|
||||
if defaultHostname != "" {
|
||||
tags["source"] = defaultHostname
|
||||
}
|
||||
fields["priority"] = priorityNormal
|
||||
ts := now
|
||||
if len(message) < 2 {
|
||||
s.acc.AddFields(name, fields, tags, ts)
|
||||
return nil
|
||||
}
|
||||
|
||||
rawMetadataFields := strings.Split(message[1:], "|")
|
||||
for i := range rawMetadataFields {
|
||||
if len(rawMetadataFields[i]) < 2 {
|
||||
return errors.New("too short metadata field")
|
||||
}
|
||||
switch rawMetadataFields[i][:2] {
|
||||
case "d:":
|
||||
ts, err := strconv.ParseInt(rawMetadataFields[i][2:], 10, 64)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
fields["ts"] = ts
|
||||
case "p:":
|
||||
switch rawMetadataFields[i][2:] {
|
||||
case priorityLow:
|
||||
fields["priority"] = priorityLow
|
||||
case priorityNormal: // we already used this as a default
|
||||
default:
|
||||
continue
|
||||
}
|
||||
case "h:":
|
||||
tags["source"] = rawMetadataFields[i][2:]
|
||||
case "t:":
|
||||
switch rawMetadataFields[i][2:] {
|
||||
case eventError, eventWarning, eventSuccess, eventInfo:
|
||||
fields["alert_type"] = rawMetadataFields[i][2:] // already set for info
|
||||
default:
|
||||
continue
|
||||
}
|
||||
case "k:":
|
||||
tags["aggregation_key"] = rawMetadataFields[i][2:]
|
||||
case "s:":
|
||||
fields["source_type_name"] = rawMetadataFields[i][2:]
|
||||
default:
|
||||
if rawMetadataFields[i][0] != '#' {
|
||||
return fmt.Errorf("unknown metadata type: %q", rawMetadataFields[i])
|
||||
}
|
||||
parseDataDogTags(tags, rawMetadataFields[i][1:])
|
||||
}
|
||||
}
|
||||
// Use source tag because host is reserved tag key in Telegraf.
|
||||
// In datadog the host tag and `h:` are interchangeable, so we have to check for the host tag.
|
||||
if host, ok := tags["host"]; ok {
|
||||
delete(tags, "host")
|
||||
tags["source"] = host
|
||||
}
|
||||
s.acc.AddFields(name, fields, tags, ts)
|
||||
return nil
|
||||
}
|
||||
|
||||
func parseDataDogTags(tags map[string]string, message string) {
|
||||
if len(message) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
start, i := 0, 0
|
||||
var k string
|
||||
var inVal bool // check if we are parsing the value part of the tag
|
||||
for i = range message {
|
||||
if message[i] == ',' {
|
||||
if k == "" {
|
||||
k = message[start:i]
|
||||
tags[k] = "true" // this is because influx doesn't support empty tags
|
||||
start = i + 1
|
||||
continue
|
||||
}
|
||||
v := message[start:i]
|
||||
if v == "" {
|
||||
v = "true"
|
||||
}
|
||||
tags[k] = v
|
||||
start = i + 1
|
||||
k, inVal = "", false // reset state vars
|
||||
} else if message[i] == ':' && !inVal {
|
||||
k = message[start:i]
|
||||
start = i + 1
|
||||
inVal = true
|
||||
}
|
||||
}
|
||||
if k == "" && start < i+1 {
|
||||
tags[message[start:i+1]] = "true"
|
||||
}
|
||||
// grab the last value
|
||||
if k != "" {
|
||||
if start < i+1 {
|
||||
tags[k] = message[start : i+1]
|
||||
return
|
||||
}
|
||||
tags[k] = "true"
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue