19
19
// Note: The implementation comes from https://www.mountedthoughts.com/golang-logger-interface/
20
20
// https://github.com/amitrai48/logger
21
21
22
- package logger
22
+ package zap
23
23
24
24
import (
25
25
"os"
26
26
27
- "go.uber.org/zap"
27
+ "github.com/vmware/vmware-go-kcl/logger"
28
+ uzap "go.uber.org/zap"
28
29
"go.uber.org/zap/zapcore"
29
30
lumberjack "gopkg.in/natefinch/lumberjack.v2"
30
31
)
31
32
32
33
type ZapLogger struct {
33
- sugaredLogger * zap .SugaredLogger
34
+ sugaredLogger * uzap .SugaredLogger
34
35
}
35
36
36
37
// NewZapLogger adapts existing sugared zap logger to Logger interface.
@@ -44,15 +45,15 @@ type ZapLogger struct {
44
45
// Base zap logger can be convert to SugaredLogger by calling to add a wrapper:
45
46
// sugaredLogger := log.Sugar()
46
47
//
47
- func NewZapLogger (logger * zap .SugaredLogger ) Logger {
48
+ func NewZapLogger (logger * uzap .SugaredLogger ) logger. Logger {
48
49
return & ZapLogger {
49
50
sugaredLogger : logger ,
50
51
}
51
52
}
52
53
53
54
// NewZapLoggerWithConfig creates and configs Logger instance backed by
54
55
// zap Sugared logger.
55
- func NewZapLoggerWithConfig (config Configuration ) Logger {
56
+ func NewZapLoggerWithConfig (config logger. Configuration ) logger. Logger {
56
57
cores := []zapcore.Core {}
57
58
58
59
if config .EnableConsole {
@@ -80,9 +81,9 @@ func NewZapLoggerWithConfig(config Configuration) Logger {
80
81
81
82
// AddCallerSkip skips 2 number of callers, this is important else the file that gets
82
83
// logged will always be the wrapped file. In our case zap.go
83
- logger := zap .New (combinedCore ,
84
- zap .AddCallerSkip (2 ),
85
- zap .AddCaller (),
84
+ logger := uzap .New (combinedCore ,
85
+ uzap .AddCallerSkip (2 ),
86
+ uzap .AddCaller (),
86
87
).Sugar ()
87
88
88
89
return & ZapLogger {
@@ -114,7 +115,7 @@ func (l *ZapLogger) Panicf(format string, args ...interface{}) {
114
115
l .sugaredLogger .Fatalf (format , args ... )
115
116
}
116
117
117
- func (l * ZapLogger ) WithFields (fields Fields ) Logger {
118
+ func (l * ZapLogger ) WithFields (fields logger. Fields ) logger. Logger {
118
119
var f = make ([]interface {}, 0 )
119
120
for k , v := range fields {
120
121
f = append (f , k )
@@ -125,7 +126,7 @@ func (l *ZapLogger) WithFields(fields Fields) Logger {
125
126
}
126
127
127
128
func getEncoder (isJSON bool ) zapcore.Encoder {
128
- encoderConfig := zap .NewProductionEncoderConfig ()
129
+ encoderConfig := uzap .NewProductionEncoderConfig ()
129
130
encoderConfig .EncodeTime = zapcore .ISO8601TimeEncoder
130
131
if isJSON {
131
132
return zapcore .NewJSONEncoder (encoderConfig )
@@ -135,15 +136,15 @@ func getEncoder(isJSON bool) zapcore.Encoder {
135
136
136
137
func getZapLevel (level string ) zapcore.Level {
137
138
switch level {
138
- case Info :
139
+ case logger . Info :
139
140
return zapcore .InfoLevel
140
- case Warn :
141
+ case logger . Warn :
141
142
return zapcore .WarnLevel
142
- case Debug :
143
+ case logger . Debug :
143
144
return zapcore .DebugLevel
144
- case Error :
145
+ case logger . Error :
145
146
return zapcore .ErrorLevel
146
- case Fatal :
147
+ case logger . Fatal :
147
148
return zapcore .FatalLevel
148
149
default :
149
150
return zapcore .InfoLevel
0 commit comments