Improve custom logging

config.go

@@ -24,6 +24,7 @@ var lookupPaths = []string{"", "./config", "/config", "../", "../config", "../..
 var ConfigName = "database.yml"
 
 func init() {
+	SetStandardLogger(defaultStdLogger)
 	SetLogger(defaultLogger)
 
 	ap := os.Getenv("APP_PATH")

connection_instrumented_test.go

@@ -19,7 +19,7 @@ func testInstrumentedDriver(p *suite.Suite) {
 	ctx, cancel := context.WithTimeout(context.TODO(), time.Second*5)
 	defer cancel()
 
-	// The WaitGroup and channel ensures that the logger is properly called. This can only happen
+	// The WaitGroup and channel ensures that the Logger is properly called. This can only happen
 	// when the instrumented driver is working as expected and returns the expected query.
 	var (
 		queryMySQL = "SELECT 1 FROM DUAL WHERE 1=?"

logger.go

@@ -9,15 +9,16 @@ import (
 	"github.com/gobuffalo/pop/v5/logging"
 )
 
-type logger func(lvl logging.Level, s string, args ...interface{})
+type Logger func(lvl logging.Level, s string, args ...interface{})
 
 // Debug mode, to toggle verbose log traces
 var Debug = false
 
 // Color mode, to toggle colored logs
 var Color = true
 
-var log logger
+var log Logger
+var standardLogger *stdlog.Logger
 
 var defaultStdLogger = stdlog.New(os.Stdout, "[POP] ", stdlog.LstdFlags)
 var defaultLogger = func(lvl logging.Level, s string, args ...interface{}) {
@@ -46,13 +47,19 @@ var defaultLogger = func(lvl logging.Level, s string, args ...interface{}) {
 	if Color {
 		s = color.YellowString(s)
 	}
-	defaultStdLogger.Println(s)
+	if standardLogger != nil {
+		standardLogger.Println(s)
+	}
 }
 
-// SetLogger overrides the default logger.
+// SetLogger overrides the default Logger.
 //
-// The logger must implement the following interface:
-// type logger func(lvl logging.Level, s string, args ...interface{})
-func SetLogger(l logger) {
+// The Logger must implement the following interface:
+// type Logger func(lvl logging.Level, s string, args ...interface{})
+func SetLogger(l Logger) {
 	log = l
 }
+
+func SetStandardLogger(logger *stdlog.Logger) {
+	standardLogger = logger
+}