添加change level 测试用例

Author: yumaojun03

logger/zap/core.go

@@ -23,12 +23,14 @@ var (
 	_log unsafe.Pointer // Pointer to a coreLogger. Access via atomic.LoadPointer.
 )
 
+// 初始化默认Logger
 func init() {
 	storeLogger(&coreLogger{
 		selectors:    map[string]struct{}{},
 		rootLogger:   zap.NewNop(),
 		globalLogger: zap.NewNop(),
 		logger:       newLogger(zap.NewNop(), ""),
+		atom:         zap.NewAtomicLevel(),
 	})
 }
 

logger/zap/core_test.go

@@ -161,3 +161,35 @@ func TestL(t *testing.T) {
 		assert.Equal(t, "warning 1", log.Message)
 	}
 }
+
+func TestChangeLevel(t *testing.T) {
+	if err := zap.DevelopmentSetup(zap.ToObserverOutput()); err != nil {
+		t.Fatal(err)
+	}
+
+	const loggerName = "tester"
+	logger := zap.NewLogger(loggerName)
+
+	logger.Debug("debug")
+	logs := zap.ObserverLogs().TakeAll()
+	if assert.Len(t, logs, 1) {
+		assert.Equal(t, loggerName, logs[0].LoggerName)
+		assert.Equal(t, "debug", logs[0].Message)
+	}
+
+	zap.SetLevel(zap.ErrorLevel)
+	logger.Info("info")
+	logs = zap.ObserverLogs().TakeAll()
+	assert.Len(t, logs, 0)
+
+	logger.Warn("warn")
+	logs = zap.ObserverLogs().TakeAll()
+	assert.Len(t, logs, 0)
+
+	logger.Error("error")
+	logs = zap.ObserverLogs().TakeAll()
+	if assert.Len(t, logs, 1) {
+		assert.Equal(t, loggerName, logs[0].LoggerName)
+		assert.Equal(t, "error", logs[0].Message)
+	}
+}

logger/zap/logger.go

@@ -60,8 +60,6 @@ func (l *Logger) Named(name string) logger.Logger {
 	return &Logger{logger, logger.Sugar()}
 }
 
-// Sprint
-
 // Debug uses fmt.Sprint to construct and log a message.
 func (l *Logger) Debug(args ...interface{}) {
 	l.sugar.Debug(args...)
@@ -211,7 +209,7 @@ func (l *Logger) Recover(msg string) {
 }
 
 // SetLevel 动态设置日志等级
-func (l *Logger) SetLevel(lv Level) {
+func SetLevel(lv Level) {
 	loadLogger().atom.SetLevel(lv.zapLevel())
 }