78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
package models
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/plugins/outputs"
|
||
|
|
|
||
|
|
"github.com/influxdata/influxdb/client/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
const DEFAULT_POINT_BUFFER_LIMIT = 10000
|
||
|
|
|
||
|
|
type RunningOutput struct {
|
||
|
|
Name string
|
||
|
|
Output outputs.Output
|
||
|
|
Config *OutputConfig
|
||
|
|
Quiet bool
|
||
|
|
PointBufferLimit int
|
||
|
|
|
||
|
|
points []*client.Point
|
||
|
|
overwriteCounter int
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRunningOutput(
|
||
|
|
name string,
|
||
|
|
output outputs.Output,
|
||
|
|
conf *OutputConfig,
|
||
|
|
) *RunningOutput {
|
||
|
|
ro := &RunningOutput{
|
||
|
|
Name: name,
|
||
|
|
points: make([]*client.Point, 0),
|
||
|
|
Output: output,
|
||
|
|
Config: conf,
|
||
|
|
PointBufferLimit: DEFAULT_POINT_BUFFER_LIMIT,
|
||
|
|
}
|
||
|
|
return ro
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ro *RunningOutput) AddPoint(point *client.Point) {
|
||
|
|
if ro.Config.Filter.IsActive {
|
||
|
|
if !ro.Config.Filter.ShouldPointPass(point) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(ro.points) < ro.PointBufferLimit {
|
||
|
|
ro.points = append(ro.points, point)
|
||
|
|
} else {
|
||
|
|
if ro.overwriteCounter == len(ro.points) {
|
||
|
|
ro.overwriteCounter = 0
|
||
|
|
}
|
||
|
|
ro.points[ro.overwriteCounter] = point
|
||
|
|
ro.overwriteCounter++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ro *RunningOutput) Write() error {
|
||
|
|
start := time.Now()
|
||
|
|
err := ro.Output.Write(ro.points)
|
||
|
|
elapsed := time.Since(start)
|
||
|
|
if err == nil {
|
||
|
|
if !ro.Quiet {
|
||
|
|
log.Printf("Wrote %d metrics to output %s in %s\n",
|
||
|
|
len(ro.points), ro.Name, elapsed)
|
||
|
|
}
|
||
|
|
ro.points = make([]*client.Point, 0)
|
||
|
|
ro.overwriteCounter = 0
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// OutputConfig containing name and filter
|
||
|
|
type OutputConfig struct {
|
||
|
|
Name string
|
||
|
|
Filter Filter
|
||
|
|
}
|