104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package file
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/internal/globpath"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
"github.com/influxdata/telegraf/plugins/parsers"
|
|
)
|
|
|
|
type File struct {
|
|
Files []string `toml:"files"`
|
|
FileTag string `toml:"file_tag"`
|
|
parser parsers.Parser
|
|
|
|
filenames []string
|
|
}
|
|
|
|
const sampleConfig = `
|
|
## Files to parse each interval. Accept standard unix glob matching rules,
|
|
## as well as ** to match recursive files and directories.
|
|
files = ["/tmp/metrics.out"]
|
|
|
|
## The dataformat to be read from files
|
|
## Each data format has its own unique set of configuration options, read
|
|
## more about them here:
|
|
## https://github.com/influxdata/telegraf/blob/master/docs/DATA_FORMATS_INPUT.md
|
|
data_format = "influx"
|
|
|
|
## Name a tag containing the name of the file the data was parsed from. Leave empty
|
|
## to disable.
|
|
# file_tag = ""
|
|
`
|
|
|
|
// SampleConfig returns the default configuration of the Input
|
|
func (f *File) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (f *File) Description() string {
|
|
return "Parse a complete file each interval"
|
|
}
|
|
|
|
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))
|
|
}
|
|
acc.AddFields(m.Name(), m.Fields(), m.Tags(), m.Time())
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (f *File) SetParser(p parsers.Parser) {
|
|
f.parser = p
|
|
}
|
|
|
|
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 %v: %v", file, err)
|
|
}
|
|
files := g.Match()
|
|
if len(files) <= 0 {
|
|
return fmt.Errorf("could not find file: %v", file)
|
|
}
|
|
allFiles = append(allFiles, files...)
|
|
}
|
|
|
|
f.filenames = allFiles
|
|
return nil
|
|
}
|
|
|
|
func (f *File) readMetric(filename string) ([]telegraf.Metric, error) {
|
|
fileContents, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("E! Error file: %v could not be read, %s", filename, err)
|
|
}
|
|
return f.parser.Parse(fileContents)
|
|
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("file", func() telegraf.Input {
|
|
return &File{}
|
|
})
|
|
}
|