Go Gin:创建通用自定义验证器

but5z9lq  于 2023-06-27  发布在  Go
关注(0)|答案(2)|浏览(192)

我正在使用go Gin在我的项目中创建API。我需要创建自定义验证器,所以我创建了如下:

var valueone validator.Func = func(fl validator.FieldLevel) bool {
    value, ok := fl.Field()
    if ok {
        if value != "one" {
                  return true
                }
    }
    return true
}

var valuetwo validator.Func = func(fl validator.FieldLevel) bool {
    value, ok := fl.Field()
    if ok {
        if value != "two" {
                  return true
                }
    }
    return true
}
....

与其多次创建几乎相同的自定义验证器,是否有一种方法可以创建一个更通用的验证器,并且可以用于这两种情况,比如:

var value validator.Func = func(fl validator.FieldLevel, param) bool {
    value, ok := fl.Field()
    if ok {
        if value != param {
                  return true
                }
    }
    return true
}

我无法找到一种方法将参数传递给gin中的自定义验证器,或者通过任何其他可能的方式使这些通用验证器。我需要创建成千上万个几乎相似的验证器,我不想为每个验证器创建自定义验证器。

u3r8eeie

u3r8eeie1#

你不能改变函数的结构,因为这是它在包中的定义。

// Func accepts a FieldLevel interface for all validation needs. The return
// value should be true when validation succeeds.

type Func func(fl FieldLevel) bool

相反,我们可以尝试使用参数自定义验证标记
请参见下面的示例

package main

import (
    "github.com/go-playground/validator"
)

type Data struct {
    One string `json:"one" validate:"custom_validation=one"`
    Two string `json:"two" validate:"custom_validation=two"`
}

var validate *validator.Validate

func main() {
    validate = validator.New()

    err := validate.RegisterValidation("custom_validation", func(fl validator.FieldLevel) bool {
        value := fl.Field()
        param := fl.Param()

        return value.String() == param
    })

    if err != nil {
        panic(err)
    }

    // this will succeed
    {
        data := &Data{
            One: "one",
            Two: "two",
        }

        err = validate.Struct(data)
        if err != nil {
            panic(err)
        }
    }

    // it will fail here
    {
        data := &Data{
            Two: "one",
            One: "two",
        }

        err = validate.Struct(data)
        if err != nil {
            panic(err)
        }
    }
}

查看更多示例here
注意:Golang不支持!==

eimct9ow

eimct9ow2#

在go-playground/validator(Gin用于验证)中,您可以定义自定义验证器以使用参数,并且可以将参数包含在结构体的验证标记中。
你可以这样做:
首先,用标记名equals注册一个函数。函数中的第二个参数是标记的参数,它将是您要检查是否相等的值:

validate := validator.New()
_ = validate.RegisterValidation("equals", func(fl validator.FieldLevel) bool {
    param := fl.Param()
    return fl.Field().String() == param
})

然后,在你的结构体中,你可以使用equals标记,后跟**=**和值:

type MyStruct struct {
    FieldOne string `validate:"equals=one"`
    FieldTwo string `validate:"equals=two"`
}

在这种情况下,FieldOne必须等于“one”,FieldTwo必须等于“two”。
因此,使用这种方法,您只需定义一次自定义验证函数,然后就可以使用不同字段的不同参数来重用它。
请注意,**fl.Param()将参数作为字符串返回。如果参数不是字符串,则可能需要在验证函数中将它们解析为实际类型。另外,当validator.New()validate.RegisterValidation()**可能返回错误时,请确保进行错误处理。

相关问题