allow to use specific logrus level in the adaptor

log/logrus/logrus_logger.go

@@ -10,18 +10,38 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
-type logrusLogger struct {
+type Logger struct {
 	logrus.FieldLogger
+	logrus.Level
 }
 
+type Option func(*Logger)
+
 var errMissingValue = errors.New("(MISSING)")
 
-// NewLogrusLogger returns a go-kit log.Logger that sends log events to a Logrus logger.
-func NewLogrusLogger(logger logrus.FieldLogger) log.Logger {
-	return &logrusLogger{logger}
+// NewLogger returns a go-kit log.Logger that sends log events to a Logrus logger.
+func NewLogger(logger logrus.FieldLogger, options ...Option) log.Logger {
+	l := &Logger{
+		FieldLogger: logger,
+		Level:       logrus.InfoLevel,
+	}
+
+	for _, optFunc := range options {
+		optFunc(l)
+	}
+
+	return l
 }
 
-func (l logrusLogger) Log(keyvals ...interface{}) error {
+// WithLevel configures a logrus logger to set specific log
+// level to log messages with
+func WithLevel(level logrus.Level) Option {
+	return func(c *Logger) {
+		c.Level = level
+	}
+}
+
+func (l Logger) Log(keyvals ...interface{}) error {
 	fields := logrus.Fields{}
 	for i := 0; i < len(keyvals); i += 2 {
 		if i+1 < len(keyvals) {
@@ -30,6 +50,19 @@ func (l logrusLogger) Log(keyvals ...interface{}) error {
 			fields[fmt.Sprint(keyvals[i])] = errMissingValue
 		}
 	}
-	l.WithFields(fields).Info()
+
+	switch l.Level {
+	case logrus.InfoLevel:
+		l.WithFields(fields).Info()
+	case logrus.ErrorLevel:
+		l.WithFields(fields).Error()
+	case logrus.DebugLevel:
+		l.WithFields(fields).Debug()
+	case logrus.WarnLevel:
+		l.WithFields(fields).Warn()
+	default:
+		l.WithFields(fields).Print()
+	}
+
 	return nil
 }

log/logrus/logrus_logger_test.go

@@ -16,7 +16,7 @@ func TestLogrusLogger(t *testing.T) {
 	logrusLogger := logrus.New()
 	logrusLogger.Out = buf
 	logrusLogger.Formatter = &logrus.TextFormatter{TimestampFormat: "02-01-2006 15:04:05", FullTimestamp: true}
-	logger := log.NewLogrusLogger(logrusLogger)
+	logger := log.NewLogger(logrusLogger)
 
 	if err := logger.Log("hello", "world"); err != nil {
 		t.Fatal(err)