65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
type gormLogger struct {
|
||
|
|
logger zerolog.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
func newGormLogger(zerologger zerolog.Logger) *gormLogger {
|
||
|
|
return &gormLogger{zerologger}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *gormLogger) LogMode(newLevel logger.LogLevel) logger.Interface {
|
||
|
|
switch newLevel {
|
||
|
|
case logger.Error:
|
||
|
|
g.logger = g.logger.Level(zerolog.ErrorLevel)
|
||
|
|
case logger.Warn:
|
||
|
|
g.logger = g.logger.Level(zerolog.WarnLevel)
|
||
|
|
case logger.Info:
|
||
|
|
g.logger = g.logger.Level(zerolog.InfoLevel)
|
||
|
|
case logger.Silent:
|
||
|
|
g.logger = g.logger.Level(zerolog.Disabled)
|
||
|
|
}
|
||
|
|
return g
|
||
|
|
}
|
||
|
|
func (g *gormLogger) Info(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
g.logger.Info().Ctx(ctx).Msgf(format, args...)
|
||
|
|
}
|
||
|
|
func (g *gormLogger) Warn(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
g.logger.Warn().Ctx(ctx).Msgf(format, args...)
|
||
|
|
}
|
||
|
|
func (g *gormLogger) Error(ctx context.Context, format string, args ...interface{}) {
|
||
|
|
g.logger.Error().Ctx(ctx).Msgf(format, args...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *gormLogger) Trace(
|
||
|
|
ctx context.Context,
|
||
|
|
begin time.Time,
|
||
|
|
fc func() (sql string, rowsAffected int64),
|
||
|
|
err error,
|
||
|
|
) {
|
||
|
|
sql, rowsAffected := fc()
|
||
|
|
g.logger.Trace().
|
||
|
|
Ctx(ctx).
|
||
|
|
Time("gorm-begin", begin).
|
||
|
|
Err(err).
|
||
|
|
Str("gorm-query", sql).
|
||
|
|
Int64("gorm-rows-affected", rowsAffected).
|
||
|
|
Send()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *gormLogger) OverwriteLoggingLevel(new zerolog.Level) {
|
||
|
|
g.logger = g.logger.Level(new)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (g *gormLogger) OverwriteLogger(new zerolog.Logger) {
|
||
|
|
g.logger = new
|
||
|
|
}
|