From 290042decb3aff96a1e3f9b4fd73345da35b5786 Mon Sep 17 00:00:00 2001
From: gondor <jeremyunruh@sbcglobal.net>
Date: Sun, 12 Mar 2017 21:36:45 -0700
Subject: [PATCH] update logger - allow hooks

---
 logger/logger.go | 37 +++++++++++++++++++++++++++++--------
 1 file changed, 29 insertions(+), 8 deletions(-)

diff --git a/logger/logger.go b/logger/logger.go
index f64f121..459d60d 100644
--- a/logger/logger.go
+++ b/logger/logger.go
@@ -21,12 +21,26 @@ type Log struct {
 	logrus.Logger
 }
 
+type LogConfigurationHook interface {
+
+	// ConfigureLogger is a hook which allows custom setup and overrides for the default
+	// category based logger.  This allows the implementation to override formatting, output
+	// and level hooks.
+	ConfigureLogger(logger *Log)
+}
+
+var configHook LogConfigurationHook
 var loggers map[string]*Log
-var globalLogger *Log
 
 func init() {
 	loggers = map[string]*Log{}
-	globalLogger = GetLogger(DefaultCategory)
+}
+
+func SetLogConfigurationHook(hook LogConfigurationHook) {
+	configHook = hook
+	for _, l := range loggers {
+		hook.ConfigureLogger(l)
+	}
 }
 
 // GetLogger returns an existing logger for the specified category or
@@ -42,7 +56,7 @@ func GetLogger(category string) *Log {
 
 // Logger is a simple accessor to the global logging instance
 func Logger() *Log {
-	return globalLogger
+	return GetLogger(DefaultCategory)
 }
 
 // SetLevel enforces the specified level on the specified category.  If a logger
@@ -57,10 +71,17 @@ func SetLevel(level logrus.Level, category string) {
 
 func newLogger(c string) *Log {
 	log := &Log{category: c}
-	log.Out = os.Stderr
-	log.Formatter = &logrus.TextFormatter{ DisableTimestamp: false, TimestampFormat: DefaultTimeFormat}
-	log.Hooks = make(logrus.LevelHooks)
-	log.Level = logrus.InfoLevel
-
+	if configHook == nil {
+		log.Out = os.Stderr
+		log.Formatter = &logrus.TextFormatter{DisableTimestamp: false, TimestampFormat: DefaultTimeFormat}
+		log.Hooks = make(logrus.LevelHooks)
+		log.Level = logrus.InfoLevel
+	} else {
+		configHook.ConfigureLogger(log)
+	}
 	return log
 }
+
+func (l *Log) Category() string {
+	return l.category
+}