109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
|
package valuecounter
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf"
|
||
|
|
"github.com/influxdata/telegraf/plugins/aggregators"
|
||
|
|
)
|
||
|
|
|
||
|
|
type aggregate struct {
|
||
|
|
name string
|
||
|
|
tags map[string]string
|
||
|
|
fieldCount map[string]int
|
||
|
|
}
|
||
|
|
|
||
|
|
// ValueCounter an aggregation plugin
|
||
|
|
type ValueCounter struct {
|
||
|
|
cache map[uint64]aggregate
|
||
|
|
Fields []string
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewValueCounter create a new aggregation plugin which counts the occurances
|
||
|
|
// of fields and emits the count.
|
||
|
|
func NewValueCounter() telegraf.Aggregator {
|
||
|
|
vc := &ValueCounter{}
|
||
|
|
vc.Reset()
|
||
|
|
return vc
|
||
|
|
}
|
||
|
|
|
||
|
|
var sampleConfig = `
|
||
|
|
## General Aggregator Arguments:
|
||
|
|
## The period on which to flush & clear the aggregator.
|
||
|
|
period = "30s"
|
||
|
|
## If true, the original metric will be dropped by the
|
||
|
|
## aggregator and will not get sent to the output plugins.
|
||
|
|
drop_original = false
|
||
|
|
## The fields for which the values will be counted
|
||
|
|
fields = []
|
||
|
|
`
|
||
|
|
|
||
|
|
// SampleConfig generates a sample config for the ValueCounter plugin
|
||
|
|
func (vc *ValueCounter) SampleConfig() string {
|
||
|
|
return sampleConfig
|
||
|
|
}
|
||
|
|
|
||
|
|
// Description returns the description of the ValueCounter plugin
|
||
|
|
func (vc *ValueCounter) Description() string {
|
||
|
|
return "Count the occurance of values in fields."
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add is run on every metric which passes the plugin
|
||
|
|
func (vc *ValueCounter) Add(in telegraf.Metric) {
|
||
|
|
id := in.HashID()
|
||
|
|
|
||
|
|
// Check if the cache already has an entry for this metric, if not create it
|
||
|
|
if _, ok := vc.cache[id]; !ok {
|
||
|
|
a := aggregate{
|
||
|
|
name: in.Name(),
|
||
|
|
tags: in.Tags(),
|
||
|
|
fieldCount: make(map[string]int),
|
||
|
|
}
|
||
|
|
vc.cache[id] = a
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check if this metric has fields which we need to count, if so increment
|
||
|
|
// the count.
|
||
|
|
for fk, fv := range in.Fields() {
|
||
|
|
for _, cf := range vc.Fields {
|
||
|
|
if fk == cf {
|
||
|
|
// Do not process float types to prevent memory from blowing up
|
||
|
|
switch fv.(type) {
|
||
|
|
default:
|
||
|
|
log.Printf("I! Valuecounter: Unsupported field type. " +
|
||
|
|
"Must be an int, string or bool. Ignoring.")
|
||
|
|
continue
|
||
|
|
case uint64, int64, string, bool:
|
||
|
|
}
|
||
|
|
fn := fmt.Sprintf("%v_%v", fk, fv)
|
||
|
|
vc.cache[id].fieldCount[fn]++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Push emits the counters
|
||
|
|
func (vc *ValueCounter) Push(acc telegraf.Accumulator) {
|
||
|
|
for _, agg := range vc.cache {
|
||
|
|
fields := map[string]interface{}{}
|
||
|
|
|
||
|
|
for field, count := range agg.fieldCount {
|
||
|
|
fields[field] = count
|
||
|
|
}
|
||
|
|
|
||
|
|
acc.AddFields(agg.name, fields, agg.tags)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Reset the cache, executed after each push
|
||
|
|
func (vc *ValueCounter) Reset() {
|
||
|
|
vc.cache = make(map[uint64]aggregate)
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
aggregators.Add("valuecounter", func() telegraf.Aggregator {
|
||
|
|
return NewValueCounter()
|
||
|
|
})
|
||
|
|
}
|