1. 钩子
钩子是一个在 插入/查询/更新/删除 之前或之后被调用的方法。
如果你在一个模型中定义了特殊的方法,它将会在插入,更新,查询,删除的时候被自动调用,如果任何的回调抛出错误,GORM 将会停止将要执行的操作并且回滚当前的改变。
1.2. 钩子
可用于创建的钩子
if u.IsValid() {
err = errors.New("can't save invalid data")
}
return
}
func (u *User) AfterCreate(scope *gorm.Scope) (err error) {
if u.ID == 1 {
scope.DB().Model(u).Update("role", "admin")
return
}
注意,在 GORM 中的保存/删除 操作会默认进行事务处理,所以在事物中,所有的改变都是无效的,直到它被提交为止:
func (u *User) AfterCreate(tx *gorm.DB) (err error) {
tx.Model(u).Update("role", "admin")
}
可用于更新的钩子
代码示例:
func (u *User) BeforeUpdate() (err error) {
if u.readonly() {
err = errors.New("read only user")
}
return
}
// 在事务结束后,进行更新数据
func (u *User) AfterUpdate(tx *gorm.DB) (err error) {
if u.Confirmed {
}
return
// 开启事务
BeforeDelete
// 删除自身
AfterDelete
// 提交或回滚事务
代码示例:
可用于查询的钩子
// 从数据库中读取数据
// 加载之前 (急于加载)
AfterFind
代码示例:
func (u *User) AfterFind() (err error) {
if u.MemberShip == "" {
u.MemberShip = "user"
}
return