ryan.merrick - 24/01/2016 08:50:07 PM

: Add basic newrelic input

plugins/inputs/all/all.go

@@ -20,6 +20,7 @@ import (
 	_ "github.com/influxdata/telegraf/plugins/inputs/memcached"
 	_ "github.com/influxdata/telegraf/plugins/inputs/mongodb"
 	_ "github.com/influxdata/telegraf/plugins/inputs/mysql"
+	_ "github.com/influxdata/telegraf/plugins/inputs/newrelic"
 	_ "github.com/influxdata/telegraf/plugins/inputs/nginx"
 	_ "github.com/influxdata/telegraf/plugins/inputs/nsq"
 	_ "github.com/influxdata/telegraf/plugins/inputs/passenger"

plugins/inputs/newrelic/newrelic.go

@@ -0,0 +1,178 @@
+package newrelic
+
+import (
+	"fmt"
+	"net/url"
+	"time"
+
+	"github.com/influxdata/telegraf/plugins/inputs"
+	"github.com/rmanocha/go_newrelic_api"
+)
+
+// Metric struct
+type Metric struct {
+	Values []string
+}
+
+// NewRelic Plugin Struct
+type NewRelic struct {
+	Name                    string
+	APIKey                  string
+	APPID                   int
+	Metrics                 map[string]Metric
+	PollServerAverages      bool
+	PollApplicationAverages bool
+}
+
+// Description of Plugin
+func (nr *NewRelic) Description() string {
+	return "Newrelic Metrics Query Plugin"
+}
+
+var sampleConfig = `
+# Name
+Name = ""
+
+#APIKey
+APIKey = ""
+
+#Application ID
+APPID =
+
+# Poll for Averages
+PollServerAverages=false
+PollApplicationAverages=false
+
+# Poll for individual metrics
+[inputs.newrelic.Metrics]
+  [inputs.newrelic.Metrics.Apdex]
+  Values = ["f", "count", "value", "threshold", "threshold_min", "score", "s"]
+
+  [inputs.newrelic.Metrics."Errors/all"]
+  Values = ["errors_per_minute", "error_count"]
+
+  [inputs.newrelic.Metrics."EndUser"]
+  Values = ["average_response_time", "calls_per_minute", "call_count"]
+
+  [inputs.newrelic.Metrics."HttpDispatcher"]
+  Values = ["average_response_time", "calls_per_minute", "call_count"]
+`
+
+// SampleConfig of Plugin
+func (nr *NewRelic) SampleConfig() string {
+	return sampleConfig
+}
+
+func (nr *NewRelic) GatherServerAverages(acc inputs.Accumulator, nrapi *go_newrelic_api.Newrelic) error {
+	dataserver := nrapi.GetServers()
+
+	for v := range dataserver.Servers {
+		tags := map[string]string{
+			"ServerName": fmt.Sprintf("%s", dataserver.Servers[v].Name),
+		}
+
+		fieldsServers := map[string]interface{}{
+			"CPU":             dataserver.Servers[v].ServerSummary.CPU,
+			"CPUStolen":       dataserver.Servers[v].ServerSummary.CPUStolen,
+			"DiskIOPercent":   dataserver.Servers[v].ServerSummary.DiskIO,
+			"MemoryPercent":   dataserver.Servers[v].ServerSummary.Memory,
+			"MemoryUsed":      dataserver.Servers[v].ServerSummary.MemoryUsed,
+			"MemoryTotal":     dataserver.Servers[v].ServerSummary.MemoryTotal,
+			"FullestDisk":     dataserver.Servers[v].ServerSummary.FullestDisk,
+			"FullestDiskFree": dataserver.Servers[v].ServerSummary.FullestDiskFree,
+		}
+
+		acc.AddFields("Servers", fieldsServers, tags)
+	}
+
+	return nil
+}
+
+func (nr *NewRelic) GatherApplicationAverages(acc inputs.Accumulator, nrapi *go_newrelic_api.Newrelic) error {
+	dataapplication := nrapi.GetApplication(nr.APPID)
+
+	fieldsApplication := map[string]interface{}{
+		"ResponceTime": dataapplication.Application.ApplicationSummary.ResponseTime,
+		"Throughput":   dataapplication.Application.ApplicationSummary.Throughput,
+		"ErrorRate":    dataapplication.Application.ApplicationSummary.ErrorRate,
+		"ApdexTarget":  dataapplication.Application.ApplicationSummary.ApdexTarget,
+		"ApdexScore":   dataapplication.Application.ApplicationSummary.ApdexScore,
+	}
+
+	fieldsEndUser := map[string]interface{}{
+		"ResponceTime": dataapplication.Application.EndUserSummary.ResponseTime,
+		"Throughput":   dataapplication.Application.EndUserSummary.Throughput,
+		"ApdexTarget":  dataapplication.Application.EndUserSummary.ApdexTarget,
+		"ApdexScore":   dataapplication.Application.EndUserSummary.ApdexScore,
+	}
+
+	acc.AddFields("Application", fieldsApplication, nil)
+	acc.AddFields("EndUser", fieldsEndUser, nil)
+
+	return nil
+}
+
+func (nr *NewRelic) GatherMetrics(acc inputs.Accumulator, nrapi *go_newrelic_api.Newrelic) error {
+	return nil
+}
+
+// Gather requested metrics
+func (nr *NewRelic) Gather(acc inputs.Accumulator) error {
+	conn := go_newrelic_api.NewNewrelic(nr.APIKey)
+
+	if nr.PollServerAverages {
+		nr.GatherServerAverages(acc, conn)
+	}
+
+	if nr.PollApplicationAverages {
+		nr.GatherApplicationAverages(acc, conn)
+	}
+
+	if len(nr.Metrics) > 0 {
+		t := time.Now()
+		tFrom := fmt.Sprintf("%d-%02d-%02dT%02d:%02d:%02d-00:00\n",
+			t.Year(), t.Month(), t.Day(),
+			t.Hour(), t.Minute()-1, t.Second())
+
+		tTo := fmt.Sprintf("%d-%02d-%02dT%02d:%02d:%02d-00:00\n",
+			t.Year(), t.Month(), t.Day(),
+			t.Hour(), t.Minute(), t.Second())
+
+		vals := url.Values{}
+
+		for k := range nr.Metrics {
+			vals.Add("names[]", k)
+		}
+
+		vals.Add("from", tFrom)
+		vals.Add("to", tTo)
+
+		result := conn.GetMetricData(nr.APPID, vals)
+
+		var fieldsMetrics map[string]interface{}
+		fieldsMetrics = make(map[string]interface{})
+
+		for metricRequest := range nr.Metrics {
+			for metricResult := range result.MetricData.Metrics {
+				if metricRequest == result.MetricData.Metrics[metricResult].Name {
+					// If new relic returned a metric we are after
+					for valueRequest := range nr.Metrics[metricRequest].Values {
+						for valueResult := range result.MetricData.Metrics[metricResult].Timeslices[0].Values {
+							if valueResult == nr.Metrics[metricRequest].Values[valueRequest] {
+								// If we matched a returned metric value that was requested
+								fieldsMetrics[metricRequest+"_"+valueResult] = result.MetricData.Metrics[metricResult].Timeslices[0].Values[valueResult]
+							}
+						}
+					}
+				}
+			}
+		}
+		acc.AddFields("Metrics", fieldsMetrics, nil)
+	}
+
+	return nil
+}
+
+func init() {
+	inputs.Add("newrelic", func() inputs.Input { return &NewRelic{} })
+}