add golog.IsEnabledFor()

Author: keakon

log/log.go

@@ -45,6 +45,9 @@ func SetLogfFunc(f func(msg string, args ...interface{}), level golog.Level) {
 // This function should be called before using below functions.
 func SetDefaultLogger(l *golog.Logger) {
 	defaultLogger = l
+	if l == nil {
+		return
+	}
 	minLevel := l.GetMinLevel()
 	for level := golog.DebugLevel; level <= golog.CritLevel; level++ {
 		if level < minLevel {
@@ -87,6 +90,11 @@ func SetDefaultLogger(l *golog.Logger) {
 	}
 }
 
+// IsEnabledFor returns whether the default logger is enabled for the level.
+func IsEnabledFor(level golog.Level) bool {
+	return defaultLogger != nil && defaultLogger.IsEnabledFor(level)
+}
+
 func nop(args ...interface{})              {}
 func nopf(msg string, args ...interface{}) {}
 

log/log_test.go

@@ -156,6 +156,25 @@ func TestLogFuncs(t *testing.T) {
 	l.Close()
 }
 
+func TestIsEnabledFor(t *testing.T) {
+	SetDefaultLogger(nil)
+	if IsEnabledFor(golog.DebugLevel) {
+		t.Error("nil logger is enabled for debug level")
+	}
+
+	l := golog.NewStdoutLogger()
+	SetDefaultLogger(l)
+	if IsEnabledFor(golog.DebugLevel) {
+		t.Error("info logger is enabled for debug level")
+	}
+	if !IsEnabledFor(golog.InfoLevel) {
+		t.Error("info logger is disabled for info level")
+	}
+	if !IsEnabledFor(golog.ErrorLevel) {
+		t.Error("info logger is disabled for error level")
+	}
+}
+
 func BenchmarkDiscardLogger(b *testing.B) {
 	golog.StartFastTimer()
 	defer golog.StopFastTimer()

logger.go

@@ -71,7 +71,7 @@ func (l *Logger) AddHandler(h *Handler) {
 	}
 }
 
-// IsEnabledFor returns whether it's enabled for the level
+// IsEnabledFor returns whether it's enabled for the level.
 func (l *Logger) IsEnabledFor(level Level) bool {
 	return l.minLevel <= level
 }