mirror of https://github.com/jackc/pgx.git
43 lines
1009 B
Go
43 lines
1009 B
Go
// Package zerologadapter provides a logger that writes to a github.com/rs/zerolog.
|
|
package zerologadapter
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v4"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type Logger struct {
|
|
logger zerolog.Logger
|
|
}
|
|
|
|
// NewLogger accepts a zerolog.Logger as input and returns a new custom pgx
|
|
// logging fascade as output.
|
|
func NewLogger(logger zerolog.Logger) *Logger {
|
|
return &Logger{
|
|
logger: logger.With().Str("module", "pgx").Logger(),
|
|
}
|
|
}
|
|
|
|
func (pl *Logger) Log(ctx context.Context, level pgx.LogLevel, msg string, data map[string]interface{}) {
|
|
var zlevel zerolog.Level
|
|
switch level {
|
|
case pgx.LogLevelNone:
|
|
zlevel = zerolog.NoLevel
|
|
case pgx.LogLevelError:
|
|
zlevel = zerolog.ErrorLevel
|
|
case pgx.LogLevelWarn:
|
|
zlevel = zerolog.WarnLevel
|
|
case pgx.LogLevelInfo:
|
|
zlevel = zerolog.InfoLevel
|
|
case pgx.LogLevelDebug:
|
|
zlevel = zerolog.DebugLevel
|
|
default:
|
|
zlevel = zerolog.DebugLevel
|
|
}
|
|
|
|
pgxlog := pl.logger.With().Fields(data).Logger()
|
|
pgxlog.WithLevel(zlevel).Msg(msg)
|
|
}
|