added custom inspector, will probably have no chart only output

Author: deven96

driver/ssh.go

@@ -2,6 +2,7 @@ package driver
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/melbahja/goph"
 	log "github.com/sirupsen/logrus"
@@ -25,6 +26,8 @@ type SSH struct {
 	KeyPass string
 	// Check known hosts (only disable for tests
 	CheckKnownHosts bool
+	// set environmental vars for server e.g []string{"DEBUG=1", "FAKE=echo"}
+	Vars []string
 }
 
 func (d *SSH) String() string {
@@ -76,6 +79,11 @@ func (d *SSH) RunCommand(command string) (string, error) {
 		return ``, err
 	}
 	defer client.Close()
+	if len(d.Vars) != 0 {
+		// add env variable to command
+		envline := strings.Join(d.Vars, ";")
+		command = strings.Join([]string{envline, command}, ";")
+	}
 	out, err := client.Run(command)
 	if err != nil {
 		return ``, err

inspector/custom.go

@@ -0,0 +1,39 @@
+package inspector
+
+import (
+	log "github.com/sirupsen/logrus"
+)
+
+// CustomMetrics : Metrics used by DF
+type CustomMetrics struct {
+	Output string
+}
+
+// Custom : Parsing the `df` output for disk monitoring
+type Custom struct {
+	fields
+	Values CustomMetrics
+}
+
+// Parse : run custom parsing on output of the command
+func (i *Custom) Parse(output string) {
+	log.Debug("Parsing ouput string in Custom inspector")
+	i.Values = i.createMetric(output)
+}
+
+func (i Custom) createMetric(output string) CustomMetrics {
+	return CustomMetrics{
+		Output: output,
+	}
+}
+
+// NewCustom : Initialize a new Custom instance
+func NewCustom(custom string) *Custom {
+	return &Custom{
+		fields: fields{
+			Type:    Command,
+			Command: custom,
+		},
+	}
+
+}

inspector/process.go

@@ -1 +1,93 @@
 package inspector
+
+import (
+	log "github.com/sirupsen/logrus"
+	"strconv"
+	"strings"
+)
+
+// ProcessMetrics : Metrics used by Process
+type ProcessMetrics struct {
+	Command string
+	User    string
+	Pid     int
+	// Percentage value of CPU used
+	CPU float64
+	// Percentage value of memory used
+	Memory float64
+	// Number of seconds the process has been running
+	Time int64
+	TTY  string
+}
+
+// Process : Parsing the `ps -A u` output for process monitoring
+type Process struct {
+	fields
+	// Track this particular PID
+	TrackPID int
+	// Values of metrics being read
+	Values []ProcessMetrics
+}
+
+// Parse : run custom parsing on output of the command
+func (i *Process) Parse(output string) {
+	var values []ProcessMetrics
+	lines := strings.Split(output, "\n")
+	for index, line := range lines {
+		// skip title line
+		if index == 0 {
+			continue
+		}
+		columns := strings.Fields(line)
+		if len(columns) >= 10 {
+			pid, err := strconv.Atoi(columns[1])
+			if err != nil {
+				log.Fatal("Could not parse pid in Process")
+			}
+			// If we are tracking only a particular ID then break loop
+			if i.TrackPID != 0 && i.TrackPID == pid {
+				value := i.createMetric(columns, pid)
+				values = append(values, value)
+				break
+			} else if i.TrackPID == 0 {
+				value := i.createMetric(columns, pid)
+				values = append(values, value)
+			}
+		}
+	}
+	i.Values = values
+}
+
+func (i Process) createMetric(columns []string, pid int) ProcessMetrics {
+	var parseErr error
+	cpu, parseErr := strconv.ParseFloat(columns[2], 64)
+	mem, parseErr := strconv.ParseFloat(columns[3], 64)
+	unparsedTime := columns[9]
+	tty := columns[6]
+	minutesStr := strings.Split(unparsedTime, ":")
+	minute, parseErr := strconv.Atoi(minutesStr[0])
+	second, parseErr := strconv.Atoi(minutesStr[1])
+	if parseErr != nil {
+		log.Fatal(parseErr)
+	}
+
+	return ProcessMetrics{
+		Command: strings.Join(columns[10:], " "),
+		User:    columns[0],
+		CPU:     cpu,
+		Memory:  mem,
+		Time:    int64((minute * 60) + second),
+		TTY:     tty,
+	}
+}
+
+// NewProcess : Initialize a new Process instance
+func NewProcess() *Process {
+	return &Process{
+		fields: fields{
+			Type:    Command,
+			Command: `ps -A u`,
+		},
+	}
+
+}

integration/integration_test.go

@@ -2,6 +2,7 @@ package integration
 
 import (
 	"fmt"
+	"strings"
 	"testing"
 
 	"github.com/bisoncorps/saido/driver"
@@ -46,3 +47,31 @@ func TestResponseTimeonWeb(t *testing.T) {
 	}
 	fmt.Printf(`%#v`, i.Values)
 }
+
+func TestProcessonSSH(t *testing.T) {
+	d := driver.NewSSHForTest()
+	i := inspector.NewProcess()
+	output, err := d.RunCommand(i.String())
+	if err != nil {
+		t.Error(err)
+	}
+	i.Parse(output)
+	if len(i.Values) <= 2 {
+		t.Error(err)
+	}
+}
+
+func TestCustomonSSH(t *testing.T) {
+	d := driver.NewSSHForTest()
+	// set vars
+	d.Vars = []string{"MONKEY=true"}
+	i := inspector.NewCustom(`echo $MONKEY`)
+	output, err := d.RunCommand(i.String())
+	if err != nil {
+		t.Error(err)
+	}
+	i.Parse(output)
+	if strings.TrimSpace(i.Values.Output) != "true" {
+		t.Errorf("%s", i.Values.Output)
+	}
+}

integration/integration_unix_test.go

@@ -54,3 +54,22 @@ func TestDockerStatsonLocal(t *testing.T) {
 	}
 	fmt.Printf(`%#v`, i.Values)
 }
+
+func TestProcessonLocal(t *testing.T) {
+	d := driver.Local{}
+	i := inspector.NewProcess()
+	output, err := d.RunCommand(i.String())
+	if err != nil {
+		t.Error(err)
+	}
+	i.Parse(output)
+	if len(i.Values) <= 2 {
+		t.Error(err)
+	}
+	// Track just root PID of 1
+	i.TrackPID = 1
+	i.Parse(output)
+	if len(i.Values) != 1 {
+		t.Error("unexpected size of single PID tracking")
+	}
+}