自定义规则-完整数据校验

大家也许已经注意到,当我们给定一个struct时,我们的规则只能对其中的键值或者属性进行校验,如果我们想要通过规则完整校验struct这个对象时,居然无法注册校验组件的自定义校验规则。当然,没有是不可能的,我们的校验组件当然也支持直接校验当前的struct对象。

我们来看一个例子,在这个例子中,我们需要对创建的用户请求进行完整的自定义校验,并注册一个UserCreateReq的校验规则来实现。

package main

import (
    "context"
    "fmt"
    "github.com/gogf/gf/errors/gerror"
    "github.com/gogf/gf/frame/g"
    "github.com/gogf/gf/util/gmeta"
)

type UserCreateReq struct {
    gmeta.Meta `v:"UserCreateReq"`
    Name       string
    Pass       string
}

func UserCreateReqChecker(ctx context.Context, rule string, value interface{}, message string, data interface{}) error {
    user := &UserCreateReq{}
    if v, ok := data.(*UserCreateReq); ok {
        user = v
    }
    // SELECT COUNT(*) FROM `user` WHERE `name` = xxx
    count, err := g.Model("user").Ctx(ctx).Where("name", user.Name).Count()
    if err != nil {
        return err
    }
    if count > 0 {
        return gerror.Newf(`The name "%s" is already token`, user.Name)
    }
    return nil
}

func main() {
    user := &UserCreateReq{
        Name: "john",
        Pass: "123456",
    }
    err := g.Validator().RuleFunc("UserCreateReq", UserCreateReqChecker).CheckStruct(user)
    fmt.Println(err)
}

可以看到,我们通过给结构体一个gmeta.Meta嵌入的元数据,并绑定UserCreateReq的自定义规则,gmeta.Meta作为结构体的一部分,当我们通过CheckStruct校验该结构体对象时,便可以通过UserCreateReq来实现校验。

上面的例子,我们执行后,终端输出:

The name "john" is already token