如何在GORM中统计MySQL READ UNCOMMITTED隔离级别的数据？

你的代码存在几个核心问题，导致无法正确读取未提交事务中的订单数据：

1. 事务开启冲突：同时使用GORM的tx := o.db.Begin()和手动SQL的START TRANSACTION，GORM的事务管理会覆盖手动的事务起始指令，导致后续设置隔离级别的语句没有作用在目标事务上。
2. 多语句执行不可靠：将事务启动、隔离级别设置、查询语句拼接成单条多语句SQL，GORM的Raw方法无法保证所有语句都在同一个事务上下文内生效。
3. 硬编码参数风险：直接在SQL中写死attendee_id = 1和event_id = 1，既不符合业务逻辑（应该用传入的参数），还存在SQL注入隐患。

修复后的代码示例（使用GORM链式调用） #

func (o *OrderRepositoryAdapter) CountOrderByAttendeeIDAndEventID(ctx context.Context, attendeeID uint, eventID uint) (uint, error) {
    var count int64
    // 开启带上下文的事务
    tx := o.db.WithContext(ctx).Begin()
    if tx.Error != nil {
        return 0, tx.Error
    }
    // 延迟处理异常回滚
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
        }
    }()

    // 为当前事务设置READ UNCOMMITTED隔离级别
    if err := tx.Exec("SET TRANSACTION ISOLATION LEVEL READ UNCOMMITTED").Error; err != nil {
        tx.Rollback()
        return 0, err
    }

    // 执行统计查询，使用参数绑定传入变量
    if err := tx.Model(&Order{}).Where("attendee_id = ? AND event_id = ?", attendeeID, eventID).Count(&count).Error; err != nil {
        tx.Rollback()
        return 0, err
    }

    // 查询完成后回滚事务，释放数据库连接
    tx.Rollback()
    return uint(count), nil
}

若偏好使用Raw查询的修复版本 #

func (o *OrderRepositoryAdapter) CountOrderByAttendeeIDAndEventID(ctx context.Context, attendeeID uint, eventID uint) (uint, error) {
    var count int64
    tx := o.db.WithContext(ctx).Begin()
    if tx.Error != nil {
        return 0, tx.Error
    }

    // 仅针对当前事务设置隔离级别，避免污染会话
    if err := tx.Exec("SET TRANSACTION ISOLATION LEVEL READ UNCOMMITTED").Error; err != nil {
        tx.Rollback()
        return 0, err
    }

    // 使用参数化查询，避免SQL注入
    query := "SELECT count(*) FROM orders WHERE attendee_id = ? AND event_id = ?"
    if err := tx.Raw(query, attendeeID, eventID).Scan(&count).Error; err != nil {
        tx.Rollback()
        return 0, err
    }

    tx.Rollback()
    return uint(count), nil
}

关键注意事项 #

• 隔离级别作用范围：使用SET TRANSACTION ISOLATION LEVEL而非SET SESSION TRANSACTION ISOLATION LEVEL，仅修改当前事务的隔离级别，不会影响后续其他数据库操作。
• 事务生命周期：GORM的事务需通过Begin()开启，Commit()提交或Rollback()回滚，不要手动写START TRANSACTION/COMMIT语句，避免冲突。
• 参数绑定：始终使用?作为占位符传递参数，杜绝硬编码，防止SQL注入。

内容的提问来源于stack exchange，提问作者Loint