From 25fd4297a8d908af2a755fd6fb6d13865532be6b Mon Sep 17 00:00:00 2001
From: Simone Aiello <s.aiello@beeweeb.com>
Date: Wed, 28 Oct 2015 09:13:22 +0100
Subject: [PATCH] Jolokia plugin first commit

---
 plugins/all/all.go         |   1 +
 plugins/jolokia/jolokia.go | 189 +++++++++++++++++++++++++++++++++++++
 2 files changed, 190 insertions(+)
 create mode 100644 plugins/jolokia/jolokia.go

diff --git a/plugins/all/all.go b/plugins/all/all.go
index f29a4987b..d714dd81e 100644
--- a/plugins/all/all.go
+++ b/plugins/all/all.go
@@ -9,6 +9,7 @@ import (
 	_ "github.com/influxdb/telegraf/plugins/exec"
 	_ "github.com/influxdb/telegraf/plugins/haproxy"
 	_ "github.com/influxdb/telegraf/plugins/httpjson"
+	_ "github.com/influxdb/telegraf/plugins/jolokia"
 	_ "github.com/influxdb/telegraf/plugins/kafka_consumer"
 	_ "github.com/influxdb/telegraf/plugins/leofs"
 	_ "github.com/influxdb/telegraf/plugins/lustre2"
diff --git a/plugins/jolokia/jolokia.go b/plugins/jolokia/jolokia.go
new file mode 100644
index 000000000..ee43246da
--- /dev/null
+++ b/plugins/jolokia/jolokia.go
@@ -0,0 +1,189 @@
+package jolokia
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	// "net/url"
+	"strings"
+	// "sync"
+
+	"github.com/influxdb/telegraf/plugins"
+)
+
+
+type Server struct {
+  Name string
+  Host string
+  Port string
+}
+
+type Metric struct {
+  Name string
+  Jmx string
+  Pass []string
+  Drop []string
+}
+
+type Jolokia struct {
+
+  Context   string
+  Servers   []Server
+  Metrics   []Metric
+
+}
+
+
+func (j *Jolokia) SampleConfig() string {
+	return `[jolokia]
+  context = "/jolokia/read"
+
+  [[jolokia.servers]]
+        name = "stable"
+        host = "192.168.103.2"
+        port = "8180"
+
+  [[jolokia.metrics]]
+        name = "heap_memory_usage"
+        jmx  = "/java.lang:type=Memory/HeapMemoryUsage"
+        pass = ["used"]
+
+  [[jolokia.metrics]]
+        name = "memory_eden"
+        jmx  = "/java.lang:type=MemoryPool,name=PS Eden Space/Usage"
+        pass = ["used"]
+
+  [[jolokia.metrics]]
+        name = "heap_threads"
+        jmx  = "/java.lang:type=Threading"
+ #      drop = ["AllThread"]
+        pass = ["CurrentThreadCpuTime","CurrentThreadUserTime","DaemonThreadCount","ThreadCount","TotalStartedThreadCount"]
+`
+}
+
+func (j *Jolokia) Description() string {
+	return "Read JMX metrics through Jolokia"
+}
+
+
+
+func getAttr(url string) (map[string]interface{}, error) {
+  //make request
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+  // Process response
+  if resp.StatusCode != http.StatusOK {
+    err = fmt.Errorf("Response from url \"%s\" has status code %d (%s), expected %d (%s)",
+      url,
+      resp.StatusCode,
+      http.StatusText(resp.StatusCode),
+      http.StatusOK,
+      http.StatusText(http.StatusOK))
+    return nil, err
+  }
+
+  // read body
+  body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+  // Unmarshal json
+  var jsonOut map[string]interface{}
+  if err = json.Unmarshal([]byte(body), &jsonOut); err != nil {
+    return nil, errors.New("Error decoding JSON response")
+  }
+
+  return jsonOut, nil
+}
+
+func (m *Metric) shouldPass(field string) bool {
+
+  if m.Pass != nil {
+
+    for _, pass := range m.Pass{
+      if strings.HasPrefix(field, pass) {
+				return true
+			}
+    }
+
+    return false
+  }
+
+  if m.Drop != nil {
+
+    for _, drop := range m.Drop{
+      if strings.HasPrefix(field, drop) {
+				return false
+			}
+    }
+
+    return true
+  }
+
+  return true
+}
+
+func (m *Metric) filterFields(fields map[string]interface{}) map[string]interface{} {
+
+  for field, _ := range fields{
+    if !m.shouldPass(field) {
+			delete(fields, field)
+		}
+  }
+
+  return fields
+}
+
+
+func (j *Jolokia) Gather(acc plugins.Accumulator) error {
+
+  context := j.Context //"/jolokia/read"
+  servers := j.Servers
+  metrics := j.Metrics
+
+  var tags = map[string]string{
+    "group": "application_server",
+  }
+
+  for _, server := range servers {
+    for _, metric := range metrics {
+
+      measurement := metric.Name
+      jmxPath := metric.Jmx
+
+      tags["server"] = server.Name
+			tags["port"] = server.Port
+			tags["host"] = server.Host
+
+      url := "http://" + server.Host + ":" + server.Port + context + jmxPath
+      //fmt.Println(url)
+      out, _ := getAttr(url)
+
+      if values, ok := out["value"]; ok {
+        switch values.(type) {
+          case map[string]interface{}:
+            acc.AddFields(measurement, metric.filterFields(values.(map[string]interface{})), tags)
+          case interface{}:
+            acc.Add(measurement, values.(interface{}), tags)
+        }
+      }else{
+        fmt.Println("Missing key value")
+      }
+    }
+  }
+
+  return nil
+}
+
+func init() {
+	plugins.Add("jolokia", func() plugins.Plugin {
+		return &Jolokia{}
+	})
+}