package gormx

import (
	"context"
	"time"

	"gorm.io/gorm/logger"

	"github.com/yuninks/loggerx"
)

// TODO:待进一步封装

type GormxLogger struct {
	Logger loggerx.LoggerInterface
}

func NewGormxLogger(logger loggerx.LoggerInterface) *GormxLogger {
	return &GormxLogger{
		Logger: logger,
	}
}

func (g *GormxLogger) LogMode(LogLevel logger.LogLevel) logger.Interface {
	return g
}

func (g *GormxLogger) Info(ctx context.Context, msg string, val ...interface{}) {
	g.Logger.Info(ctx, val)
}

func (g *GormxLogger) Warn(ctx context.Context, msg string, val ...interface{}) {
	g.Logger.Info(ctx, val)
}

func (g *GormxLogger) Error(ctx context.Context, msg string, val ...interface{}) {
	g.Logger.Error(ctx, val)
}

func (g *GormxLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {

	sql, rows := fc()
	g.Logger.Infof(ctx, "trace:  begin:%+v, err:%+v, sql:%+v, rows:%+v", begin, err, sql, rows)

}