feat: make stacktrace of the logger configurable

snapp-incubator · Sep 10, 2024 · 06da692 · 06da692
1 parent c16fed2
commit 06da692
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/internal/logger/logger.go b/internal/logger/logger.go
@@ -9,7 +9,8 @@ import (
 )
 
 type Config struct {
-	Level string `json:"level,omitempty" koanf:"level"`
+	Level      string `json:"level,omitempty" koanf:"level"`
+	Stacktrace bool   `json:"stacktrace,omitempty" koanf:"stacktrace"`
 }
 
 // New creates a zap logger for console.
@@ -28,7 +29,12 @@ func New(cfg Config) *zap.Logger {
 	}
 
 	core := zapcore.NewTee(cores...)
-	logger := zap.New(core, zap.AddCaller(), zap.AddStacktrace(zap.ErrorLevel))
+	var zapOpts = make([]zap.Option, 0, 2)
+	zapOpts = append(zapOpts, zap.AddCaller())
+	if cfg.Stacktrace {
+		zapOpts = append(zapOpts, zap.AddStacktrace(zap.ErrorLevel))
+	}
+	logger := zap.New(core, zapOpts...)
 
 	return logger
 }