logger: add level support

Signed-off-by: Maciek Borzecki <maciek.borzecki@gmail.com>

logger/logger.go

@@ -35,4 +35,17 @@ type Logger interface {
 	Print(args ...interface{})
 	Printf(format string, args ...interface{})
 	Println(args ...interface{})
+
+	SetLevel(level Level)
 }
+
+type Level int
+
+const (
+	PanicLevel Level = iota
+	FatalLevel
+	ErrorLevel
+	WarnLevel
+	InfoLevel
+	DebugLevel
+)

logger/logrus.go

@@ -17,19 +17,29 @@ import (
 	"github.com/Sirupsen/logrus"
 )
 
-var log *logrus.Logger
+type logger struct {
+	logrus.Logger
+}
+
+var log *logger
 
 func init() {
-	log = &logrus.Logger{
-		Out: os.Stderr,
-		Formatter: &logrus.TextFormatter{
-			FullTimestamp: true,
+	log = &logger{
+		Logger: logrus.Logger{
+			Out: os.Stderr,
+			Formatter: &logrus.TextFormatter{
+				FullTimestamp: true,
+			},
+			Hooks: make(logrus.LevelHooks),
+			Level: logrus.DebugLevel,
 		},
-		Hooks: make(logrus.LevelHooks),
-		Level: logrus.DebugLevel,
 	}
 }
 
 func New() Logger {
 	return log
 }
+
+func (l *logger) SetLevel(level Level) {
+	l.Logger.SetLevel(logrus.Level(level))
+}