在 Go 语言中，GORM 是一个功能强大的 ORM 框架，它支持事务操作，可以保证多个数据库操作要么全部成功，要么全部失败。下面详细讲解 GORM 的事务操作及其使用方法，并附带示例代码和中文注释。
GORM 事务操作的基本步骤
- 开启事务：使用
db.Begin()方法开启事务。 - 执行数据库操作：在事务上下文中执行相关操作。
- 提交事务：操作成功时，调用
tx.Commit()提交事务。 - 回滚事务：操作失败时，调用
tx.Rollback()回滚事务。
代码示例：事务的使用
以下是一个完整的代码示例，展示如何使用 GORM 的事务操作，并附带详细中文注释。
数据库模型
假设我们有两个表：用户表和订单表。
package main
import (
"fmt"
"log"
"time"
"gorm.io/driver/sqlite" // 使用 SQLite 驱动
"gorm.io/gorm"
)
// 用户表模型
type User struct {
ID uint `gorm:"primaryKey"` // 主键
Name string // 用户名
Balance float64 // 余额
CreatedAt time.Time // 创建时间
UpdatedAt time.Time // 更新时间
}
// 订单表模型
type Order struct {
ID uint `gorm:"primaryKey"` // 主键
UserID uint // 用户 ID
Amount float64 // 订单金额
CreatedAt time.Time // 创建时间
UpdatedAt time.Time // 更新时间
}
func main() {
// 初始化数据库
db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
if err != nil {
log.Fatalf("无法连接到数据库: %v", err)
}
// 自动迁移数据库
if err := db.AutoMigrate(&User{}, &Order{}); err != nil {
log.Fatalf("数据库迁移失败: %v", err)
}
// 创建示例用户
db.Create(&User{Name: "Alice", Balance: 100.0})
// 调用事务函数
err = performTransaction(db, 1, 50.0)
if err != nil {
fmt.Printf("事务失败: %v\n", err)
} else {
fmt.Println("事务成功")
}
}
// performTransaction 处理用户下单事务
func performTransaction(db *gorm.DB, userID uint, orderAmount float64) error {
// 开启事务
tx := db.Begin()
// 检查事务是否成功开启
if tx.Error != nil {
return fmt.Errorf("无法开启事务: %w", tx.Error)
}
// 查询用户
var user User
if err := tx.First(&user, userID).Error; err != nil {
// 查询失败，回滚事务
tx.Rollback()
return fmt.Errorf("查询用户失败: %w", err)
}
// 检查用户余额是否足够
if user.Balance < orderAmount {
// 余额不足，回滚事务
tx.Rollback()
return fmt.Errorf("用户余额不足")
}
// 扣除用户余额
user.Balance -= orderAmount
if err := tx.Save(&user).Error; err != nil {
// 更新用户余额失败，回滚事务
tx.Rollback()
return fmt.Errorf("更新用户余额失败: %w", err)
}
// 创建订单
order := Order{
UserID: user.ID,
Amount: orderAmount,
}
if err := tx.Create(&order).Error; err != nil {
// 创建订单失败，回滚事务
tx.Rollback()
return fmt.Errorf("创建订单失败: %w", err)
}
// 提交事务
if err := tx.Commit().Error; err != nil {
return fmt.Errorf("提交事务失败: %w", err)
}
return nil
}代码运行说明
1. 数据库初始化
- 使用 SQLite 创建一个名为
test.db的本地数据库文件。 - 自动迁移两个表：
users和orders。
2. 事务执行逻辑
- 用户下单时，会检查用户余额是否充足。
- 如果余额不足或任一操作失败，则事务会回滚，保证数据库状态一致性。
- 如果所有操作成功，则事务提交。
3. 运行代码
运行程序后：
- 如果用户余额足够，则扣除余额并创建订单，输出 "事务成功"。
- 如果余额不足或发生错误，则回滚操作，输出错误信息。
输出示例
用户余额足够：
事务成功用户余额不足：
事务失败: 用户余额不足查询用户失败：
事务失败: 查询用户失败: record not found
注意事项
- 事务生命周期
GORM 的事务是基于*gorm.DB的，每次调用tx.Commit()或tx.Rollback()后，事务的生命周期结束，不能再用该tx执行其他操作。 - 并发操作
如果有高并发场景，建议在事务中锁定相关记录，避免并发问题。例如，可以使用SELECT ... FOR UPDATE。 - 嵌套事务
GORM 不支持真正的嵌套事务，但可以通过保存点（Save Point）实现类似功能。
通过此代码示例和详细讲解，你可以在实际开发中安全地使用 GORM 事务操作。
