gorm事务默认是开启的。为了确保数据一致性,GORM 会在事务里执行写入操作(创建、更新、删除)。
如果没有这方面的要求,您可以在初始化时禁用它,这将获得大约 30%+ 性能提升。
一般不推荐禁用
// 全局禁用 db, err := gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{ SkipDefaultTransaction: true, })
普通事务
type TMG struct { ID uint Name string } func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) db.Transaction(func(tx *gorm.DB) error { // 在事务中执行一些 db 操作(从这里开始,您应该使用 'tx' 而不是 'db') if err := tx.Create(&TMG{Name: "Giraffe"}).Error; err != nil { // 返回任何错误都会回滚事务 return err } if err := tx.Create(&TMG{Name: "Lion"}).Error; err != nil { return err } // 返回 nil 提交事务 return nil }) }
执行下面代码,其中执行返回了一个test for err的错误,此时事务自动回滚,数据表中无数据
type TMG struct { ID uint Name string } func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) flag := false err := db.Transaction(func(tx *gorm.DB) error { // 在事务中执行一些 db 操作(从这里开始,您应该使用 'tx' 而不是 'db') if err := tx.Create(&TMG{Name: "Giraffe"}).Error; err != nil { // 返回任何错误都会回滚事务 return err } if !flag { return errors.New("test for err") } if err := tx.Create(&TMG{Name: "Lion"}).Error; err != nil { return err } // 返回 nil 提交事务 return nil }) if err != nil { fmt.Println(err)//test for err } }
嵌套事务
GORM 支持嵌套事务,您可以回滚较大事务内执行的一部分操作,例如:
在下面代码中,我们知道包含name2的事务返回了err,是会回滚的,而外层事务返回nil,则会提交
func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) db.Transaction(func(tx *gorm.DB) error { tx.Create(&TMG{Name: "name 1"}) tx.Transaction(func(tx2 *gorm.DB) error { tx2.Create(&TMG{Name: "name 2"}) return errors.New("rollback user2") // Rollback user2 }) tx.Transaction(func(tx2 *gorm.DB) error { tx2.Create(&TMG{Name: "name 3"}) return nil }) return nil }) }
手动事务
func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) // 开始事务 tx := db.Begin() // 在事务中执行一些 db 操作(从这里开始,您应该使用 'tx' 而不是 'db') tx.Create(&TMG{Name: "wxf1"}) tx.Create(&TMG{Name: "wxf2"}) tx.Create(&TMG{Name: "wxf3"}) // ... ... if err!=nil{ // 遇到错误时回滚事务 fmt.Println(tx.Rollback().Error) } ... // 否则,提交事务 fmt.Println(tx.Commit().Error) }
一个特殊的示例
type TMG struct { ID uint Name string } func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) // 事务 CreateTMG(db) } func CreateTMG(db *gorm.DB) error { // 再唠叨一下,事务一旦开始,你就应该使用 tx 处理数据 tx := db.Begin() defer func() { if r := recover(); r != nil { tx.Rollback() } }() if err := tx.Error; err != nil { return err } if err := tx.Create(&TMG{Name: "Giraffe"}).Error; err != nil { tx.Rollback() return err } if err := tx.Create(&TMG{Name: "Lion"}).Error; err != nil { tx.Rollback() return err } return tx.Commit().Error }
SavePoint、RollbackTo
GORM 提供了 SavePoint、Rollbackto 方法,来提供保存点以及回滚至保存点功能
这里rollback到了sp1的位置,也就是说,数据库中只存了wxf666这条数据
func main() { db, _ := gorm.Open(mysql.New(mysql.Config{DSN: "root:123456@tcp(127.0.0.1:3306)/gormDB?charset=utf8mb4&parseTime=True&loc=Local"}), &gorm.Config{}) db.AutoMigrate(&TMG{}) // 事务 tx := db.Begin() tx.Create(&TMG{Name: "wxf666"}) tx.SavePoint("sp1") tx.Create(&TMG{Name: "wxf777"}) tx.Create(&TMG{Name: "wxf888"}) tx.SavePoint("sp2") tx.RollbackTo("sp1") tx.Commit() }