GORM中使用事务学习指南(小白必会)

事务处理可以用来维护数据库的完整性,保证成批的 SQL 语句要么全执行,要么全不执行。

1 禁用默认事务

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

package models
import ( "fmt"
"gorm.io/driver/mysql"
"gorm.io/gorm"
)
var DB *gorm.DB
var err error
func init() {
dsn := "root:123456@tcp(192.168.0.6:3306)/gin?charset=utf8mb4&parseTime=True&loc=L
ocal"DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
SkipDefaultTransaction: true, })
DB.Debug()
if err != nil {
fmt.Println(err)
}
}

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

2 事务

https://gorm.io/zh_CN/docs/transactions.html 01、事务执行流程 要在事务中执行一系列操作,通常您可以参照下面的流程来执行。

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

02、事务(手动控制)

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

03、张三给李四转账

package admin
import ( "fmt"
"gindemo13/models"
"github.com/gin-gonic/gin"
)
type TransitionController struct {
BaseController
}
func (con TransitionController) Index(c *gin.Context) {
tx := models.DB.Begin()
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)
}

更多Go知识点b站学习地址:https://www.bilibili.com/video/BV1Rm421N7Jy


回到顶部