Add zap log adapter
This commit is contained in:
@@ -57,6 +57,7 @@ install:
|
|||||||
- go get -u github.com/satori/go.uuid
|
- go get -u github.com/satori/go.uuid
|
||||||
- go get -u github.com/sirupsen/logrus
|
- go get -u github.com/sirupsen/logrus
|
||||||
- go get -u github.com/pkg/errors
|
- go get -u github.com/pkg/errors
|
||||||
|
- go get -u go.uber.org/zap
|
||||||
|
|
||||||
script:
|
script:
|
||||||
- go test -v -race ./...
|
- go test -v -race ./...
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
// Package zapadapter provides a logger that writes to a go.uber.org/zap.Logger.
|
||||||
|
package zapadapter
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/jackc/pgx"
|
||||||
|
"go.uber.org/zap"
|
||||||
|
"go.uber.org/zap/zapcore"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Logger struct {
|
||||||
|
logger *zap.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLogger(logger *zap.Logger) *Logger {
|
||||||
|
return &Logger{logger: logger.WithOptions(zap.AddCallerSkip(1))}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pl *Logger) Log(level pgx.LogLevel, msg string, data map[string]interface{}) {
|
||||||
|
fields := make([]zapcore.Field, len(data))
|
||||||
|
i := 0
|
||||||
|
for k, v := range data {
|
||||||
|
fields[i] = zap.Reflect(k, v)
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
|
||||||
|
switch level {
|
||||||
|
case pgx.LogLevelTrace:
|
||||||
|
pl.logger.Debug(msg, append(fields, zap.Stringer("PGX_LOG_LEVEL", level))...)
|
||||||
|
case pgx.LogLevelDebug:
|
||||||
|
pl.logger.Debug(msg, fields...)
|
||||||
|
case pgx.LogLevelInfo:
|
||||||
|
pl.logger.Info(msg, fields...)
|
||||||
|
case pgx.LogLevelWarn:
|
||||||
|
pl.logger.Warn(msg, fields...)
|
||||||
|
case pgx.LogLevelError:
|
||||||
|
pl.logger.Error(msg, fields...)
|
||||||
|
default:
|
||||||
|
pl.logger.Error(msg, append(fields, zap.Stringer("PGX_LOG_LEVEL", level))...)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user