diff --git a/log/logger.go b/log/logger.go index ed5270775..768352520 100644 --- a/log/logger.go +++ b/log/logger.go @@ -10,8 +10,14 @@ import ( "github.com/sirupsen/logrus" ) +// Logger allows overriding the logrus logger behavior +type Logger interface { + logrus.FieldLogger + WriterLevel(logrus.Level) *io.PipeWriter +} + var ( - logger *logrus.Entry + logger Logger logFilePath string logFile *os.File ) @@ -41,6 +47,11 @@ func SetLevel(level logrus.Level) { logrus.SetLevel(level) } +// SetLogger sets the logger. +func SetLogger(l Logger) { + logger = l +} + // GetLevel returns the standard logger level. func GetLevel() logrus.Level { return logrus.GetLevel()