Merge pull request jenningsloy318#69 from chiveturkey/set-log-level

Add ability to set log level globally for the app.

Author: jenningsloy318

config.go

@@ -9,8 +9,9 @@ import (
 )
 
 type Config struct {
-	Hosts  map[string]HostConfig `yaml:"hosts"`
-	Groups map[string]HostConfig `yaml:"groups"`
+	Hosts    map[string]HostConfig `yaml:"hosts"`
+	Groups   map[string]HostConfig `yaml:"groups"`
+	Loglevel string                `yaml:"loglevel"`
 }
 
 type SafeConfig struct {
@@ -69,3 +70,13 @@ func (sc *SafeConfig) HostConfigForGroup(group string) (*HostConfig, error) {
 	}
 	return &HostConfig{}, fmt.Errorf("no credentials found for group %s", group)
 }
+
+func (sc *SafeConfig) AppLogLevel() (string) {
+	sc.Lock()
+	defer sc.Unlock()
+	logLevel := sc.C.Loglevel
+	if logLevel != "" {
+		return logLevel
+	}
+	return "info"
+}

config.yml.example

@@ -9,3 +9,5 @@ groups:
   group1:
     username: group1_user
     password: group1_pass
+# loglevel can be one of "debug", "info", "warn", "error", or "fatal"
+# loglevel: info

main.go

@@ -72,6 +72,15 @@ func reloadHandler(configLoggerCtx *alog.Entry) http.HandlerFunc {
 	}
 }
 
+func SetLogLevel() {
+	logLevel, err := alog.ParseLevel(sc.AppLogLevel())
+	if err != nil {
+		logLevel = alog.InfoLevel
+	}
+
+	alog.SetLevel(logLevel)
+}
+
 // define new http handleer
 func metricsHandler() http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
@@ -138,6 +147,8 @@ func main() {
 
 	configLoggerCtx.WithField("operation", "sc.ReloadConfig").Info("config file loaded")
 
+	SetLogLevel()
+
 	// load config in background to watch for config changes
 	hup := make(chan os.Signal, 1)
 	reloadCh = make(chan chan error)