120 lines
2.3 KiB
Go
120 lines
2.3 KiB
Go
package filestat
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/internal/globpath"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
)
|
|
|
|
const sampleConfig = `
|
|
## Files to gather stats about.
|
|
## These accept standard unix glob matching rules, but with the addition of
|
|
## ** as a "super asterisk". See https://github.com/gobwas/glob.
|
|
files = ["/etc/telegraf/telegraf.conf", "/var/log/**.log"]
|
|
## If true, read the entire file and calculate an md5 checksum.
|
|
md5 = false
|
|
`
|
|
|
|
type FileStat struct {
|
|
Md5 bool
|
|
Files []string
|
|
|
|
// maps full file paths to globmatch obj
|
|
globs map[string]*globpath.GlobPath
|
|
}
|
|
|
|
func NewFileStat() *FileStat {
|
|
return &FileStat{
|
|
globs: make(map[string]*globpath.GlobPath),
|
|
}
|
|
}
|
|
|
|
func (_ *FileStat) Description() string {
|
|
return "Read stats about given file(s)"
|
|
}
|
|
|
|
func (_ *FileStat) SampleConfig() string { return sampleConfig }
|
|
|
|
func (f *FileStat) Gather(acc telegraf.Accumulator) error {
|
|
var errS string
|
|
var err error
|
|
|
|
for _, filepath := range f.Files {
|
|
// Get the compiled glob object for this filepath
|
|
g, ok := f.globs[filepath]
|
|
if !ok {
|
|
if g, err = globpath.Compile(filepath); err != nil {
|
|
errS += err.Error() + " "
|
|
continue
|
|
}
|
|
f.globs[filepath] = g
|
|
}
|
|
|
|
files := g.Match()
|
|
if len(files) == 0 {
|
|
acc.AddFields("filestat",
|
|
map[string]interface{}{
|
|
"exists": int64(0),
|
|
},
|
|
map[string]string{
|
|
"file": filepath,
|
|
})
|
|
continue
|
|
}
|
|
|
|
for fileName, fileInfo := range files {
|
|
tags := map[string]string{
|
|
"file": fileName,
|
|
}
|
|
fields := map[string]interface{}{
|
|
"exists": int64(1),
|
|
"size_bytes": fileInfo.Size(),
|
|
}
|
|
|
|
if f.Md5 {
|
|
md5, err := getMd5(fileName)
|
|
if err != nil {
|
|
errS += err.Error() + " "
|
|
} else {
|
|
fields["md5_sum"] = md5
|
|
}
|
|
}
|
|
|
|
acc.AddFields("filestat", fields, tags)
|
|
}
|
|
}
|
|
|
|
if errS != "" {
|
|
return fmt.Errorf(errS)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Read given file and calculate an md5 hash.
|
|
func getMd5(file string) (string, error) {
|
|
of, err := os.Open(file)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer of.Close()
|
|
|
|
hash := md5.New()
|
|
_, err = io.Copy(hash, of)
|
|
if err != nil {
|
|
// fatal error
|
|
return "", err
|
|
}
|
|
return fmt.Sprintf("%x", hash.Sum(nil)), nil
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("filestat", func() telegraf.Input {
|
|
return NewFileStat()
|
|
})
|
|
}
|