70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package models
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
)
|
|
|
|
type RunningProcessor struct {
|
|
Name string
|
|
|
|
sync.Mutex
|
|
Processor telegraf.Processor
|
|
Config *ProcessorConfig
|
|
}
|
|
|
|
type RunningProcessors []*RunningProcessor
|
|
|
|
func (rp RunningProcessors) Len() int { return len(rp) }
|
|
func (rp RunningProcessors) Swap(i, j int) { rp[i], rp[j] = rp[j], rp[i] }
|
|
func (rp RunningProcessors) Less(i, j int) bool { return rp[i].Config.Order < rp[j].Config.Order }
|
|
|
|
// FilterConfig containing a name and filter
|
|
type ProcessorConfig struct {
|
|
Name string
|
|
Order int64
|
|
Filter Filter
|
|
}
|
|
|
|
func (rp *RunningProcessor) metricFiltered(metric telegraf.Metric) {
|
|
metric.Drop()
|
|
}
|
|
|
|
func containsMetric(item telegraf.Metric, metrics []telegraf.Metric) bool {
|
|
for _, m := range metrics {
|
|
if item == m {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (rp *RunningProcessor) Apply(in ...telegraf.Metric) []telegraf.Metric {
|
|
rp.Lock()
|
|
defer rp.Unlock()
|
|
|
|
ret := []telegraf.Metric{}
|
|
|
|
for _, metric := range in {
|
|
// In processors when a filter selects a metric it is sent through the
|
|
// processor. Otherwise the metric continues downstream unmodified.
|
|
if ok := rp.Config.Filter.Select(metric); !ok {
|
|
ret = append(ret, metric)
|
|
continue
|
|
}
|
|
|
|
rp.Config.Filter.Modify(metric)
|
|
if len(metric.FieldList()) == 0 {
|
|
rp.metricFiltered(metric)
|
|
continue
|
|
}
|
|
|
|
// This metric should pass through the filter, so call the filter Apply
|
|
// function and append results to the output slice.
|
|
ret = append(ret, rp.Processor.Apply(metric)...)
|
|
}
|
|
|
|
return ret
|
|
}
|