40 lines
849 B
Go
40 lines
849 B
Go
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
"gorm.io/gorm/logger"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DBLogger struct {
|
||
|
|
Logger zerolog.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
func (D DBLogger) LogMode(_ logger.LogLevel) logger.Interface {
|
||
|
|
return D
|
||
|
|
}
|
||
|
|
|
||
|
|
func (D DBLogger) Info(ctx context.Context, s string, i ...interface{}) {
|
||
|
|
D.Logger.Info().Ctx(ctx).Msgf(s, i...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (D DBLogger) Warn(ctx context.Context, s string, i ...interface{}) {
|
||
|
|
D.Logger.Warn().Ctx(ctx).Msgf(s, i...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (D DBLogger) Error(ctx context.Context, s string, i ...interface{}) {
|
||
|
|
D.Logger.Error().Ctx(ctx).Msgf(s, i...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (D DBLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
|
||
|
|
tupl, rows := fc()
|
||
|
|
D.Logger.Trace().Ctx(ctx).
|
||
|
|
Time("begin", begin).
|
||
|
|
Err(err).
|
||
|
|
Str("sql", tupl).
|
||
|
|
Int64("rows", rows).
|
||
|
|
Msg("Query performed")
|
||
|
|
|
||
|
|
}
|