Adding upstream version 1.34.4.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
e393c3af3f
commit
4978089aab
4963 changed files with 677545 additions and 0 deletions
116
plugins/inputs/multifile/multifile.go
Normal file
116
plugins/inputs/multifile/multifile.go
Normal file
|
@ -0,0 +1,116 @@
|
|||
//go:generate ../../../tools/readme_config_includer/generator
|
||||
package multifile
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"math"
|
||||
"os"
|
||||
"path"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/plugins/inputs"
|
||||
)
|
||||
|
||||
//go:embed sample.conf
|
||||
var sampleConfig string
|
||||
|
||||
type MultiFile struct {
|
||||
BaseDir string `toml:"base_dir"`
|
||||
FailEarly bool `toml:"fail_early"`
|
||||
Files []file `toml:"file"`
|
||||
}
|
||||
|
||||
type file struct {
|
||||
Name string `toml:"file"`
|
||||
Dest string `toml:"dest"`
|
||||
Conversion string `toml:"conversion"`
|
||||
}
|
||||
|
||||
func (*MultiFile) SampleConfig() string {
|
||||
return sampleConfig
|
||||
}
|
||||
|
||||
func (m *MultiFile) Init() error {
|
||||
for i, file := range m.Files {
|
||||
if m.BaseDir != "" {
|
||||
m.Files[i].Name = path.Join(m.BaseDir, file.Name)
|
||||
}
|
||||
if file.Dest == "" {
|
||||
m.Files[i].Dest = path.Base(file.Name)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *MultiFile) Gather(acc telegraf.Accumulator) error {
|
||||
now := time.Now()
|
||||
fields := make(map[string]interface{})
|
||||
tags := make(map[string]string)
|
||||
|
||||
for _, file := range m.Files {
|
||||
fileContents, err := os.ReadFile(file.Name)
|
||||
|
||||
if err != nil {
|
||||
if m.FailEarly {
|
||||
return err
|
||||
}
|
||||
continue
|
||||
}
|
||||
|
||||
vStr := string(bytes.TrimSpace(bytes.Trim(fileContents, "\x00")))
|
||||
|
||||
if file.Conversion == "tag" {
|
||||
tags[file.Dest] = vStr
|
||||
continue
|
||||
}
|
||||
|
||||
var value interface{}
|
||||
|
||||
var d int
|
||||
if _, errfmt := fmt.Sscanf(file.Conversion, "float(%d)", &d); errfmt == nil || file.Conversion == "float" {
|
||||
var v float64
|
||||
v, err = strconv.ParseFloat(vStr, 64)
|
||||
value = v / math.Pow10(d)
|
||||
}
|
||||
|
||||
if file.Conversion == "int" {
|
||||
value, err = strconv.ParseInt(vStr, 10, 64)
|
||||
}
|
||||
|
||||
if file.Conversion == "string" || file.Conversion == "" {
|
||||
value = vStr
|
||||
}
|
||||
|
||||
if file.Conversion == "bool" {
|
||||
value, err = strconv.ParseBool(vStr)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
if m.FailEarly {
|
||||
return err
|
||||
}
|
||||
continue
|
||||
}
|
||||
|
||||
if value == nil {
|
||||
return fmt.Errorf("invalid conversion %v", file.Conversion)
|
||||
}
|
||||
|
||||
fields[file.Dest] = value
|
||||
}
|
||||
|
||||
acc.AddGauge("multifile", fields, tags, now)
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
inputs.Add("multifile", func() telegraf.Input {
|
||||
return &MultiFile{
|
||||
FailEarly: true,
|
||||
}
|
||||
})
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue