tablet/gorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update migrator.go

Browse Source
This commit is contained in:
Jinzhu 2022-12-24 12:06:26 +08:00 committed by GitHub
parent ac9262f736
commit 011eeccd21
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
migrator/migrator.go
View File

@ -32,15 +32,14 @@ type Config struct {
gorm.Dialector gorm.Dialector
} }
type printSQLLogger struct{} type printSQLLogger struct {
logger.Interface
}
func (l *printSQLLogger) LogMode(ll logger.LogLevel) logger.Interface { return l }
func (l *printSQLLogger) Info(ctx context.Context, log string, args ...interface{}) {}
func (l *printSQLLogger) Warn(ctx context.Context, log string, args ...interface{}) {}
func (l *printSQLLogger) Error(ctx context.Context, log string, args ...interface{}) {}
func (l *printSQLLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) { func (l *printSQLLogger) Trace(ctx context.Context, begin time.Time, fc func() (sql string, rowsAffected int64), err error) {
sql, _ := fc() sql, _ := fc()
fmt.Println(sql + ";") fmt.Println(sql + ";")
l.Interface.Trace(ctx, begin, fc, err)
} }
// GormDataTypeInterface gorm data type interface // GormDataTypeInterface gorm data type interface
@ -109,7 +108,7 @@ func (m Migrator) AutoMigrate(values ...interface{}) error {
execTx := queryTx execTx := queryTx
if m.DB.DryRun { if m.DB.DryRun {
queryTx.DryRun = false queryTx.DryRun = false
execTx = m.DB.Session(&gorm.Session{Logger: &printSQLLogger{}}) execTx = m.DB.Session(&gorm.Session{Logger: &printSQLLogger{Interface: m.DB.Logger}})
} }
if !queryTx.Migrator().HasTable(value) { if !queryTx.Migrator().HasTable(value) {
if err := execTx.Migrator().CreateTable(value); err != nil { if err := execTx.Migrator().CreateTable(value); err != nil {