forked from go-kit/kit
/
zap_sugar_logger.go
40 lines (36 loc) · 967 Bytes
/
zap_sugar_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
package zap
import (
"github.com/go-kit/kit/log"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
type zapSugarLogger func(msg string, keysAndValues ...interface{})
func (l zapSugarLogger) Log(kv ...interface{}) error {
l("", kv...)
return nil
}
// NewZapSugarLogger returns a Go kit log.Logger that sends
// log events to a zap.Logger.
func NewZapSugarLogger(logger *zap.Logger, level zapcore.Level) log.Logger {
sugarLogger := logger.WithOptions(zap.AddCallerSkip(2)).Sugar()
var sugar zapSugarLogger
switch level {
case zapcore.DebugLevel:
sugar = sugarLogger.Debugw
case zapcore.InfoLevel:
sugar = sugarLogger.Infow
case zapcore.WarnLevel:
sugar = sugarLogger.Warnw
case zapcore.ErrorLevel:
sugar = sugarLogger.Errorw
case zapcore.DPanicLevel:
sugar = sugarLogger.DPanicw
case zapcore.PanicLevel:
sugar = sugarLogger.Panicw
case zapcore.FatalLevel:
sugar = sugarLogger.Fatalw
default:
sugar = sugarLogger.Infow
}
return sugar
}