slogutil: imp code

AdguardTeam · Oct 18, 2024 · da3d5e0 · da3d5e0
1 parent b50b124
commit da3d5e0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/logutil/slogutil/slogutil.go b/logutil/slogutil/slogutil.go
@@ -76,7 +76,7 @@ func New(c *Config) (l *slog.Logger) {
 	case FormatJSONL:
 		h = NewJSONHybridHandler(output, &slog.HandlerOptions{
 			Level:       lvl,
-			ReplaceAttr: newJSONLReplaceAttr(),
+			ReplaceAttr: newJSONLReplaceAttr(!c.AddTimestamp),
 		})
 	case FormatText:
 		h = slog.NewTextHandler(output, &slog.HandlerOptions{
@@ -141,7 +141,11 @@ func ReplaceLevel(groups []string, a slog.Attr) (res slog.Attr) {
 
 // newJSONLReplaceAttr is a function that returns
 // [slog.HandlerOptions.ReplaceAttr] function for [FormatJSONHybrid] format.
-func newJSONLReplaceAttr() func(groups []string, a slog.Attr) (res slog.Attr) {
+func newJSONLReplaceAttr(removeTime bool) func(groups []string, a slog.Attr) (res slog.Attr) {
+	if !removeTime {
+		return SimplifyLevel
+	}
+
 	return func(groups []string, a slog.Attr) (res slog.Attr) {
 		return SimplifyLevel(groups, RemoveTime(groups, a))
 	}