Merge pull request #1 from KevinPike/master

set log level using environment

main.go

@@ -2,13 +2,17 @@ package main
 
 import (
 	"net/http"
+	"os"
 	"strings"
 
 	log "github.com/Sirupsen/logrus"
 	"github.com/prometheus/client_golang/prometheus"
 )
 
+const defaultLogLevel = log.InfoLevel
+
 func initLogger() {
+	log.SetLevel(getLogLevel())
 	if strings.ToUpper(config.OutputFormat) == "JSON" {
 		log.SetFormatter(&log.JSONFormatter{})
 	} else {
@@ -44,3 +48,12 @@ func main() {
 
 	log.Fatal(http.ListenAndServe(":"+config.PublishPort, nil))
 }
+
+func getLogLevel() log.Level {
+	lvl := strings.ToLower(os.Getenv("LOG_LEVEL"))
+	level, err := log.ParseLevel(lvl)
+	if err != nil {
+		level = defaultLogLevel
+	}
+	return level
+}