/
logger.go
105 lines (82 loc) · 2.43 KB
/
logger.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
101
102
103
104
105
package log
import (
"context"
"encoding/binary"
"go.opencensus.io/trace"
"go.uber.org/zap"
)
type logger struct {
logger *zap.SugaredLogger
}
func (l logger) WithField(key string, value interface{}) Logger {
return logger{l.logger.With(zap.Any(key, value))}
}
func (l logger) CheckWrite(lvl Level, msg string, fields ...Field) {
if ce := l.logger.Desugar().Check(lvl, msg); ce != nil {
ce.Write(fields...)
}
}
func (l logger) WithFields(fields Fields) Logger {
return logger{l.logger.Desugar().With(fields...).Sugar()}
}
func (l logger) WithError(err error) Logger {
return logger{l.logger.With(zap.Error(err))}
}
// WithTracing will take an OpenCensus trace and add log fields for Datadog.
// based on convertSpan in https://github.com/DataDog/opencensus-go-exporter-datadog/blob/master/span.go
// and https://docs.datadoghq.com/tracing/advanced/connect_logs_and_traces/?tab=go
func (l logger) WithTracing(ctx context.Context) Logger {
if span := trace.FromContext(ctx); span != nil {
traceID := span.SpanContext().TraceID
spanID := span.SpanContext().SpanID
return l.
WithField("dd.trace_id", binary.BigEndian.Uint64(traceID[8:])).
WithField("dd.span_id", binary.BigEndian.Uint64(spanID[:]))
}
return l
}
func (l logger) Debugf(format string, args ...interface{}) {
l.logger.Debugf(format, args...)
}
func (l logger) Infof(format string, args ...interface{}) {
l.logger.Infof(format, args...)
}
func (l logger) Warnf(format string, args ...interface{}) {
l.logger.Warnf(format, args...)
}
func (l logger) Warningf(format string, args ...interface{}) {
l.logger.Warnf(format, args...)
}
func (l logger) Errorf(format string, args ...interface{}) {
l.logger.Errorf(format, args...)
}
func (l logger) Fatalf(format string, args ...interface{}) {
l.logger.Fatalf(format, args...)
}
func (l logger) Panicf(format string, args ...interface{}) {
l.logger.Panicf(format, args...)
}
func (l logger) Debug(args ...interface{}) {
l.logger.Debug(args...)
}
func (l logger) Info(args ...interface{}) {
l.logger.Info(args...)
}
func (l logger) Warn(args ...interface{}) {
l.logger.Warn(args...)
}
func (l logger) Warning(args ...interface{}) {
l.logger.Warn(args...)
}
func (l logger) Error(args ...interface{}) {
l.logger.Error(args...)
}
func (l logger) Fatal(args ...interface{}) {
l.logger.Fatal(args...)
}
func (l logger) Panic(args ...interface{}) {
l.logger.Panic(args...)
}
func (l logger) Sync() error {
return l.logger.Sync()
}