feat(logger): add debug parameter for logger init

heyshaquib · Jan 2, 2025 · a8dceaf · a8dceaf
1 parent 40cad00
commit a8dceaf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/cmd/fsb/run.go b/cmd/fsb/run.go
@@ -27,7 +27,7 @@ var runCmd = &cobra.Command{
 var startTime time.Time = time.Now()
 
 func runApp(cmd *cobra.Command, args []string) {
-	utils.InitLogger()
+	utils.InitLogger(config.ValueOf.Dev)
 	log := utils.Logger
 	mainLogger := log.Named("Main")
 	mainLogger.Info("Starting server")

diff --git a/internal/utils/logger.go b/internal/utils/logger.go
@@ -11,7 +11,7 @@ import (
 
 var Logger *zap.Logger
 
-func InitLogger() {
+func InitLogger(debugMode bool) {
 	customTimeEncoder := func(t time.Time, enc zapcore.PrimitiveArrayEncoder) {
 		enc.AppendString(t.Format("02/01/2006 03:04 PM"))
 	}
@@ -32,8 +32,15 @@ func InitLogger() {
 		Compress:   true,
 	})
 
+	var consoleLevel zapcore.Level
+	if debugMode {
+		consoleLevel = zapcore.DebugLevel
+	} else {
+		consoleLevel = zapcore.InfoLevel
+	}
+
 	core := zapcore.NewTee(
-		zapcore.NewCore(consoleEncoder, zapcore.AddSync(os.Stdout), zapcore.InfoLevel),
+		zapcore.NewCore(consoleEncoder, zapcore.AddSync(os.Stdout), consoleLevel),
 		zapcore.NewCore(fileEncoder, fileWriter, zapcore.DebugLevel),
 	)