forked from go-kratos/kratos
/
helper_test.go
96 lines (82 loc) · 2.11 KB
/
helper_test.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
package log
import (
"context"
"io"
"os"
"testing"
)
func TestHelper(_ *testing.T) {
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
log := NewHelper(logger)
log.Log(LevelDebug, "msg", "test debug")
log.Debug("test debug")
log.Debugf("test %s", "debug")
log.Debugw("log", "test debug")
log.Warn("test warn")
log.Warnf("test %s", "warn")
log.Warnw("log", "test warn")
}
func TestHelperWithMsgKey(_ *testing.T) {
logger := With(DefaultLogger, "ts", DefaultTimestamp, "caller", DefaultCaller)
log := NewHelper(logger, WithMessageKey("message"))
log.Debugf("test %s", "debug")
log.Debugw("log", "test debug")
}
func TestHelperLevel(_ *testing.T) {
log := NewHelper(DefaultLogger)
log.Debug("test debug")
log.Info("test info")
log.Infof("test %s", "info")
log.Warn("test warn")
log.Error("test error")
log.Errorf("test %s", "error")
log.Errorw("log", "test error")
}
func BenchmarkHelperPrint(b *testing.B) {
log := NewHelper(NewStdLogger(io.Discard))
for i := 0; i < b.N; i++ {
log.Debug("test")
}
}
func BenchmarkHelperPrintFilterLevel(b *testing.B) {
log := NewHelper(NewFilter(NewStdLogger(io.Discard), FilterLevel(LevelDebug)))
for i := 0; i < b.N; i++ {
log.Debug("test")
}
}
func BenchmarkHelperPrintf(b *testing.B) {
log := NewHelper(NewStdLogger(io.Discard))
for i := 0; i < b.N; i++ {
log.Debugf("%s", "test")
}
}
func BenchmarkHelperPrintfFilterLevel(b *testing.B) {
log := NewHelper(NewFilter(NewStdLogger(io.Discard), FilterLevel(LevelInfo)))
for i := 0; i < b.N; i++ {
log.Debugf("%s", "test")
}
}
func BenchmarkHelperPrintw(b *testing.B) {
log := NewHelper(NewStdLogger(io.Discard))
for i := 0; i < b.N; i++ {
log.Debugw("key", "value")
}
}
type traceKey struct{}
func TestContext(_ *testing.T) {
logger := With(NewStdLogger(os.Stdout),
"trace", Trace(),
)
log := NewHelper(logger)
ctx := context.WithValue(context.Background(), traceKey{}, "2233")
log.WithContext(ctx).Info("got trace!")
}
func Trace() Valuer {
return func(ctx context.Context) interface{} {
s, ok := ctx.Value(traceKey{}).(string)
if !ok {
return nil
}
return s
}
}