9.4 KiB
9.4 KiB
tags, create time
| tags | create time | ||||||||
|---|---|---|---|---|---|---|---|---|---|
|
2026-04-28 00:00 |
事务管理
概述
事务是数据库操作的「安全网」——它保证一组操作要么全部成功，要么全部失败。在现实业务中，你几乎处处离不开事务：转账时「扣 A 加 B」、下单时「减库存 + 创建订单 + 生成流水」，这些都需要事务来保证数据一致性。
flowchart TD
Start[开始事务 tx = db.Begin] --> Op1["执行操作 1"]
Op1 --> OK1{"操作 1 成功?"}
OK1 --> |是| Op2["执行操作 2"]
OK1 --> |否| Rollback["tx.Rollback()"]
OK2{"操作 2 成功?"} --> |是| Op3["执行操作 3"]
Op2 --> OK2
OK2 --> |否| Rollback
Op3 --> OK3{"操作 3 成功?"}
OK3 --> |是| Commit["tx.Commit()"]
OK3 --> |否| Rollback
style Start fill:#4FC08D,color:#fff
style Commit fill:#3B82F6,color:#fff
style Rollback fill:#EF4444,color:#fff
[!definition] 事务四大特性（ACID）
| 特性 | 含义 | 示例 |
|---|---|---|
| 原子性（Atomicity） | 全部提交或全部回滚 | 转账：A 扣钱和 B 加钱不可分割 |
| 一致性（Consistency） | 事务前后数据满足业务规则 | 转账前后总金额不变 |
| 隔离性（Isolation） | 并发事务互不干扰 | 两个用户同时下单不会超卖 |
| 持久性（Durability） | 提交后永不过期 | 断电后数据仍在 |
基本用法
Begin / Commit / Rollback
// 1. 开启事务
tx := db.Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback() // panic 时确保回滚
}
}()
// 2. 所有操作通过 tx 执行
user := User{Name: "Alice", Age: 30}
if err := tx.Create(&user).Error; err != nil {
tx.Rollback()
return err
}
order := Order{UserID: user.ID, Amount: 99.9}
if err := tx.Create(&order).Error; err != nil {
tx.Rollback()
return err
}
// 3. 全部成功，提交
if err := tx.Commit().Error; err != nil {
return err
}
return nil
[!warning] defer Rollback 的注意事项
Commit()成功后再执行defer Rollback()会报错（事务已提交）。所以推荐显式处理错误路径的回滚，defer只用于兜底 panic。
简洁写法（Err 链式判断）
tx := db.Begin()
if err := tx.Error; err != nil {
return err
}
// ... 操作 ...
if err := tx.Commit().Error; err != nil {
tx.Rollback()
return err
}
[!tip] 理解 tx.Transaction 返回值 GORM 的
Begin()返回*gorm.DB，但内部包装了事务。这个实例只在事务范围内有效，不能跨事务使用。
事务中的错误处理
func TransferMoney(fromID, toID uint, amount float64) error {
// 开启事务
tx := db.Begin()
// 查入账和出账账户（锁行！）
var fromAccount Account
if err := tx.Set("gorm:query_option", "FOR UPDATE").First(&fromAccount, fromID).Error; err != nil {
tx.Rollback()
return fmt.Errorf("查询转出账户失败: %w", err)
}
var toAccount Account
if err := tx.First(&toAccount, toID).Error; err != nil {
tx.Rollback()
return fmt.Errorf("查询入账账户失败: %w", err)
}
// 余额不足
if fromAccount.Balance < amount {
tx.Rollback()
return errors.New("余额不足")
}
// 扣款
fromAccount.Balance -= amount
if err := tx.Save(&fromAccount).Error; err != nil {
tx.Rollback()
return fmt.Errorf("扣款失败: %w", err)
}
// 入账
toAccount.Balance += amount
if err := tx.Save(&toAccount).Error; err != nil {
tx.Rollback()
return fmt.Errorf("入账失败: %w", err)
}
return tx.Commit().Error
}
[!important] FOR UPDATE 锁行 事务中的查询如果要用更新的数据做决策，必须用
FOR UPDATE（排他锁），否则可能读到未提交的旧数据：// ❌ 竞态条件：两个并发请求可能都读到余额足够并扣款 db.Where("id = ?", id).First(&account) account.Balance -= 100 db.Save(&account) // ✅ 加锁读取，另一个事务只能等待 db.Set("gorm:query_option", "FOR UPDATE").Where("id = ?", id).First(&account)
嵌套事务
当代码结构有分层时，内部函数可能需要发起自己的事务。GORM 支持嵌套事务机制——内部实际上是保存点（Savepoint），而非真正的事务：
func CreateOrderWithAudit(tx *gorm.DB, order Order) error {
// 检查外层是否已有事务
if !tx.IsTransaction() {
// 外层不是事务，自己开一个
tx = db.Begin()
defer func() {
if v := recover(); v != nil {
tx.Rollback()
panic(v)
}
}()
}
// 保存点：如果内部失败可以回滚到这里，不影响外部
tx.SavePoint("create_order")
// 主逻辑
if err := tx.Create(&order).Error; err != nil {
tx.RollbackTo("create_order") // 回滚到保存点
return err
}
// 审计日志记录
audit := Audit{Table: "orders", Action: "create", RecordID: order.ID}
if err := tx.Create(&audit).Error; err != nil {
tx.RollbackTo("create_order") // 审计失败也可以回滚
return err
}
// 成功 —— 如果不外层事务则提交，否则由外层统一提交
if !tx.IsTransaction() {
return tx.Commit().Error
}
return nil
}
[!tip] 嵌套事务的本质 MySQL/PostgreSQL 不支持真正的嵌套事务，它们用的是 Savepoint。这意味着：
- 内部回滚只会回滚到最近的 savepoint，不会影响外层事务
- 最终
COMMIT仍然是一次性的，从外层事务发出- 如果外层事务
ROLLBACK，整个事务树（包括所有 savepoint）都会被回滚
事务级别控制
import "database/sql"
tx := db.Session(&gorm.Session{
DryRun: false, // 模拟模式：构建 SQL 但不执行
}).Begin(&sql.TxOptions{
Isolation: sql.LevelRepeatableRead, // 设置隔离级别
ReadOnly: false, // 读写事务
})
// 或者针对特定查询
db.Set("gorm:prepare_stmt", true).Find(&users) // 预编译语句
[!example] 隔离级别速查
| 级别 | 脏读 | 不可重复读 | 幻读 | 性能影响 |
|---|---|---|---|---|
| Read Uncommitted | ❌ 允许 | ❌ 允许 | ❌ 允许 | 最快 |
| Read Committed | ✅ 阻止 | ❌ 允许 | ❌ 允许 | 快 |
| Repeatable Read（MySQL 默认） | ✅ 阻止 | ✅ 阻止 | ⚠️ 部分阻止 | 中等 |
| Serializable | ✅ 阻止 | ✅ 阻止 | ✅ 阻止 | 最慢 |
事务与中间件结合
在实际项目中，经常需要将事务与 Gin 等 Web 框架集成，实现自动事务管理：
// Gin 中间件：每个 HTTP 请求自带一个事务
func TransactionMiddleware(db *gorm.DB) gin.HandlerFunc {
return func(c *gin.Context) {
// 开始事务
tx := db.Begin()
c.Set("db", tx) // 存入 Context
defer func() {
if p := recover(); p != nil {
tx.Rollback()
panic(p) // 重新抛出让上层捕获
}
// 请求处理完毕且无错误，提交事务
if c.Writer.Status() >= 200 && c.Writer.Status() < 400 {
tx.Commit()
} else {
tx.Rollback()
}
}()
c.Next()
}
}
// 在 handler 中使用
func CreateUser(c *gin.Context) {
// 从 Context 获取当前事务
txVal, _ := c.Get("db")
tx := txVal.(*gorm.DB)
user := User{Name: c.PostForm("name")}
if err := tx.Create(&user).Error; err != nil {
c.JSON(500, gin.H{"error": err.Error()})
return
}
c.JSON(201, user)
}
事务决策流程图
flowchart TD
Start[需要多步数据操作] --> InTx{"是否在已有事务中?"}
InTx --> |是| SaveQ{需要局部回滚能力_}
InTx --> |否| NewTx{"操作数量?"}
NewTx --> |单条 SQL| SimpleTx["不需要事务<br/>直接操作即可"]
NewTx --> |多条 SQL| BeginTx["db.Begin()"]
SaveQ --> |需要| Savepoint["SavePoint('name')"]
SaveQ --> |不需要| Continue["继续执行"]
BeginTx --> Ops["依次执行各操作"]
Continue --> Ops
Savepoint --> Ops
Ops --> AllOK{"全部成功?"}
AllOK --> |是| Commit["Commit"]
AllOK --> |否| RollBack["Rollback"]
style Start fill:#4FC08D,color:#fff
style Commit fill:#3B82F6,color:#fff
style RollBack fill:#EF4444,color:#fff
style SimpleTx fill:#A0AEC0,color:#fff
常见坑点速查
| 问题 | 原因 | 解决方案 |
|---|---|---|
| 忘记 Commit/Rollback | 漏了错误分支的回滚逻辑 | 对所有错误分支显式调用 Rollback |
| 事务内查询被其他事务修改 | 没加 FOR UPDATE | 关键查询加 Set("gorm:query_option", "FOR UPDATE") |
| 嵌套事务外层的 Commit 失效 | 内部使用了独立连接 | 始终复用同一个 *gorm.DB 对象 |
| 长事务导致行锁堆积 | 事务中包含耗时操作（RPC、HTTP） | 将耗时操作移出事务范围 |
| defer Rollback 导致已提交事务报错 | Commit 后再执行 deferred Rollback | 用 panic 兜底，正常路径不依赖 defer |