set log level using env variables

Signed-off-by: Amruta Kale <amrutakale24.1991@gmail.com>

pkg/log/log.go

@@ -25,6 +25,10 @@ const (
 	InfoLevel Level = Level(logrus.InfoLevel)
 	// ErrorLevel log level.
 	ErrorLevel Level = Level(logrus.ErrorLevel)
+
+	// LevelVarName is the environment variable that controls
+	// init log levels
+	LevelVarName = "LOG_LEVEL"
 )
 
 // OutputSink describes the current output sink.
@@ -121,6 +125,21 @@ func SetFormatter(format OutputFormat) {
 func init() {
 	SetFormatter(TextFormat)
 	initEnvVarFields()
+	level, err := logrus.ParseLevel(os.Getenv(LevelVarName))
+
+	if err != nil {
+		level = logrus.InfoLevel
+	}
+	SetLevel(Level(level))
+}
+
+// SetLevel sets the current log level.
+func SetLevel(level Level) {
+	log.SetLevel(logrus.Level(level))
+}
+
+func GetLevel() logrus.Level {
+	return log.GetLevel()
 }
 
 func Info() Logger {

pkg/log/log_test.go

@@ -105,3 +105,27 @@ func testLogMessage(c *C, msg string, print func(string, ...field.M), fields ...
 	c.Assert(entry["msg"], Equals, msg)
 	return entry
 }
+
+func (s *LogSuite) TestLogLevel(c *C) {
+	log.SetFormatter(&logrus.JSONFormatter{TimestampFormat: time.RFC3339Nano})
+
+	var output bytes.Buffer
+	log.SetOutput(&output)
+	ctx := field.Context(context.Background(), "key", "value")
+	var entry map[string]interface{}
+	//Check if debug level log is printed when log level is info
+	Debug().WithContext(ctx).Print("Testing debug level")
+	err := json.Unmarshal(output.Bytes(), &entry)
+
+	c.Assert(err, NotNil)
+	c.Assert(output.String(), HasLen, 0)
+
+	//Check if debug level log is printed when log level is debug
+	log.SetLevel(logrus.DebugLevel)
+	Debug().WithContext(ctx).Print("Testing debug level")
+	err = json.Unmarshal(output.Bytes(), &entry)
+	c.Assert(err, IsNil)
+	c.Assert(entry, NotNil)
+	c.Assert(entry["msg"], Equals, "Testing debug level")
+
+}