-
Notifications
You must be signed in to change notification settings - Fork 1
/
log.go
73 lines (61 loc) · 1.33 KB
/
log.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
package avl
import (
"os"
"github.com/rs/zerolog"
)
var (
nilLog zerolog.Logger = zerolog.Nop()
log zerolog.Logger = SetDefaultLog()
)
// SetDefaultLog returns the predefined(default) zerolog.Logger.
func SetDefaultLog() zerolog.Logger {
if os.Getenv("AVL_DEBUG") != "1" {
return zerolog.Nop()
}
return zerolog.
New(os.Stderr).
With().
Timestamp().
Caller().
Stack().
Logger().
Level(zerolog.DebugLevel)
}
// Logger is basic log for this package.
type Logger struct {
root zerolog.Logger
l *zerolog.Logger
contextFunc []func(zerolog.Context) zerolog.Context
}
// NewLogger returns new Logger. With argument function you can pass the context
// to Logger. For example,
//
// logger := NewLogger(func(c zerolog.Context) zerolog.Context {
// return c.Str("module", "avl_tree_validator")
// })
func NewLogger(cf func(zerolog.Context) zerolog.Context) *Logger {
zl := &Logger{}
if cf != nil {
zl.contextFunc = append(zl.contextFunc, cf)
}
return zl
}
// SetLogger set the new zerolog.Logger.
func (zl *Logger) SetLogger(l zerolog.Logger) *Logger {
zl.root = l
if len(zl.contextFunc) > 0 {
for _, cf := range zl.contextFunc {
l = cf(l.With()).Logger()
}
zl.l = &l
} else {
zl.l = &l
}
return zl
}
func (zl *Logger) Log() *zerolog.Logger {
if zl.l == nil {
return &nilLog
}
return zl.l
}