58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
//go:generate ../../../tools/readme_config_includer/generator
|
|
package template
|
|
|
|
import (
|
|
_ "embed"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/plugins/processors"
|
|
)
|
|
|
|
//go:embed sample.conf
|
|
var sampleConfig string
|
|
|
|
type TemplateProcessor struct {
|
|
Tag string `toml:"tag"`
|
|
Template string `toml:"template"`
|
|
Log telegraf.Logger `toml:"-"`
|
|
tmpl *template.Template
|
|
}
|
|
|
|
func (*TemplateProcessor) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (r *TemplateProcessor) Apply(in ...telegraf.Metric) []telegraf.Metric {
|
|
// for each metric in "in" array
|
|
for _, metric := range in {
|
|
var b strings.Builder
|
|
newM := TemplateMetric{metric}
|
|
|
|
// supply TemplateMetric and Template from configuration to Template.Execute
|
|
err := r.tmpl.Execute(&b, &newM)
|
|
if err != nil {
|
|
r.Log.Errorf("failed to execute template: %v", err)
|
|
continue
|
|
}
|
|
|
|
metric.AddTag(r.Tag, b.String())
|
|
}
|
|
return in
|
|
}
|
|
|
|
func (r *TemplateProcessor) Init() error {
|
|
// create template
|
|
t, err := template.New("configured_template").Parse(r.Template)
|
|
|
|
r.tmpl = t
|
|
return err
|
|
}
|
|
|
|
func init() {
|
|
processors.Add("template", func() telegraf.Processor {
|
|
return &TemplateProcessor{}
|
|
})
|
|
}
|