From 7e17755f146c797a4a05e9f26fa70f7c21242cad Mon Sep 17 00:00:00 2001
From: Xaelias <GitHub@ALesieur.net>
Date: Fri, 10 Apr 2020 13:24:23 -0500
Subject: [PATCH] secondary indexes] Scrape metrics specific to secondary
 indexes.

---
 main.go   |   2 ++
 sindex.go | 100 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)
 create mode 100644 sindex.go

diff --git a/main.go b/main.go
index f643938..978803f 100644
--- a/main.go
+++ b/main.go
@@ -33,6 +33,7 @@ const (
 	systemLatencyHist = "latency_hist" // total number of ops
 	systemOps         = "ops"
 	systemSet         = "set"
+	secondaryIndex    = "sindex"
 )
 
 var (
@@ -129,6 +130,7 @@ func newAsCollector(nodeAddr, username, password string) *asCollector {
 			newNSCollector(),
 			newLatencyCollector(),
 			newSetCollector(),
+			newSindexCollector(),
 		},
 	}
 }
diff --git a/sindex.go b/sindex.go
new file mode 100644
index 0000000..c932878
--- /dev/null
+++ b/sindex.go
@@ -0,0 +1,100 @@
+package main
+
+import (
+  "strings"
+
+  as "github.com/aerospike/aerospike-client-go"
+  "github.com/prometheus/client_golang/prometheus"
+)
+
+var (
+  // SindexMetrics lists the keys we report from aero's sindex statistics command.
+  // See `asinfo -l -v sindex` for a list of secondary indexes.
+  // See `asinfo -l -v sindex/<namespace>` for a list of secondary indexes for a given namespace.
+  // See `asinfo -l -v sindex/<namespace>/<sindex_name>` for detailed metrics for a given secondary index.
+  SindexMetrics = []metric{
+    gauge("keys", "keys"),
+    gauge("entries", "entries"),
+    gauge("ibtr_memory_used", "ibtr_memory_used"),
+    gauge("nbtr_memory_used", "nbtr_memory_used"),
+    gauge("si_accounted_memory", "si_accounted_memory"),
+    gauge("load_pct", "load_pct"),
+    counter("loadtime", "loadtime"),
+    counter("write_success", "write_success"),
+    counter("write_error", "write_error"),
+    counter("delete_success", "delete_success"),
+    counter("delete_error", "delete_error"),
+    counter("stat_gc_recs", "stat_gc_recs"),
+    counter("stat_gc_time", "stat_gc_time"),
+    counter("query_reqs", "query_reqs"),
+    gauge("query_avg_rec_count", "query_avg_rec_count"),
+    gauge("query_avg_record_size", "query_avg_record_size"),
+    counter("query_agg", "query_agg"),
+    gauge("query_agg_avg_rec_count", "query_agg_avg_rec_count"),
+    gauge("query_agg_avg_record_size", "query_agg_avg_record_size"),
+    counter("query_lookups", "query_lookups"),
+    gauge("query_lookup_avg_rec_count", "query_lookup_avg_rec_count"),
+    gauge("query_lookup_avg_record_size", "query_lookup_avg_record_size"),
+  }
+)
+
+type sindexCollector cmetrics
+
+func newSindexCollector() sindexCollector {
+  sindex := map[string]cmetric {}
+  for _, m := range SindexMetrics {
+    sindex[m.aeroName] = cmetric{
+      typ: m.typ,
+      desc: prometheus.NewDesc(
+        promkey(secondaryIndex, m.aeroName),
+        m.desc,
+        []string{"namespace", "sindex", "set", "bin", "type", "indextype", "path"},
+        nil,
+      ),
+    }
+  }
+  return sindex
+}
+
+func (sindexc sindexCollector) describe(ch chan<- *prometheus.Desc) {
+  for _, s := range sindexc {
+    ch <- s.desc
+  }
+}
+
+func (sic sindexCollector) collect(conn *as.Connection) ([]prometheus.Metric, error) {
+  info, err := as.RequestInfo(conn, "sindex")
+  if err != nil {
+    return nil, err
+  }
+
+  var metrics []prometheus.Metric
+  for _, sindexInfo := range strings.Split(info["sindex"], ";") {
+    if sindexInfo == "" {
+      continue
+    }
+    sindexStats := parseInfo(sindexInfo)
+    ns := sindexStats["ns"]
+    sindexName := sindexStats["indexname"]
+    sindexDetails, err := as.RequestInfo(conn, "sindex/"+ns+"/"+sindexName)
+    if err != nil {
+      return nil, err
+    }
+
+    metrics = append(
+      metrics,
+      infoCollect(
+        cmetrics(sic),
+        sindexDetails["sindex/"+ns+"/"+sindexName],
+        ns,
+        sindexName,
+        sindexStats["set"],
+        sindexStats["bin"],
+        sindexStats["type"],
+        sindexStats["indextype"],
+        sindexStats["path"],
+      )...,
+    )
+  }
+  return metrics, nil
+}