Merge pull request #77 from albertsj1/add_statsd_env_variable

Add support for STATSD_* environment variables

configuration/configuration.go

@@ -5,6 +5,7 @@ import (
 	"io/ioutil"
 	"log"
 	"os"
+	"strconv"
 )
 
 var logger = log.New(os.Stdout, "", log.Ldate|log.Ltime|log.Lshortfile)
@@ -52,6 +53,9 @@ func FromFile(filePath string) (Configuration, error) {
 	setValueFromEnv(&conf.HAProxy.TemplatePath, "HAPROXY_TEMPLATE_PATH")
 	setValueFromEnv(&conf.HAProxy.OutputPath, "HAPROXY_OUTPUT_PATH")
 	setValueFromEnv(&conf.HAProxy.ReloadCommand, "HAPROXY_RELOAD_CMD")
+	setValueFromEnv(&conf.StatsD.Host, "STATSD_HOST")
+	setValueFromEnv(&conf.StatsD.Prefix, "STATSD_PREFIX")
+	setBoolValueFromEnv(&conf.StatsD.Enabled, "STATSD_ENABLED")
 	return *conf, err
 }
 
@@ -62,3 +66,17 @@ func setValueFromEnv(field *string, envVar string) {
 		*field = env
 	}
 }
+
+func setBoolValueFromEnv(field *bool, envVar string) {
+env := os.Getenv(envVar)
+if len(env) > 0 {
+	log.Printf("Using environment override %s=%t", envVar, env)
+	x, err := strconv.ParseBool(env)
+	if err != nil {
+		log.Printf("Error converting boolean value: %s\n", err)
+	}
+	*field = x
+} else {
+	log.Printf("Environment variable not set: %s", envVar)
+}
+}