GO语言-数据操作-GORM 中使用事务

943 阅读2分钟

GORM 中使用事务

事务处理可以用来维护数据库的完整性,保证成批的 SQL 语句要么全执行,要么全不执行。 事务的处理也很简单,用 db.Begin() 声明开启事务,结束的时候调用 tx.Commit(),异常的时候调用 tx.Rollback()

禁用默认事务

为了确保数据一致性,GORM 会在事务里执行写入操作(创建、更新、删除等)。如果没有这方面的要求,我们就可以在初始化时禁用它,这将提高大约 30%+ 性能提升。

dsn := "root:password@tcp(127.0.0.1:3306)/gin?charset=utf8mb4&parseTime=True&loc=Local"
    DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
        SkipDefaultTransaction: true, //打开禁用默认事务
    })
    DB.Debug()

GORM 默认会将单个的 create, update, delete 操作封装在事务内进行处理,以确保数据的完整性。 如果你想把多个 create, update, delete 操作作为一个原子操作,Transaction 就是用来完成这个的。

事务执行流程

要在事务中执行一系列操作,我们可以参照下面的流程来执行。

db.Transaction(func(tx *gorm.DB) error {
// 在事务中执行一些 db 操作(从这里开始,应该使用 'tx' 而不是'db')
    if err := tx.Create(&Animal{Name: "cxk"}).Error; err != nil {
// 返回任何错误都会回滚事务
return err
}
    if err := tx.Create(&Animal{Name: "ak"}).Error; err != nil {
        return err
    }
        // 返回 nil 提交事务
        return nil
})

手动控制事务

db.Begin() 声明开启事务,结束的时候调用 tx.Commit(),异常的时候调用 tx.Rollback(),我们通常使用判断语句来实现一些功能

// 开启事务
tx := db.Begin()
// 在事务中做一些数据库操作 (这里应该使用 'tx' ,而不是 'db')
tx.Create(...)
// 有错误时,手动调用事务的 Rollback()
tx.Rollback()
// 无错误时,手动调用事务的 Commit()
tx.Commit()

举一个经典的例子,老板给我转账

func (con TransitionController) Index(c *gin.Context) {
tx := models.DB.Begin()
//这里需要使用defer语句
defer func() {
if r := recover(); r != nil {
tx.Rollback()
con.error(c)
}
}()
if err := tx.Error; err != nil {
fmt.Println(err)
con.error(c)
}
// 老板账户减去 100
u1 := models.Bank{Id: 1}
//查找
tx.Find(&u1)
u1.Balance = u1.Balance - 100
//保存
if err := tx.Save(&u1).Error; err != nil {
tx.Rollback()
con.error(c)
}
// panic("遇到了错误")
// 我的账户增加 100
u2 := models.Bank{Id: 2}
tx.Find(&u2)
u2.Balance = u2.Balance + 100
// panic("失败")
if err := tx.Save(&u2).Error; err != nil {
tx.Rollback()
con.error(c)
}
tx.Commit()
con.success(c)
}

总的来说,GORM的事务功能简单好学,功能强大,性能强悍,GOlang是世界上最好的语言!!!!

本文正在参加技术专题18期-聊聊Go语言框架