/
context.go
100 lines (86 loc) · 2.36 KB
/
context.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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package zc
import (
"context"
"go.uber.org/zap"
)
type loggerContextMark struct{}
var (
defaultLogger *zap.Logger
defaultSugaredLogger *zap.SugaredLogger
nopLogger = zap.NewNop()
nopSugar = zap.NewNop().Sugar()
)
// SetDefaultLogger set logger, which will return with L and S function if no logger in context.
// It is NOT thread safe and no syncronization with L and S
func SetDefaultLogger(logger *zap.Logger) {
defaultLogger = logger
if logger == nil {
defaultSugaredLogger = nil
} else {
defaultSugaredLogger = logger.Sugar()
}
}
// Return copy of ctx with associated logger
func WithLogger(ctx context.Context, l *zap.Logger) context.Context {
return context.WithValue(ctx, loggerContextMark{}, l)
}
// Return copy of ctx with associated logger
func WithSugarLogger(ctx context.Context, l *zap.SugaredLogger) context.Context {
return context.WithValue(ctx, loggerContextMark{}, l)
}
// Return associated logger. Default logger if no ctx logger yet. (default of default is nil)
func L(ctx context.Context) *zap.Logger {
if ctx == nil {
return defaultLogger
}
val := ctx.Value(loggerContextMark{})
if logger, ok := val.(*zap.Logger); ok {
return logger
}
if logger, ok := val.(*zap.SugaredLogger); ok {
return logger.Desugar()
}
return defaultLogger
}
// Return associated logger. Nop if no ctx logger yet.
func LNop(ctx context.Context) (res *zap.Logger) {
if ctx == nil {
return nopLogger
}
val := ctx.Value(loggerContextMark{})
if logger, ok := val.(*zap.Logger); ok {
return logger
}
if logger, ok := val.(*zap.SugaredLogger); ok {
return logger.Desugar()
}
return nopLogger
}
// Return associated logger. Default logger if no ctx logger yet. (default of default is nil)
func S(ctx context.Context) *zap.SugaredLogger {
if ctx == nil {
return defaultSugaredLogger
}
res := ctx.Value(loggerContextMark{})
if logger, ok := res.(*zap.SugaredLogger); ok {
return logger
}
if logger, ok := res.(*zap.Logger); ok {
return logger.Sugar()
}
return defaultSugaredLogger
}
// Return associated logger. Nop if no ctx logger yet.
func SNop(ctx context.Context) *zap.SugaredLogger {
if ctx == nil {
return nopSugar
}
res := ctx.Value(loggerContextMark{})
if logger, ok := res.(*zap.SugaredLogger); ok {
return logger
}
if logger, ok := res.(*zap.Logger); ok {
return logger.Sugar()
}
return nopSugar
}