heimoshuiyu/gonic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

use db.Transaction helper for migrations

Browse Source
This commit is contained in:
sentriz
2023-10-09 23:31:51 +01:00
parent 219aa747c9
commit 422a4b88d7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
db/migrations.go
View File

@@ -74,10 +74,10 @@ func (db *DB) Migrate(ctx MigrationContext) error {
func construct(ctx MigrationContext, id string, f func(*gorm.DB, MigrationContext) error) *gormigrate.Migration { func construct(ctx MigrationContext, id string, f func(*gorm.DB, MigrationContext) error) *gormigrate.Migration {
return constructNoTx(ctx, id, func(db *gorm.DB, ctx MigrationContext) error { return constructNoTx(ctx, id, func(db *gorm.DB, ctx MigrationContext) error {
tx := db.Begin() return db.Transaction(func(tx *gorm.DB) error {
defer tx.Commit()
return f(tx, ctx) return f(tx, ctx)
}) })
})
} }
func constructNoTx(ctx MigrationContext, id string, f func(*gorm.DB, MigrationContext) error) *gormigrate.Migration { func constructNoTx(ctx MigrationContext, id string, f func(*gorm.DB, MigrationContext) error) *gormigrate.Migration {