use callback to handle transaction
- make transaction have before and after hooks, so plugin can have hack before or after transaction
This commit is contained in:
		
							parent
							
								
									a70af2a4c0
								
							
						
					
					
						commit
						93f28bc116
					
				
							
								
								
									
										37
									
								
								callbacks.go
									
									
									
									
									
								
							
							
						
						
									
										37
									
								
								callbacks.go
									
									
									
									
									
								
							| @ -2,6 +2,7 @@ package gorm | ||||
| 
 | ||||
| import ( | ||||
| 	"context" | ||||
| 	"database/sql" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"reflect" | ||||
| @ -15,12 +16,13 @@ import ( | ||||
| func initializeCallbacks(db *DB) *callbacks { | ||||
| 	return &callbacks{ | ||||
| 		processors: map[string]*processor{ | ||||
| 			"create": {db: db}, | ||||
| 			"query":  {db: db}, | ||||
| 			"update": {db: db}, | ||||
| 			"delete": {db: db}, | ||||
| 			"row":    {db: db}, | ||||
| 			"raw":    {db: db}, | ||||
| 			"create":      {db: db}, | ||||
| 			"query":       {db: db}, | ||||
| 			"update":      {db: db}, | ||||
| 			"delete":      {db: db}, | ||||
| 			"row":         {db: db}, | ||||
| 			"raw":         {db: db}, | ||||
| 			"transaction": {db: db}, | ||||
| 		}, | ||||
| 	} | ||||
| } | ||||
| @ -72,6 +74,29 @@ func (cs *callbacks) Raw() *processor { | ||||
| 	return cs.processors["raw"] | ||||
| } | ||||
| 
 | ||||
| func (cs *callbacks) Transaction() *processor { | ||||
| 	return cs.processors["transaction"] | ||||
| } | ||||
| 
 | ||||
| func (p *processor) Begin(tx *DB, opt *sql.TxOptions) *DB { | ||||
| 	var err error | ||||
| 
 | ||||
| 	switch beginner := tx.Statement.ConnPool.(type) { | ||||
| 	case TxBeginner: | ||||
| 		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt) | ||||
| 	case ConnPoolBeginner: | ||||
| 		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt) | ||||
| 	default: | ||||
| 		err = ErrInvalidTransaction | ||||
| 	} | ||||
| 
 | ||||
| 	if err != nil { | ||||
| 		tx.AddError(err) | ||||
| 	} | ||||
| 
 | ||||
| 	return tx | ||||
| } | ||||
| 
 | ||||
| func (p *processor) Execute(db *DB) *DB { | ||||
| 	// call scopes
 | ||||
| 	for len(db.Statement.scopes) > 0 { | ||||
|  | ||||
| @ -619,27 +619,13 @@ func (db *DB) Begin(opts ...*sql.TxOptions) *DB { | ||||
| 		// clone statement
 | ||||
| 		tx  = db.getInstance().Session(&Session{Context: db.Statement.Context, NewDB: db.clone == 1}) | ||||
| 		opt *sql.TxOptions | ||||
| 		err error | ||||
| 	) | ||||
| 
 | ||||
| 	if len(opts) > 0 { | ||||
| 		opt = opts[0] | ||||
| 	} | ||||
| 
 | ||||
| 	switch beginner := tx.Statement.ConnPool.(type) { | ||||
| 	case TxBeginner: | ||||
| 		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt) | ||||
| 	case ConnPoolBeginner: | ||||
| 		tx.Statement.ConnPool, err = beginner.BeginTx(tx.Statement.Context, opt) | ||||
| 	default: | ||||
| 		err = ErrInvalidTransaction | ||||
| 	} | ||||
| 
 | ||||
| 	if err != nil { | ||||
| 		tx.AddError(err) | ||||
| 	} | ||||
| 
 | ||||
| 	return tx | ||||
| 	return tx.callbacks.Transaction().Begin(tx, opt) | ||||
| } | ||||
| 
 | ||||
| // Commit commit a transaction
 | ||||
|  | ||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user
	 Joe
						Joe