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
129
plugins/inputs/file/file.go
Normal file
129
plugins/inputs/file/file.go
Normal file
|
@ -0,0 +1,129 @@
|
|||
//go:generate ../../../tools/readme_config_includer/generator
|
||||
package file
|
||||
|
||||
import (
|
||||
_ "embed"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
|
||||
"github.com/dimchansky/utfbom"
|
||||
|
||||
"github.com/influxdata/telegraf"
|
||||
"github.com/influxdata/telegraf/internal"
|
||||
"github.com/influxdata/telegraf/internal/globpath"
|
||||
"github.com/influxdata/telegraf/plugins/common/encoding"
|
||||
"github.com/influxdata/telegraf/plugins/inputs"
|
||||
)
|
||||
|
||||
//go:embed sample.conf
|
||||
var sampleConfig string
|
||||
|
||||
var once sync.Once
|
||||
|
||||
type File struct {
|
||||
Files []string `toml:"files"`
|
||||
FileTag string `toml:"file_tag"`
|
||||
FilePathTag string `toml:"file_path_tag"`
|
||||
CharacterEncoding string `toml:"character_encoding"`
|
||||
Log telegraf.Logger `toml:"-"`
|
||||
|
||||
parserFunc telegraf.ParserFunc
|
||||
filenames []string
|
||||
decoder *encoding.Decoder
|
||||
}
|
||||
|
||||
func (*File) SampleConfig() string {
|
||||
return sampleConfig
|
||||
}
|
||||
|
||||
func (f *File) Init() error {
|
||||
var err error
|
||||
f.decoder, err = encoding.NewDecoder(f.CharacterEncoding)
|
||||
return err
|
||||
}
|
||||
|
||||
func (f *File) SetParserFunc(fn telegraf.ParserFunc) {
|
||||
f.parserFunc = fn
|
||||
}
|
||||
|
||||
func (f *File) Gather(acc telegraf.Accumulator) error {
|
||||
err := f.refreshFilePaths()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, k := range f.filenames {
|
||||
metrics, err := f.readMetric(k)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, m := range metrics {
|
||||
if f.FileTag != "" {
|
||||
m.AddTag(f.FileTag, filepath.Base(k))
|
||||
}
|
||||
if f.FilePathTag != "" {
|
||||
if absPath, err := filepath.Abs(k); err == nil {
|
||||
m.AddTag(f.FilePathTag, absPath)
|
||||
}
|
||||
}
|
||||
acc.AddMetric(m)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *File) refreshFilePaths() error {
|
||||
var allFiles []string
|
||||
for _, file := range f.Files {
|
||||
g, err := globpath.Compile(file)
|
||||
if err != nil {
|
||||
return fmt.Errorf("could not compile glob %q: %w", file, err)
|
||||
}
|
||||
files := g.Match()
|
||||
if len(files) == 0 {
|
||||
return fmt.Errorf("could not find file(s): %v", file)
|
||||
}
|
||||
allFiles = append(allFiles, files...)
|
||||
}
|
||||
|
||||
f.filenames = allFiles
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *File) readMetric(filename string) ([]telegraf.Metric, error) {
|
||||
file, err := os.Open(filename)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
r, _ := utfbom.Skip(f.decoder.Reader(file))
|
||||
fileContents, err := io.ReadAll(r)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("could not read %q: %w", filename, err)
|
||||
}
|
||||
parser, err := f.parserFunc()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("could not instantiate parser: %w", err)
|
||||
}
|
||||
metrics, err := parser.Parse(fileContents)
|
||||
if err != nil {
|
||||
return metrics, fmt.Errorf("could not parse %q: %w", filename, err)
|
||||
}
|
||||
|
||||
if len(metrics) == 0 {
|
||||
once.Do(func() {
|
||||
f.Log.Debug(internal.NoMetricsCreatedMsg)
|
||||
})
|
||||
}
|
||||
return metrics, err
|
||||
}
|
||||
|
||||
func init() {
|
||||
inputs.Add("file", func() telegraf.Input {
|
||||
return &File{}
|
||||
})
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue