From cba55bc2d40f27f7126a277d9e8886488fe1e7f6 Mon Sep 17 00:00:00 2001
From: Troy Denton <tdenton@24-7intouch.com>
Date: Thu, 12 Mar 2020 15:16:39 -0500
Subject: [PATCH] add custom logger functionality

---
 logger.go | 47 ++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 40 insertions(+), 7 deletions(-)

diff --git a/logger.go b/logger.go
index 345607a..c6ae152 100644
--- a/logger.go
+++ b/logger.go
@@ -13,8 +13,8 @@ import (
 )
 
 var (
-	log = logging.MustGetLogger("goesl")
-
+	log       = logging.MustGetLogger("goesl")
+	customLog Logging
 	// Example format string. Everything except the message has a custom color
 	// which is dependent on the log level. Many fields have a custom output
 	// formatting too, eg. the time returns the hour down to the milli second.
@@ -24,27 +24,60 @@ var (
 )
 
 func Debug(message string, args ...interface{}) {
-	log.Debugf(message, args...)
+	if customLog == nil {
+		log.Debugf(message, args...)
+	} else {
+		customLog.Debugf(message, args...)
+	}
 }
 
 func Error(message string, args ...interface{}) {
-	log.Errorf(message, args...)
+	if customLog == nil {
+		log.Errorf(message, args...)
+	} else {
+		customLog.Errorf(message, args...)
+	}
 }
 
 func Notice(message string, args ...interface{}) {
-	log.Noticef(message, args...)
+	if customLog == nil {
+		log.Noticef(message, args...)
+	} else {
+		customLog.Noticef(message, args...)
+	}
 }
 
 func Info(message string, args ...interface{}) {
-	log.Infof(message, args...)
+	if customLog == nil {
+		log.Infof(message, args...)
+	} else {
+		customLog.Infof(message, args...)
+	}
 }
 
 func Warning(message string, args ...interface{}) {
-	log.Warningf(message, args...)
+	if customLog == nil {
+		log.Warningf(message, args...)
+	} else {
+		customLog.Warningf(message, args...)
+	}
+}
+
+type Logging interface {
+	Debugf(string, ...interface{})
+	Errorf(string, ...interface{})
+	Noticef(string, ...interface{})
+	Infof(string, ...interface{})
+	Warningf(string, ...interface{})
 }
 
 func init() {
 	backend := logging.NewLogBackend(os.Stderr, "", 0)
 	formatter := logging.NewBackendFormatter(backend, format)
 	logging.SetBackend(formatter)
+	customLog = nil
+}
+
+func SetCustomLogger(l Logging) {
+	customLog = l
 }