134 lines
2.6 KiB
Go
134 lines
2.6 KiB
Go
package exec
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os/exec"
|
|
"sync"
|
|
|
|
"github.com/gonuts/go-shellquote"
|
|
|
|
"github.com/influxdata/telegraf"
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
|
"github.com/influxdata/telegraf/plugins/parsers"
|
|
)
|
|
|
|
const sampleConfig = `
|
|
### Commands array
|
|
commands = ["/tmp/test.sh", "/usr/bin/mycollector --foo=bar"]
|
|
|
|
### measurement name suffix (for separating different commands)
|
|
name_suffix = "_mycollector"
|
|
|
|
### Data format to consume. This can be "json", "influx" or "graphite"
|
|
### Each data format has it's 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"
|
|
`
|
|
|
|
type Exec struct {
|
|
Commands []string
|
|
Command string
|
|
|
|
parser parsers.Parser
|
|
|
|
wg sync.WaitGroup
|
|
|
|
runner Runner
|
|
errChan chan error
|
|
}
|
|
|
|
func NewExec() *Exec {
|
|
return &Exec{
|
|
runner: CommandRunner{},
|
|
}
|
|
}
|
|
|
|
type Runner interface {
|
|
Run(*Exec, string) ([]byte, error)
|
|
}
|
|
|
|
type CommandRunner struct{}
|
|
|
|
func (c CommandRunner) Run(e *Exec, command string) ([]byte, error) {
|
|
split_cmd, err := shellquote.Split(command)
|
|
if err != nil || len(split_cmd) == 0 {
|
|
return nil, fmt.Errorf("exec: unable to parse command, %s", err)
|
|
}
|
|
|
|
cmd := exec.Command(split_cmd[0], split_cmd[1:]...)
|
|
|
|
var out bytes.Buffer
|
|
cmd.Stdout = &out
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
return nil, fmt.Errorf("exec: %s for command '%s'", err, command)
|
|
}
|
|
|
|
return out.Bytes(), nil
|
|
}
|
|
|
|
func (e *Exec) ProcessCommand(command string, acc telegraf.Accumulator) {
|
|
defer e.wg.Done()
|
|
|
|
out, err := e.runner.Run(e, command)
|
|
if err != nil {
|
|
e.errChan <- err
|
|
return
|
|
}
|
|
|
|
metrics, err := e.parser.Parse(out)
|
|
if err != nil {
|
|
e.errChan <- err
|
|
} else {
|
|
for _, metric := range metrics {
|
|
acc.AddFields(metric.Name(), metric.Fields(), metric.Tags(), metric.Time())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (e *Exec) SampleConfig() string {
|
|
return sampleConfig
|
|
}
|
|
|
|
func (e *Exec) Description() string {
|
|
return "Read metrics from one or more commands that can output to stdout"
|
|
}
|
|
|
|
func (e *Exec) SetParser(parser parsers.Parser) {
|
|
e.parser = parser
|
|
}
|
|
|
|
func (e *Exec) Gather(acc telegraf.Accumulator) error {
|
|
// Legacy single command support
|
|
if e.Command != "" {
|
|
e.Commands = append(e.Commands, e.Command)
|
|
e.Command = ""
|
|
}
|
|
|
|
e.errChan = make(chan error, len(e.Commands))
|
|
|
|
e.wg.Add(len(e.Commands))
|
|
for _, command := range e.Commands {
|
|
go e.ProcessCommand(command, acc)
|
|
}
|
|
e.wg.Wait()
|
|
|
|
select {
|
|
default:
|
|
close(e.errChan)
|
|
return nil
|
|
case err := <-e.errChan:
|
|
close(e.errChan)
|
|
return err
|
|
}
|
|
|
|
}
|
|
|
|
func init() {
|
|
inputs.Add("exec", func() telegraf.Input {
|
|
return NewExec()
|
|
})
|
|
}
|