bosun-monitor · captncraig · Jul 27, 2015 · Jul 27, 2015
diff --git a/cmd/scollector/collectors/riak.go b/cmd/scollector/collectors/riak.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net/http"
+	"net/url"
 	"strings"
 
 	"bosun.org/metadata"
@@ -460,16 +461,35 @@ func init() {
 }
 
 const (
-	riakURL string = "http://localhost:8098/stats"
+	localRiakURL string = "http://localhost:8098/stats"
 )
 
+func Riak(s string) error {
+	u, err := url.Parse(s)
+	if err != nil {
+		return err
+	}
+	collectors = append(collectors,
+		&IntervalCollector{
+			F: func() (opentsdb.MultiDataPoint, error) {
+				return riak(s)
+			},
+			name: fmt.Sprintf("riak-%s", u.Host),
+		})
+	return nil
+}
+
 func enableRiak() bool {
-	return enableURL(riakURL)()
+	return enableURL(localRiakURL)()
 }
 
 func c_riak() (opentsdb.MultiDataPoint, error) {
+	return riak(localRiakURL)
+}
+
+func riak(s string) (opentsdb.MultiDataPoint, error) {
 	var md opentsdb.MultiDataPoint
-	res, err := http.Get(riakURL)
+	res, err := http.Get(s)
 	if err != nil {
 		return nil, err
 	}

diff --git a/cmd/scollector/conf/conf.go b/cmd/scollector/conf/conf.go
@@ -42,6 +42,7 @@ type Conf struct {
 	Process       []ProcessParams
 	ProcessDotNet []ProcessDotNet
 	HTTPUnit      []HTTPUnit
+	Riak          []Riak
 }
 
 type HAProxy struct {
@@ -112,3 +113,7 @@ type HTTPUnit struct {
 	TOML  string
 	Hiera string
 }
+
+type Riak struct {
+	URL string
+}
diff --git a/cmd/scollector/doc.go b/cmd/scollector/doc.go
@@ -202,6 +202,7 @@ AWS (array of table, keys are AccessKey, SecretKey, Region): AWS hosts to poll.
 	  SecretKey = "snch0d"
 	  Region = "somewhere"
 
+
 Process (array of table, keys are Command, Name, Args for Linux, and Name
 for Windows): processes to monitor. Name is optional, and defaults to Command.
 
@@ -238,6 +239,11 @@ or just one.
 	[[HTTPUnit]]
 	  TOML = "/some/other.toml"
 
+Riak (array of table, keys are URL): Riak hosts to poll.
+
+	[[Riak]]
+	  URL = "http://localhost:8098/stats"
+
 Windows
 
 scollector has full Windows support. It can be run standalone, or installed as a

diff --git a/cmd/scollector/main.go b/cmd/scollector/main.go
@@ -128,6 +128,9 @@ func main() {
 			check(collectors.HTTPUnitHiera(h.Hiera))
 		}
 	}
+	for _, r := range conf.Riak {
+		check(collectors.Riak(r.URL))
+	}
 	if err != nil {
 		slog.Fatal(err)
 	}