Merge branch 'master' into master

main.go

@@ -528,12 +528,12 @@ func (s *DB) Debug() *DB {
 // Transaction start a transaction as a block,
 // return error will rollback, otherwise to commit.
 func (s *DB) Transaction(fc func(tx *DB) error) (err error) {
+	panicked := true
 	tx := s.Begin()
 	defer func() {
-		if r := recover(); r != nil {
+		// Make sure to rollback when panic, Block error or Commit error
-			err = fmt.Errorf("%s", r)
+		if panicked || err != nil {
 			tx.Rollback()
-			return
 		}
 	}()
 
@@ -543,10 +543,7 @@ func (s *DB) Transaction(fc func(tx *DB) error) (err error) {
 		err = tx.Commit().Error
 	}
 
-	// Makesure rollback when Block error or Commit error
+	panicked = false
-	if err != nil {
-		tx.Rollback()
-	}
 	return
 }
 

main_test.go

@@ -470,6 +470,15 @@ func TestTransaction(t *testing.T) {
 	}
 }
 
+func assertPanic(t *testing.T, f func()) {
+	defer func() {
+		if r := recover(); r == nil {
+			t.Errorf("The code did not panic")
+		}
+	}()
+	f()
+}
+
 func TestTransactionWithBlock(t *testing.T) {
 	// rollback
 	err := DB.Transaction(func(tx *gorm.DB) error {
@@ -511,6 +520,7 @@ func TestTransactionWithBlock(t *testing.T) {
 	}
 
 	// panic will rollback
+	assertPanic(t, func() {
 		DB.Transaction(func(tx *gorm.DB) error {
 			u3 := User{Name: "transcation-3"}
 			if err := tx.Save(&u3).Error; err != nil {
@@ -523,6 +533,7 @@ func TestTransactionWithBlock(t *testing.T) {
 
 			panic("force panic")
 		})
+	})
 
 	if err := DB.First(&User{}, "name = ?", "transcation").Error; err == nil {
 		t.Errorf("Should not find record after panic rollback")