GORM中钩子了解
在 GORM 中，钩子函数（Hooks）是一种强大的机制，它允许你在模型的生命周期中，于特定的数据库操作（如创建、更新、删除、查询）前后，自动执行自定义的业务逻辑。
简单来说，钩子就是模型结构体上的方法。当你为模型定义了这些特定名称的方法后，GORM 会在执行相应操作时自动调用它们。
🪝 钩子函数的工作原理
钩子函数有固定的函数签名，必须接收一个 *gorm.DB 参数并返回一个 error。
func(模型实例)钩子函数名(tx*gorm.DB)error成功执行：如果钩子函数返回 nil，GORM 会继续执行后续的数据库操作。
中断操作：如果钩子函数返回一个非 nil 的错误，GORM 会立即停止后续操作，并回滚当前的事务（如果存在）。这个特性非常适合用于数据校验。
📜 完整的钩子生命周期
以下是 GORM 支持的所有钩子函数及其触发时机，按照操作的执行顺序排列：
创建 (Create)
- BeforeSave
- BeforeCreate
- (执行 SQL 插入)
- AfterCreate
- AfterSave
更新 (Update)
- BeforeSave
- BeforeUpdate
- (执行 SQL 更新)
- AfterUpdate
- AfterSave
删除 (Delete)
- BeforeDelete
- (执行 SQL 删除)
- AfterDelete
查询 (Query)
- (执行 SQL 查询)
- AfterFind
注意：BeforeSave 和 AfterSave 是在 Create 和 Update 操作中都会被调用的钩子。
💡 常见应用场景
钩子函数非常适合处理那些需要在数据持久化前后自动执行的通用逻辑。
- 数据校验与预处理 (BeforeCreate, BeforeUpdate)
在数据写入数据库前进行校验或修改。
// 创建前自动生成 UUIDfunc(u*User)BeforeCreate(tx*gorm.DB)(errerror){u.UUID=uuid.New()// 数据校验示例：如果名字是 "admin"，则返回错误阻止创建ifu.Name=="admin"{returnerrors.New("invalid name")}return}// 更新前自动更新时间戳func(u*User)BeforeUpdate(tx*gorm.DB)(errerror){u.UpdateTime=time.Now()return}- 关联操作与日志记录 (AfterCreate, AfterUpdate, AfterDelete)
在主操作完成后，执行一些后续任务，如记录日志或更新关联数据。
// 创建后，如果用户ID为1，则将其角色更新为 "admin"func(u*User)AfterCreate(tx*gorm.DB)(errerror){ifu.ID==1{tx.Model(u).Update("role","admin")}return}// 删除后，记录一条删除日志func(u*User)AfterDelete(tx*gorm.DB)(errerror){log.Printf("用户 %s 已被删除",u.Name)return}- 查询后处理 (AfterFind)
在数据从数据库加载后，对结果进行填充或转换。
// 查询后，如果会员信息为空，则赋予默认值func(u*User)AfterFind(tx*gorm.DB)(errerror){ifu.MemberShip==""{u.MemberShip="user"}return}⚠️ 重要注意事项
事务与钩子：GORM 的 Save、Create、Update、Delete 等操作默认在事务中运行。这意味着在 After 类型的钩子中，你仍然处于同一个事务内。如果钩子返回错误，整个事务（包括已经完成的数据库操作）都会被回滚。
使用 tx 对象：钩子函数接收的 *gorm.DB 参数（通常命名为 tx）代表了当前的数据库会话。在钩子内部进行任何数据库操作时，都应使用这个 tx 对象，以确保操作在正确的事务上下文中执行