package file import ( "fmt" "io" "log" "os" "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/internal" "github.com/influxdata/telegraf/internal/rotate" "github.com/influxdata/telegraf/plugins/outputs" "github.com/influxdata/telegraf/plugins/serializers" ) type File struct { Files []string `toml:"files"` RotationInterval internal.Duration `toml:"rotation_interval"` RotationMaxSize internal.Size `toml:"rotation_max_size"` RotationMaxArchives int `toml:"rotation_max_archives"` writer io.Writer closers []io.Closer serializer serializers.Serializer } var sampleConfig = ` ## Files to write to, "stdout" is a specially handled file. files = ["stdout", "/tmp/metrics.out"] ## The file will be rotated after the time interval specified. When set ## to 0 no time based rotation is performed. # rotation_interval = "0d" ## The logfile will be rotated when it becomes larger than the specified ## size. When set to 0 no size based rotation is performed. # rotation_max_size = "0MB" ## Maximum number of rotated archives to keep, any older logs are deleted. ## If set to -1, no archives are removed. # rotation_max_archives = 5 ## Data format to output. ## 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_OUTPUT.md data_format = "influx" ` func (f *File) SetSerializer(serializer serializers.Serializer) { f.serializer = serializer } func (f *File) Connect() error { writers := []io.Writer{} if len(f.Files) == 0 { f.Files = []string{"stdout"} } for _, file := range f.Files { if file == "stdout" { writers = append(writers, os.Stdout) } else { of, err := rotate.NewFileWriter( file, f.RotationInterval.Duration, f.RotationMaxSize.Size, f.RotationMaxArchives) if err != nil { return err } writers = append(writers, of) f.closers = append(f.closers, of) } } f.writer = io.MultiWriter(writers...) return nil } func (f *File) Close() error { var err error for _, c := range f.closers { errClose := c.Close() if errClose != nil { err = errClose } } return err } func (f *File) SampleConfig() string { return sampleConfig } func (f *File) Description() string { return "Send telegraf metrics to file(s)" } func (f *File) Write(metrics []telegraf.Metric) error { var writeErr error = nil for _, metric := range metrics { b, err := f.serializer.Serialize(metric) if err != nil { log.Printf("D! [outputs.file] Could not serialize metric: %v", err) } _, err = f.writer.Write(b) if err != nil { writeErr = fmt.Errorf("E! [outputs.file] failed to write message: %v", err) } } return writeErr } func init() { outputs.Add("file", func() telegraf.Output { return &File{} }) }