Merge pull request #637 from ghjm/pluggable_logger

Allow routing Receptor logs to another logger

Author: fosterseth

pkg/logger/logger.go

@@ -76,8 +76,17 @@ var logLevelMap = map[string]int{
 	"debug":   DebugLevel,
 }
 
+type MessageFunc func(level int, format string, v ...interface{})
+
+var logger MessageFunc
+
 // Log sends a log message at a given level.
 func Log(level int, format string, v ...interface{}) {
+	if logger != nil {
+		logger(level, format, v...)
+
+		return
+	}
 	var prefix string
 	logLevelName, err := LogLevelToName(level)
 	if err != nil {
@@ -92,6 +101,11 @@ func Log(level int, format string, v ...interface{}) {
 	}
 }
 
+// RegisterLogger registers a function for log delivery.
+func RegisterLogger(msgFunc MessageFunc) {
+	logger = msgFunc
+}
+
 // Error reports unexpected behavior, likely to result in termination.
 func Error(format string, v ...interface{}) {
 	Log(ErrorLevel, format, v...)