-
Notifications
You must be signed in to change notification settings - Fork 0
/
field_generator.go
42 lines (32 loc) · 1.12 KB
/
field_generator.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package clog
import (
"context"
"go.uber.org/zap/zapcore"
)
// FieldGenerator describes a function that generates zap fields for a given context
type FieldGenerator func(ctx context.Context) []zapcore.Field
// FieldGenerators retrieves the FieldGenerators stored on a context
func FieldGenerators(ctx context.Context) (fieldGenerators []FieldGenerator) {
if ctx != nil {
fieldGenerators, _ = ctx.Value(ctxFields).([]FieldGenerator)
} else {
fieldGenerators = []FieldGenerator{}
}
return fieldGenerators
}
// WithFieldGenerators adds the supplied FieldGenerator to those found on the supplied context and surfaces a new context
func WithFieldGenerators(ctx context.Context, generators ...FieldGenerator) context.Context {
if len(generators) == 0 {
return ctx
}
fs := FieldGenerators(ctx)
fs = append(make([]FieldGenerator, 0, len(fs)+len(generators)), fs...)
return context.WithValue(ctx, ctxFields, append(fs, generators...))
}
func generateFields(ctx context.Context) []zapcore.Field {
var fields []zapcore.Field
for _, gen := range FieldGenerators(ctx) {
fields = append(fields, gen(ctx)...)
}
return fields
}