telegraf/accumulator.go

189 lines
3.7 KiB
Go
Raw Normal View History

package telegraf
import (
"fmt"
"log"
"math"
"sync"
"time"
"github.com/influxdb/telegraf/internal/config"
"github.com/influxdb/influxdb/client/v2"
)
type Accumulator interface {
Add(measurement string, value interface{},
tags map[string]string, t ...time.Time)
AddFields(measurement string, fields map[string]interface{},
tags map[string]string, t ...time.Time)
SetDefaultTags(tags map[string]string)
AddDefaultTag(key, value string)
Prefix() string
SetPrefix(prefix string)
Debug() bool
SetDebug(enabled bool)
}
func NewAccumulator(
2016-01-07 20:39:43 +00:00
inputConfig *config.InputConfig,
points chan *client.Point,
) Accumulator {
acc := accumulator{}
acc.points = points
2016-01-07 20:39:43 +00:00
acc.inputConfig = inputConfig
return &acc
}
type accumulator struct {
sync.Mutex
points chan *client.Point
defaultTags map[string]string
debug bool
2016-01-07 20:39:43 +00:00
inputConfig *config.InputConfig
prefix string
}
func (ac *accumulator) Add(
measurement string,
value interface{},
tags map[string]string,
t ...time.Time,
) {
fields := make(map[string]interface{})
fields["value"] = value
ac.AddFields(measurement, fields, tags, t...)
}
func (ac *accumulator) AddFields(
2015-05-29 20:25:48 +00:00
measurement string,
fields map[string]interface{},
tags map[string]string,
t ...time.Time,
) {
if len(fields) == 0 || len(measurement) == 0 {
return
}
2016-01-07 20:39:43 +00:00
if !ac.inputConfig.Filter.ShouldTagsPass(tags) {
2015-12-11 20:07:32 +00:00
return
}
// Override measurement name if set
2016-01-07 20:39:43 +00:00
if len(ac.inputConfig.NameOverride) != 0 {
measurement = ac.inputConfig.NameOverride
2015-12-11 20:07:32 +00:00
}
// Apply measurement prefix and suffix if set
2016-01-07 20:39:43 +00:00
if len(ac.inputConfig.MeasurementPrefix) != 0 {
measurement = ac.inputConfig.MeasurementPrefix + measurement
2015-12-11 20:07:32 +00:00
}
2016-01-07 20:39:43 +00:00
if len(ac.inputConfig.MeasurementSuffix) != 0 {
measurement = measurement + ac.inputConfig.MeasurementSuffix
2015-12-11 20:07:32 +00:00
}
if tags == nil {
tags = make(map[string]string)
}
// Apply plugin-wide tags if set
2016-01-07 20:39:43 +00:00
for k, v := range ac.inputConfig.Tags {
2015-12-11 20:07:32 +00:00
if _, ok := tags[k]; !ok {
tags[k] = v
}
}
// Apply daemon-wide tags if set
for k, v := range ac.defaultTags {
if _, ok := tags[k]; !ok {
tags[k] = v
}
}
result := make(map[string]interface{})
for k, v := range fields {
2015-12-11 20:07:32 +00:00
// Filter out any filtered fields
2016-01-07 20:39:43 +00:00
if ac.inputConfig != nil {
if !ac.inputConfig.Filter.ShouldPass(k) {
2015-12-11 20:07:32 +00:00
continue
}
}
result[k] = v
// Validate uint64 and float64 fields
switch val := v.(type) {
case uint64:
// InfluxDB does not support writing uint64
if val < uint64(9223372036854775808) {
2015-12-11 20:07:32 +00:00
result[k] = int64(val)
} else {
2015-12-11 20:07:32 +00:00
result[k] = int64(9223372036854775807)
}
case float64:
// NaNs are invalid values in influxdb, skip measurement
if math.IsNaN(val) || math.IsInf(val, 0) {
if ac.debug {
2015-12-11 20:07:32 +00:00
log.Printf("Measurement [%s] field [%s] has a NaN or Inf "+
"field, skipping",
measurement, k)
}
2015-12-11 20:07:32 +00:00
continue
}
}
}
2015-12-11 20:07:32 +00:00
fields = nil
if len(result) == 0 {
return
}
var timestamp time.Time
if len(t) > 0 {
timestamp = t[0]
} else {
timestamp = time.Now()
}
if ac.prefix != "" {
measurement = ac.prefix + measurement
}
2015-12-11 20:07:32 +00:00
pt, err := client.NewPoint(measurement, tags, result, timestamp)
if err != nil {
log.Printf("Error adding point [%s]: %s\n", measurement, err.Error())
return
}
if ac.debug {
fmt.Println("> " + pt.String())
}
ac.points <- pt
}
func (ac *accumulator) SetDefaultTags(tags map[string]string) {
ac.defaultTags = tags
}
func (ac *accumulator) AddDefaultTag(key, value string) {
ac.defaultTags[key] = value
}
func (ac *accumulator) Prefix() string {
return ac.prefix
}
func (ac *accumulator) SetPrefix(prefix string) {
ac.prefix = prefix
}
func (ac *accumulator) Debug() bool {
return ac.debug
}
func (ac *accumulator) SetDebug(debug bool) {
ac.debug = debug
}