From dcb5e321124596481344873a0c1b03c9c832cb23 Mon Sep 17 00:00:00 2001
From: AJ Bahnken <aj@ajvb.me>
Date: Tue, 19 Feb 2019 14:35:50 -0800
Subject: [PATCH] Prefer STATSD_HOST env var over config value; fixes #13

---
 iprepd.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/iprepd.go b/iprepd.go
index 3f44443..ac020b7 100644
--- 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()
 }