Allow replacing the logger to facilitate those who wish to use no log…

…ging or non-standard logging

log.go

@@ -0,0 +1,59 @@
+package faktory_worker
+
+import (
+	"log"
+	"os"
+)
+
+type Logger interface {
+	Debug(v ...interface{})
+	Debugf(format string, args ...interface{})
+	Info(v ...interface{})
+	Infof(format string, args ...interface{})
+	Warn(v ...interface{})
+	Warnf(format string, args ...interface{})
+	Error(v ...interface{})
+	Errorf(format string, args ...interface{})
+	Fatal(v ...interface{})
+	Fatalf(format string, args ...interface{})
+}
+
+type StdLogger struct {
+	*log.Logger
+}
+
+func NewStdLogger() Logger {
+	return &StdLogger{log.New(os.Stdout, "Faktory ", log.LstdFlags)}
+}
+
+func (l *StdLogger) Debug(v ...interface{}) {
+	l.Println(v...)
+}
+
+func (l *StdLogger) Debugf(format string, v ...interface{}) {
+	l.Printf(format + "\n", v...)
+}
+
+func (l *StdLogger) Error(v ...interface{}) {
+	l.Println(v...)
+}
+
+func (l *StdLogger) Errorf(format string, v ...interface{}) {
+	l.Printf(format + "\n", v...)
+}
+
+func (l *StdLogger) Info(v ...interface{}) {
+	l.Println(v...)
+}
+
+func (l *StdLogger) Infof(format string, v ...interface{}) {
+	l.Printf(format + "\n", v...)
+}
+
+func (l *StdLogger) Warn(v ...interface{}) {
+	l.Println(v...)
+}
+
+func (l *StdLogger) Warnf(format string, v ...interface{}) {
+	l.Printf(format + "\n", v...)
+}

runner.go

@@ -3,7 +3,6 @@ package faktory_worker
 import (
 	"context"
 	"fmt"
-	"log"
 	"math/rand"
 	"os"
 	"strconv"
@@ -35,6 +34,7 @@ type Manager struct {
 	Concurrency int
 	Queues      []string
 	Pool
+	Logger      Logger
 
 	quiet bool
 	// The done channel will always block unless
@@ -54,20 +54,20 @@ func (mgr *Manager) On(event eventType, fn func()) {
 // After calling Quiet(), no more jobs will be pulled
 // from Faktory by this process.
 func (mgr *Manager) Quiet() {
-	log.Println("Quieting...")
+	mgr.Logger.Info("Quieting...")
 	mgr.quiet = true
 	mgr.fireEvent(Quiet)
 }
 
 // Terminate signals that the various components should shutdown.
 // Blocks on the shutdownWaiter until all components have finished.
 func (mgr *Manager) Terminate() {
-	log.Println("Shutting down...")
+	mgr.Logger.Info("Shutting down...")
 	close(mgr.done)
 	mgr.fireEvent(Shutdown)
 	mgr.shutdownWaiter.Wait()
 	mgr.Pool.Close()
-	log.Println("Goodbye")
+	mgr.Logger.Info("Goodbye")
 	os.Exit(0)
 }
 
@@ -76,6 +76,7 @@ func NewManager() *Manager {
 	return &Manager{
 		Concurrency: 20,
 		Queues:      []string{"default"},
+		Logger:      NewStdLogger(),
 
 		done:           make(chan interface{}),
 		shutdownWaiter: &sync.WaitGroup{},
@@ -183,7 +184,7 @@ func process(mgr *Manager, idx int) {
 		})
 
 		if err != nil {
-			log.Println(err)
+			mgr.Logger.Error(err)
 			time.Sleep(1 * time.Second)
 			continue
 		}