-
Notifications
You must be signed in to change notification settings - Fork 3
/
logger.go
63 lines (53 loc) · 1.49 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
package keeltemporal
import (
"fmt"
"strings"
"go.uber.org/zap"
)
type logger struct {
l *zap.Logger
}
func NewLogger(l *zap.Logger) *logger {
return &logger{
l: l.WithOptions(zap.AddCallerSkip(3)),
}
}
func (t *logger) Debug(msg string, keyvals ...interface{}) {
t.l.Debug(msg, t.fields(keyvals...)...)
}
func (t *logger) Info(msg string, keyvals ...interface{}) {
// TODO check with temporal why errors are being logged as info!
for _, keyval := range keyvals {
if keyval == "Error" {
t.l.Error(msg, t.fields(keyvals...)...)
return
}
}
t.l.Info(msg, t.fields(keyvals...)...)
}
func (t *logger) Warn(msg string, keyvals ...interface{}) {
t.l.Warn(msg, t.fields(keyvals...)...)
}
func (t *logger) Error(msg string, keyvals ...interface{}) {
t.l.Error(msg, t.fields(keyvals...)...)
}
func (t *logger) With(keyvals ...interface{}) *logger {
return NewLogger(t.l.With(t.fields(keyvals...)...))
}
func (t *logger) fields(keyvals ...interface{}) []zap.Field {
var fields []zap.Field
for i := 0; i < len(keyvals); i++ {
if value, ok := keyvals[i].(zap.Field); ok {
fields = append(fields, value)
} else if value, ok := keyvals[i].(string); ok && len(keyvals) > i+2 {
fields = append(fields, zap.Any(strings.ToLower(value), keyvals[i+1]))
i++
} else if len(keyvals) > i+1 {
fields = append(fields, zap.Any(strings.ToLower(fmt.Sprintf("%v", keyvals[i])), keyvals[i+1]))
i++
} else {
fields = append(fields, zap.Any("undefined", keyvals[i]))
}
}
return fields
}