metric prefix

Author: sokil

README.md

@@ -36,8 +36,22 @@ statsd-rest-server \
 	--http-port=8080 \
 	--statsd-host=127.0.0.1 \
 	--statsd-port=8125 \
-	--jwt-secret=somesecret
-```
+	--jwt-secret=somesecret \
+	--metric-prefix=prefix.subprefix
+```
+
+Command line arguments:
+
+| Parameter       | Description                          | Default value                                                                  |
+|-----------------|--------------------------------------|--------------------------------------------------------------------------------|
+| verbose         | Print debug info to stderr           | Optional. Default false                                                        |
+| http-host       | Host of HTTP server                  | Optional. Default 127.0.0.1. To accept connections on any interface, set to "" |
+| http-port       | Port of HTTP server                  | Optional. Default 80                                                           |
+| statsd-host     | Host of StatsD instance              | Optional. Default 127.0.0.1                                                    |
+| statsd-port     | Port of StatsD instance              | Optional. Default 8125                                                         |
+| jwt-secret      | JWT token secret                     | Optional. If not set, server accepts all connections                           |
+| metric-prefix   | Prefix, added to any metric name     | Optional. If not set, do not add prefix                                        |
+| version         | Print version of server and exit     | Optional                                                                       |
 
 ## Authentication
 

main.go

@@ -42,6 +42,7 @@ var httpHost = flag.String("http-host", defaultHTTPHost, "HTTP Host")
 var httpPort = flag.Int("http-port", defaultHTTPPort, "HTTP Port")
 var statsdHost = flag.String("statsd-host", defaultStatsDHost, "StatsD Host")
 var statsdPort = flag.Int("statsd-port", defaultStatsDPort, "StatsD Port")
+var metricPrefix = flag.String("metric-prefix", "", "Prefix of metric name")
 var tokenSecret = flag.String("jwt-secret", "", "Secret to encrypt JWT")
 var verbose = flag.Bool("verbose", false, "Verbose")
 var version = flag.Bool("version", false, "Show version")
@@ -66,6 +67,11 @@ func main() {
 		log.SetOutput(ioutil.Discard)
 	}
 
+	// prepare metric prefix
+	if *metricPrefix != "" && (*metricPrefix)[len(*metricPrefix) - 1:] != "." {
+		*metricPrefix = *metricPrefix + "."
+	}
+
 	// create HTTP router
 	router := mux.NewRouter().StrictSlash(true)
 
@@ -176,7 +182,7 @@ func handleHeartbeatRequest(w http.ResponseWriter, r *http.Request) {
 func handleCountRequest(w http.ResponseWriter, r *http.Request) {
 	// get key
 	vars := mux.Vars(r)
-	key := vars["key"]
+	key := *metricPrefix + vars["key"]
 
 	// get count value
 	var value = 1
@@ -209,7 +215,7 @@ func handleCountRequest(w http.ResponseWriter, r *http.Request) {
 func handleGaugeRequest(w http.ResponseWriter, r *http.Request) {
 	// get key
 	vars := mux.Vars(r)
-	key := vars["key"]
+	key := *metricPrefix + vars["key"]
 
 	// get gauge shift
 	shiftPostFormValue := r.PostFormValue("shift")
@@ -246,7 +252,7 @@ func handleGaugeRequest(w http.ResponseWriter, r *http.Request) {
 func handleTimingRequest(w http.ResponseWriter, r *http.Request) {
 	// get key
 	vars := mux.Vars(r)
-	key := vars["key"]
+	key := *metricPrefix + vars["key"]
 
 	// get timing
 	time, err := strconv.ParseInt(r.PostFormValue("time"), 10, 64)
@@ -273,7 +279,7 @@ func handleTimingRequest(w http.ResponseWriter, r *http.Request) {
 func handleSetRequest(w http.ResponseWriter, r *http.Request) {
 	// get key
 	vars := mux.Vars(r)
-	key := vars["key"]
+	key := *metricPrefix + vars["key"]
 
 	// get set value
 	var value = 1