62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
|
package system
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf"
|
||
|
|
"github.com/influxdata/telegraf/plugins/inputs"
|
||
|
|
"github.com/shirou/gopsutil/process"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Processes struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func (_ *Processes) Description() string {
|
||
|
|
return "Get the number of processes and group them by status (Linux only)"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (_ *Processes) SampleConfig() string { return "" }
|
||
|
|
|
||
|
|
func (s *Processes) Gather(acc telegraf.Accumulator) error {
|
||
|
|
pids, err := process.Pids()
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("error getting pids list: %s", err)
|
||
|
|
}
|
||
|
|
// TODO handle other OS (Windows/BSD/Solaris/OSX)
|
||
|
|
fields := map[string]interface{}{
|
||
|
|
"paging": uint64(0),
|
||
|
|
"blocked": uint64(0),
|
||
|
|
"zombie": uint64(0),
|
||
|
|
"stopped": uint64(0),
|
||
|
|
"running": uint64(0),
|
||
|
|
"sleeping": uint64(0),
|
||
|
|
}
|
||
|
|
for _, pid := range pids {
|
||
|
|
process, err := process.NewProcess(pid)
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("Can not get process %d status: %s", pid, err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
status, err := process.Status()
|
||
|
|
if err != nil {
|
||
|
|
log.Printf("Can not get process %d status: %s\n", pid, err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
_, exists := fields[status]
|
||
|
|
if !exists {
|
||
|
|
log.Printf("Status '%s' for process with pid: %d\n", status, pid)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
fields[status] = fields[status].(uint64) + uint64(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
acc.AddFields("processes", fields, nil)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func init() {
|
||
|
|
inputs.Add("processes", func() telegraf.Input {
|
||
|
|
return &Processes{}
|
||
|
|
})
|
||
|
|
}
|