118 lines
2.5 KiB
Go
118 lines
2.5 KiB
Go
package agent
|
|
|
|
import (
|
|
"log"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
)
|
|
|
|
type MetricMaker interface {
|
|
Name() string
|
|
MakeMetric(
|
|
measurement string,
|
|
fields map[string]interface{},
|
|
tags map[string]string,
|
|
mType telegraf.ValueType,
|
|
t time.Time,
|
|
) telegraf.Metric
|
|
}
|
|
|
|
func NewAccumulator(
|
|
maker MetricMaker,
|
|
metrics chan telegraf.Metric,
|
|
) *accumulator {
|
|
acc := accumulator{
|
|
maker: maker,
|
|
metrics: metrics,
|
|
precision: time.Nanosecond,
|
|
}
|
|
return &acc
|
|
}
|
|
|
|
type accumulator struct {
|
|
metrics chan telegraf.Metric
|
|
|
|
maker MetricMaker
|
|
|
|
precision time.Duration
|
|
|
|
errCount uint64
|
|
}
|
|
|
|
func (ac *accumulator) AddFields(
|
|
measurement string,
|
|
fields map[string]interface{},
|
|
tags map[string]string,
|
|
t ...time.Time,
|
|
) {
|
|
if m := ac.maker.MakeMetric(measurement, fields, tags, telegraf.Untyped, ac.getTime(t)); m != nil {
|
|
ac.metrics <- m
|
|
}
|
|
}
|
|
|
|
func (ac *accumulator) AddGauge(
|
|
measurement string,
|
|
fields map[string]interface{},
|
|
tags map[string]string,
|
|
t ...time.Time,
|
|
) {
|
|
if m := ac.maker.MakeMetric(measurement, fields, tags, telegraf.Gauge, ac.getTime(t)); m != nil {
|
|
ac.metrics <- m
|
|
}
|
|
}
|
|
|
|
func (ac *accumulator) AddCounter(
|
|
measurement string,
|
|
fields map[string]interface{},
|
|
tags map[string]string,
|
|
t ...time.Time,
|
|
) {
|
|
if m := ac.maker.MakeMetric(measurement, fields, tags, telegraf.Counter, ac.getTime(t)); m != nil {
|
|
ac.metrics <- m
|
|
}
|
|
}
|
|
|
|
// AddError passes a runtime error to the accumulator.
|
|
// The error will be tagged with the plugin name and written to the log.
|
|
func (ac *accumulator) AddError(err error) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
atomic.AddUint64(&ac.errCount, 1)
|
|
//TODO suppress/throttle consecutive duplicate errors?
|
|
log.Printf("E! Error in plugin [%s]: %s", ac.maker.Name(), err)
|
|
}
|
|
|
|
// SetPrecision takes two time.Duration objects. If the first is non-zero,
|
|
// it sets that as the precision. Otherwise, it takes the second argument
|
|
// as the order of time that the metrics should be rounded to, with the
|
|
// maximum being 1s.
|
|
func (ac *accumulator) SetPrecision(precision, interval time.Duration) {
|
|
if precision > 0 {
|
|
ac.precision = precision
|
|
return
|
|
}
|
|
switch {
|
|
case interval >= time.Second:
|
|
ac.precision = time.Second
|
|
case interval >= time.Millisecond:
|
|
ac.precision = time.Millisecond
|
|
case interval >= time.Microsecond:
|
|
ac.precision = time.Microsecond
|
|
default:
|
|
ac.precision = time.Nanosecond
|
|
}
|
|
}
|
|
|
|
func (ac accumulator) getTime(t []time.Time) time.Time {
|
|
var timestamp time.Time
|
|
if len(t) > 0 {
|
|
timestamp = t[0]
|
|
} else {
|
|
timestamp = time.Now()
|
|
}
|
|
return timestamp.Round(ac.precision)
|
|
}
|