Update applogger.go

HuobiRDCenter · Sep 29, 2020 · cc7e268 · cc7e268
1 parent 0705717
commit cc7e268
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/logging/applogger/applogger.go b/logging/applogger/applogger.go
@@ -7,6 +7,7 @@ import (
 )
 
 var sugaredLogger *zap.SugaredLogger
+var atomicLevel zap.AtomicLevel
 
 func init() {
 	encoderCfg := zapcore.EncoderConfig {
@@ -17,10 +18,18 @@ func init() {
 		EncodeTime:     zapcore.ISO8601TimeEncoder,
 	}
 
-	core := zapcore.NewCore(zapcore.NewConsoleEncoder(encoderCfg), os.Stdout, zap.DebugLevel)
+	// define default level as debug level
+	atomicLevel = zap.NewAtomicLevel()
+	atomicLevel.SetLevel(zapcore.DebugLevel)
+
+	core := zapcore.NewCore(zapcore.NewConsoleEncoder(encoderCfg), os.Stdout, atomicLevel)
 	sugaredLogger = zap.New(core).Sugar()
 }
 
+func SetLevel(level zapcore.Level) {
+	atomicLevel.SetLevel(level)
+}
+
 func Fatal(template string, args ...interface{}) {
 	sugaredLogger.Fatalf(template, args...)
 }