Golang 使用事务的简单实践

// 需要使用事务的方法
func (s *userService) funcName(ctx context.Context, req *v1.Req) (*v1.RespData, error) {
    // 获取事务的最终结果
    err := s.tm.Transaction(ctx, func(ctx context.Context) error {
        // 内部写相关的原子性数据库操作
        // 如果任意操作报错，将触发回滚，恢复之前的状态

        // 调用数据层方法 repository
        // repository.CreateUser(ctx, req)
        // repository.CreateLog(ctx, req)

        // 所有操作均无错误，正常退出
        return nil
    })

    // 如果事务中存在错误，所有操作都会被回滚
    if err != nil {
        return nil, err
    }

    // 没有触发事务报错，正常返回结果
    return &v1.RespData{}, nil
}

func (s *userService) CreateOrder(ctx context.Context, req *v1.OrderReq) error {
    return s.db.Transaction(func(tx *gorm.DB) error {
        // 创建订单
        if err := tx.Create(&Order{UserID: req.UserID, Amount: req.Amount}).Error; err != nil {
            return err // 返回错误会触发回滚
        }

        // 扣减库存
        if err := tx.Model(&Product{}).
            Where("id = ? AND stock >= ?", req.ProductID, req.Quantity).
            Update("stock", gorm.Expr("stock - ?", req.Quantity)).Error; err != nil {
            return err
        }

        // 写入日志
        if err := tx.Create(&Log{Action: "create_order", UserID: req.UserID}).Error; err != nil {
            return err
        }

        // 所有操作成功，事务提交
        return nil
    })
}

s.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
    return tx.Create(&User{Name: "Tom"}).Error
})

func (r *UserRepo) Create(ctx context.Context, tx *gorm.DB, user *User) error {
    return tx.WithContext(ctx).Create(user).Error
}

tx.SavePoint("sp1")
// ...
tx.RollbackTo("sp1")