GORM 中使用 Struct 作为 Updates 参数时触发 BeforeUpdate 钩子的解决方案

2025-05-03 02:40:35作者：侯霆垣

在 GORM 这个流行的 Go 语言 ORM 框架中，开发者经常会遇到一个常见问题：当使用结构体(Struct)作为 Updates 方法的参数时，BeforeUpdate 钩子函数虽然会被执行，但后续生成的 SQL 语句中却不会包含钩子函数中设置的字段更新。这个问题尤其在实现乐观锁机制时显得尤为突出。

问题现象分析

让我们通过一个具体案例来说明这个问题。假设我们有一个基础模型 BaseModel，其中包含版本控制字段 version：

type BaseModel struct {
    ID        uint64
    Version   uint32
    // 其他字段...
}

func (m *BaseModel) BeforeUpdate(tx *gorm.DB) error {
    tx.Statement.SetColumn("version", clause.Expr{SQL: "version + 1"})
    return nil
}

当使用不同的参数类型调用 Updates 方法时，会出现以下差异：

使用 map 作为参数：

db.Model(&Table{}).Updates(map[string]any{"field": "value"})

生成的 SQL 会包含 version 字段的更新：UPDATE table SET field='value', version=version+1

使用 struct 作为参数：

db.Model(&Table{}).Updates(Table{Field: "value"})

生成的 SQL 不会包含 version 字段的更新：UPDATE table SET field='value'

问题根源探究

这个问题的根源在于 GORM 内部处理不同参数类型时的差异：

参数类型差异：
- 当使用 map 时，GORM 可以直接操作原始数据，添加额外的更新字段
- 当使用 struct 时，GORM 接收到的是值的拷贝，无法直接修改原始结构体
类型系统限制：
- BeforeUpdate 钩子中通过 SetColumn 设置的 clause.Expr 表达式无法直接赋值给 struct 中的基本类型字段（如 uint32）
- GORM 在处理 struct 更新时会严格遵循类型系统，而 map 则更灵活
SQL 生成时机：
- BeforeUpdate 钩子在 SQL 生成前执行
- 对于 struct 参数，字段选择在钩子执行前就已确定

解决方案

方案一：使用指针传递 struct

最简单的解决方案是始终使用指针传递 struct：

data := Table{Field: "value"}
db.Model(&Table{}).Updates(&data)

这种方法虽然能确保 BeforeUpdate 钩子被调用，但仍然无法解决 version 字段更新的问题。

方案二：自定义字段类型

更完善的解决方案是创建一个自定义的 Version 字段类型，实现必要的接口：

type VersionField uint32

// 实现必要的接口
var _ driver.Valuer = VersionField(0)
var _ sql.Scanner = (*VersionField)(nil)
var _ schema.UpdateClausesInterface = (*VersionField)(nil)

func (v *VersionField) UpdateClauses(field *schema.Field) []clause.Interface {
    return []clause.Interface{VersionUpdateClause{Field: field}}
}

// 实现 VersionUpdateClause 类型及其方法
// ...

这个自定义类型需要实现三个关键接口：

driver.Valuer：用于将 Go 值转换为数据库值
sql.Scanner：用于从数据库扫描值到 Go 类型
schema.UpdateClausesInterface：用于自定义更新时的 SQL 生成逻辑

方案三：使用乐观锁插件

GORM 社区提供了一个乐观锁插件，可以直接使用：

import "gorm.io/plugin/optimisticlock"

type Model struct {
    ID      uint
    Version optimisticlock.Version
    // 其他字段...
}

这个插件已经实现了我们需要的所有功能，包括自动版本号递增和并发控制。

实现细节解析

让我们深入分析自定义 VersionField 的实现细节：

UpdateClauses 方法：
- 返回一个自定义的 VersionUpdateClause
- 这个 clause 会修改最终的 SQL 语句
ModifyStatement 方法：
- 添加 WHERE 条件确保只更新匹配版本号的记录
- 将 struct 参数转换为 map 以便添加额外字段
- 设置 version=version+1 表达式
并发控制：
- 自动在 WHERE 子句中包含版本检查
- 如果版本不匹配，更新会失败
- 实现了标准的乐观锁机制