Added logger caller in debug mode

F1bonacc1 · Mar 25, 2024 · 2160723 · 2160723
1 parent 118a3aa
commit 2160723
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/cmd/root.go b/src/cmd/root.go
@@ -111,7 +111,12 @@ func setupLogger() *os.File {
 	}
 	writer := zerolog.MultiLevelWriter(writers...)
 
-	log.Logger = zerolog.New(writer).With().Timestamp().Logger()
+	// add caller only in debug mode
+	if os.Getenv("PC_DEBUG") != "" {
+		log.Logger = zerolog.New(writer).With().Timestamp().Caller().Logger()
+	} else {
+		log.Logger = zerolog.New(writer).With().Timestamp().Logger()
+	}
 	zerolog.TimeFieldFormat = time.RFC3339Nano
 	zerolog.SetGlobalLevel(zerolog.DebugLevel)
 	return file