support adding fields to already exists fields (#13)

yuseferi · Nov 9, 2023 · 3109e14 · 3109e14
1 parent d142945
commit 3109e14
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/zax.go b/zax.go
@@ -15,6 +15,9 @@ const loggerKey = Key("zax")
 
 // Set Add passed fields in context
 func Set(ctx context.Context, fields []zap.Field) context.Context {
+	if loggerFields, ok := ctx.Value(loggerKey).([]zap.Field); ok {
+		fields = append(fields, loggerFields...)
+	}
 	return context.WithValue(ctx, loggerKey, fields)
 }