Merge 33558ed56e9d6b9d2ea808bc46c0b77e7952d813 into 46bce170cae701615e2b2f8b2448b54524be9648

callbacks.go

@@ -79,19 +79,19 @@ func (cs *callbacks) Transaction() *processor {
 }
 
 func (p *processor) Begin(tx *DB, opt *sql.TxOptions) *DB {
-	var err error
+	// call scopes
-
+	for len(tx.Statement.scopes) > 0 {
-	switch beginner := tx.Statement.ConnPool.(type) {
+		scopes := tx.Statement.scopes
-	case TxBeginner:
+		tx.Statement.scopes = nil
-		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt)
+		for _, scope := range scopes {
-	case ConnPoolBeginner:
+			tx = scope(tx)
-		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt)
+		}
-	default:
-		err = ErrInvalidTransaction
 	}
 
-	if err != nil {
+	tx.InstanceSet("gorm:transaction_options", opt)
-		_ = tx.AddError(err)
+
+	for _, f := range p.fns {
+		f(tx)
 	}
 
 	return tx

callbacks/callbacks.go

@@ -80,4 +80,7 @@ func RegisterDefaultCallbacks(db *gorm.DB, config *Config) {
 	rawCallback := db.Callback().Raw()
 	rawCallback.Register("gorm:raw", RawExec)
 	rawCallback.Clauses = config.QueryClauses
+
+	transactionCallback := db.Callback().Transaction()
+	_ = transactionCallback.Register("gorm:begin", Begin)
 }

callbacks/transaction.go

@@ -1,6 +1,8 @@
 package callbacks
 
 import (
+	"database/sql"
+
 	"gorm.io/gorm"
 )
 
@@ -30,3 +32,33 @@ func CommitOrRollbackTransaction(db *gorm.DB) {
 		}
 	}
 }
+
+func Begin(tx *gorm.DB) {
+	err := tx.Error
+
+	if err != nil {
+		return
+	}
+
+	var opt *sql.TxOptions
+
+	if v, ok := tx.InstanceGet("gorm:transaction_options"); ok {
+		if txOpts, ok := v.(*sql.TxOptions); ok {
+			opt = txOpts
+		}
+	}
+
+	switch beginner := tx.Statement.ConnPool.(type) {
+	case gorm.TxBeginner:
+		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt)
+	case gorm.ConnPoolBeginner:
+		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt)
+	default:
+		err = gorm.ErrInvalidTransaction
+	}
+
+	if err != nil {
+		_ = tx.AddError(err)
+	}
+
+}