telegraf/plugins/inputs/execd/execd_posix.go

45 lines
844 B
Go
Raw Normal View History

2020-02-29 02:46:03 +08:00
// +build !windows
package execd
import (
"fmt"
"io"
"os"
2020-02-29 02:46:03 +08:00
"syscall"
"time"
2020-02-29 02:46:03 +08:00
"github.com/influxdata/telegraf"
)
func (e *Execd) Gather(_ telegraf.Accumulator) error {
2020-06-05 07:09:22 +08:00
if e.process == nil || e.process.Cmd == nil {
2020-02-29 02:46:03 +08:00
return nil
}
2020-06-05 07:09:22 +08:00
osProcess := e.process.Cmd.Process
if osProcess == nil {
return nil
}
2020-02-29 02:46:03 +08:00
switch e.Signal {
case "SIGHUP":
2020-06-05 07:09:22 +08:00
osProcess.Signal(syscall.SIGHUP)
2020-02-29 02:46:03 +08:00
case "SIGUSR1":
2020-06-05 07:09:22 +08:00
osProcess.Signal(syscall.SIGUSR1)
2020-02-29 02:46:03 +08:00
case "SIGUSR2":
2020-06-05 07:09:22 +08:00
osProcess.Signal(syscall.SIGUSR2)
2020-02-29 02:46:03 +08:00
case "STDIN":
2020-06-05 07:09:22 +08:00
if osStdin, ok := e.process.Stdin.(*os.File); ok {
osStdin.SetWriteDeadline(time.Now().Add(1 * time.Second))
}
2020-06-05 07:09:22 +08:00
if _, err := io.WriteString(e.process.Stdin, "\n"); err != nil {
return fmt.Errorf("Error writing to stdin: %s", err)
}
2020-02-29 02:46:03 +08:00
case "none":
default:
return fmt.Errorf("invalid signal: %s", e.Signal)
}
return nil
}