Merge pull request #14 from mozilla-services/ajvb/prefer-statsd-env-var

Prefer STATSD_HOST env var over config value; fixes #13
mozilla-services · Feb 19, 2019 · 7653bd6 · 7653bd6
2 parents 177f71d + dcb5e32
commit 7653bd6
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/iprepd.go b/iprepd.go
@@ -1,8 +1,10 @@
 package iprepd
 
 import (
+	"fmt"
 	"io/ioutil"
 	"math/rand"
+	"os"
 	"time"
 
 	log "github.com/sirupsen/logrus"
@@ -93,6 +95,15 @@ func loadCfg(confpath string) (ret serverCfg, err error) {
 	if err != nil {
 		return
 	}
+	// prefer STATSD_HOST env var over config file (#13)
+	statsdHost := os.Getenv("STATSD_HOST")
+	if statsdHost != "" {
+		statsdPort := os.Getenv("STATSD_PORT")
+		if statsdPort == "" {
+			statsdPort = "8125"
+		}
+		ret.Statsd.Addr = fmt.Sprintf("%s:%s", statsdHost, statsdPort)
+	}
 	return ret, ret.validate()
 }