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…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user
	 Cr
						Cr