forked from mirror/gorm
fix: trx in hooks clone stmt (#5338)
* fix: trx in hooks * chore: format by gofumpt
This commit is contained in:
parent
f5e77aab2f
commit
7496c3a56e
|
@ -589,8 +589,7 @@ func (db *DB) Transaction(fc func(tx *DB) error, opts ...*sql.TxOptions) (err er
|
|||
}
|
||||
}()
|
||||
}
|
||||
|
||||
err = fc(db.Session(&Session{}))
|
||||
err = fc(db.Session(&Session{NewDB: db.clone == 1}))
|
||||
} else {
|
||||
tx := db.Begin(opts...)
|
||||
if tx.Error != nil {
|
||||
|
|
|
@ -367,3 +367,33 @@ func TestTransactionOnClosedConn(t *testing.T) {
|
|||
t.Errorf("should returns error when commit with closed conn, got error %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTransactionWithHooks(t *testing.T) {
|
||||
user := GetUser("tTestTransactionWithHooks", Config{Account: true})
|
||||
DB.Create(&user)
|
||||
|
||||
var err error
|
||||
err = DB.Transaction(func(tx *gorm.DB) error {
|
||||
return tx.Model(&User{}).Limit(1).Transaction(func(tx2 *gorm.DB) error {
|
||||
return tx2.Scan(&User{}).Error
|
||||
})
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
// method with hooks
|
||||
err = DB.Transaction(func(tx1 *gorm.DB) error {
|
||||
// callMethod do
|
||||
tx2 := tx1.Find(&User{}).Session(&gorm.Session{NewDB: true})
|
||||
// trx in hooks
|
||||
return tx2.Transaction(func(tx3 *gorm.DB) error {
|
||||
return tx3.Where("user_id", user.ID).Delete(&Account{}).Error
|
||||
})
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue