From 72524e911d01b89205ea29ceaa3dbd8c2ed4d52d Mon Sep 17 00:00:00 2001
From: Ryan Uber <ru@ryanuber.com>
Date: Sun, 3 May 2015 16:46:20 -0700
Subject: [PATCH] agent: allow configuring the stats prefix

---
 command/agent/command.go     |  2 +-
 command/agent/config.go      |  8 ++++++++
 command/agent/config_test.go | 11 +++++++++++
 3 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/command/agent/command.go b/command/agent/command.go
index 294e991b9e..1020b46c65 100644
--- a/command/agent/command.go
+++ b/command/agent/command.go
@@ -567,7 +567,7 @@ func (c *Command) Run(args []string) int {
 	*/
 	inm := metrics.NewInmemSink(10*time.Second, time.Minute)
 	metrics.DefaultInmemSignal(inm)
-	metricsConf := metrics.DefaultConfig("consul")
+	metricsConf := metrics.DefaultConfig(config.StatsitePrefix)
 
 	// Configure the statsite sink
 	var fanout metrics.FanoutSink
diff --git a/command/agent/config.go b/command/agent/config.go
index 48b347c332..8319932c8a 100644
--- a/command/agent/config.go
+++ b/command/agent/config.go
@@ -160,6 +160,10 @@ type Config struct {
 	// metrics will be streamed to that instance.
 	StatsiteAddr string `mapstructure:"statsite_addr"`
 
+	// StatsitePrefix is the prefix used to write stats values to. By
+	// default this is set to 'consul'.
+	StatsitePrefix string `mapstructure:"statsite_prefix"`
+
 	// StatsdAddr is the address of a statsd instance. If provided,
 	// metrics will be sent to that instance.
 	StatsdAddr string `mapstructure:"statsd_addr"`
@@ -429,6 +433,7 @@ func DefaultConfig() *Config {
 		DNSConfig: DNSConfig{
 			MaxStale: 5 * time.Second,
 		},
+		StatsitePrefix:      "consul",
 		SyslogFacility:      "LOCAL0",
 		Protocol:            consul.ProtocolVersionMax,
 		CheckUpdateInterval: 5 * time.Minute,
@@ -811,6 +816,9 @@ func MergeConfig(a, b *Config) *Config {
 	if b.StatsiteAddr != "" {
 		result.StatsiteAddr = b.StatsiteAddr
 	}
+	if b.StatsitePrefix != "" {
+		result.StatsitePrefix = b.StatsitePrefix
+	}
 	if b.StatsdAddr != "" {
 		result.StatsdAddr = b.StatsdAddr
 	}
diff --git a/command/agent/config_test.go b/command/agent/config_test.go
index 945e2152d4..45bfd9e1fe 100644
--- a/command/agent/config_test.go
+++ b/command/agent/config_test.go
@@ -568,6 +568,16 @@ func TestDecodeConfig(t *testing.T) {
 		t.Fatalf("bad: %#v", config)
 	}
 
+	// Statsite prefix
+	input = `{"statsite_prefix": "my_prefix"}`
+	config, err = DecodeConfig(bytes.NewReader([]byte(input)))
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+	if config.StatsitePrefix != "my_prefix" {
+		t.Fatalf("bad: %#v", config)
+	}
+
 	// Address overrides
 	input = `{"addresses": {"dns": "0.0.0.0", "http": "127.0.0.1", "https": "127.0.0.1", "rpc": "127.0.0.1"}}`
 	config, err = DecodeConfig(bytes.NewReader([]byte(input)))
@@ -1114,6 +1124,7 @@ func TestMergeConfig(t *testing.T) {
 		},
 		DisableRemoteExec:         true,
 		StatsiteAddr:              "127.0.0.1:7250",
+		StatsitePrefix:            "stats_prefix",
 		StatsdAddr:                "127.0.0.1:7251",
 		DisableUpdateCheck:        true,
 		DisableAnonymousSignature: true,