fix(logger): properly SetGlobalLevel on all levels (#1504)

* fix(logger): properly SetGlobalLevel on all levels * utilize ParseLevel
autobrr · Apr 12, 2024 · b58ead7 · b58ead7
1 parent 5857945
commit b58ead7
Showing 1 changed file with 4 additions and 16 deletions.
diff --git a/internal/logger/logger.go b/internal/logger/logger.go
@@ -87,23 +87,11 @@ func (l *DefaultLogger) RegisterSSEWriter(sse *sse.Server) {
 }
 
 func (l *DefaultLogger) SetLogLevel(level string) {
-	switch level {
-	case "INFO":
-		l.level = zerolog.InfoLevel
-		zerolog.SetGlobalLevel(zerolog.InfoLevel)
-	case "DEBUG":
-		l.level = zerolog.DebugLevel
-		zerolog.SetGlobalLevel(zerolog.DebugLevel)
-	case "ERROR":
-		l.level = zerolog.ErrorLevel
-	case "WARN":
-		l.level = zerolog.WarnLevel
-	case "TRACE":
-		l.level = zerolog.TraceLevel
-		zerolog.SetGlobalLevel(zerolog.TraceLevel)
-	default:
-		l.level = zerolog.Disabled
+	lvl, err := zerolog.ParseLevel(level)
+	if err != nil {
+		lvl = zerolog.DebugLevel
 	}
+	zerolog.SetGlobalLevel(lvl)
 }
 
 // Log log something at fatal level.