From 47004c7ad73d5235a93257428288f24402c7d6df Mon Sep 17 00:00:00 2001
From: Thibault Cohen <titilambert@gmail.com>
Date: Sun, 6 Mar 2016 01:04:54 -0500
Subject: [PATCH] Add processes status stats in system input plugin

---
 plugins/inputs/system/processes.go      | 61 +++++++++++++++++++++++++
 plugins/inputs/system/processes_test.go | 21 +++++++++
 2 files changed, 82 insertions(+)
 create mode 100644 plugins/inputs/system/processes.go
 create mode 100644 plugins/inputs/system/processes_test.go

diff --git a/plugins/inputs/system/processes.go b/plugins/inputs/system/processes.go
new file mode 100644
index 000000000..c4b791e3c
--- /dev/null
+++ b/plugins/inputs/system/processes.go
@@ -0,0 +1,61 @@
+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{}
+	})
+}
diff --git a/plugins/inputs/system/processes_test.go b/plugins/inputs/system/processes_test.go
new file mode 100644
index 000000000..246884711
--- /dev/null
+++ b/plugins/inputs/system/processes_test.go
@@ -0,0 +1,21 @@
+package system
+
+import (
+	"testing"
+
+	"github.com/influxdata/telegraf/testutil"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestProcesses(t *testing.T) {
+	processes := &Processes{}
+	var acc testutil.Accumulator
+
+	err := processes.Gather(&acc)
+	require.NoError(t, err)
+
+	assert.True(t, acc.HasUIntField("processes", "running"))
+	assert.True(t, acc.HasUIntField("processes", "sleeping"))
+	assert.True(t, acc.HasUIntField("processes", "stopped"))
+}