-
Notifications
You must be signed in to change notification settings - Fork 0
/
collector.go
137 lines (112 loc) · 3.85 KB
/
collector.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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package infralog
import (
"context"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
// handlers is a map of handlers for different levels
var handlers []func(entry *LogEntry)
func Debug(message string, fields ...zap.Field) {
handleEntry(&LogEntry{Level: zapcore.DebugLevel, Message: message, Fields: fields})
}
func DebugCtx(ctx context.Context, message string, fields ...zap.Field) {
ctxFields := fieldsFromContext(ctx)
fields = append(ctxFields, fields...)
handleEntry(&LogEntry{Level: zapcore.DebugLevel, Message: message, Fields: fields})
}
func Info(message string, fields ...zap.Field) {
handleEntry(&LogEntry{Level: zapcore.InfoLevel, Message: message, Fields: fields})
}
func InfoCtx(ctx context.Context, message string, fields ...zap.Field) {
ctxFields := fieldsFromContext(ctx)
fields = append(ctxFields, fields...)
handleEntry(&LogEntry{Level: zapcore.InfoLevel, Message: message, Fields: fields})
}
func Warn(message string, fields ...zap.Field) {
handleEntry(&LogEntry{Level: zapcore.WarnLevel, Message: message, Fields: fields})
}
func WarnCtx(ctx context.Context, message string, fields ...zap.Field) {
ctxFields := fieldsFromContext(ctx)
fields = append(ctxFields, fields...)
handleEntry(&LogEntry{Level: zapcore.WarnLevel, Message: message, Fields: fields})
}
func Error(message string, fields ...zap.Field) {
handleEntry(&LogEntry{Level: zapcore.ErrorLevel, Message: message, Fields: fields})
}
func ErrorCtx(ctx context.Context, message string, fields ...zap.Field) {
ctxFields := fieldsFromContext(ctx)
fields = append(ctxFields, fields...)
handleEntry(&LogEntry{Level: zapcore.ErrorLevel, Message: message, Fields: fields})
}
func Fatal(message string, fields ...zap.Field) {
entry := &LogEntry{Level: zapcore.FatalLevel, Message: message, Fields: fields}
handleEntry(entry)
// in case if there are no fatal handlers, we exit with panic.
// should be unreachable
panic(entry)
}
func FatalCtx(ctx context.Context, message string, fields ...zap.Field) {
ctxFields := fieldsFromContext(ctx)
fields = append(ctxFields, fields...)
entry := &LogEntry{Level: zapcore.FatalLevel, Message: message, Fields: fields}
handleEntry(entry)
// in case if there are no fatal handlers, we exit with panic.
// should be unreachable
panic(entry)
}
type fieldsCtxKeyType string
const fieldsCtxKey fieldsCtxKeyType = "fields"
type logFields struct {
fields []zap.Field
}
func fieldsFromContext(ctx context.Context) []zap.Field {
if ctxFields, ok := ctx.Value(fieldsCtxKey).(*logFields); ok {
return ctxFields.fields
}
return nil
}
func WithField(ctx context.Context, field zap.Field) context.Context {
ctxFields, ok := ctx.Value(fieldsCtxKey).(*logFields)
if !ok || ctxFields == nil {
ctxFields = &logFields{}
ctx = context.WithValue(ctx, fieldsCtxKey, ctxFields)
}
ctxFields.fields = append(ctxFields.fields, field)
return ctx
}
func WithFields(ctx context.Context, fields ...zap.Field) context.Context {
ctxFields := ctx.Value(fieldsCtxKey).(*logFields)
if ctxFields == nil {
ctxFields = &logFields{}
ctx = context.WithValue(ctx, fieldsCtxKey, ctxFields)
}
ctxFields.fields = append(ctxFields.fields, fields...)
return ctx
}
func RegisterLogHandler(handler func(entry *LogEntry)) {
// add handler to the beginning of the handlers list to
// make it the first one to be called
handlers = append([]func(entry *LogEntry){handler}, handlers...)
}
func handleEntry(entry *LogEntry) {
for _, handler := range handlers {
handler(entry)
}
}
type LogEntry struct {
Level zapcore.Level
Message string
Fields []zap.Field
}
// Error searches the first error in the list of entry's fields.
// Used to extract error from the log entry. We usually have only one error.
func (e *LogEntry) Error() error {
for i := range e.Fields {
if e.Fields[i].Type == zapcore.ErrorType {
if err, ok := e.Fields[i].Interface.(error); ok {
return err
}
}
}
return nil
}