golang之数据验证validator

栏目: IT技术 · 发布时间: 4年前

内容简介：在web应用中经常会遇到数据验证问题，普通的验证方法比较繁琐，这里介绍一个使用比较多的包将验证规则写在struct对字段tag里，再通过反射（安装

前言

在web应用中经常会遇到数据验证问题，普通的验证方法比较繁琐，这里介绍一个使用比较多的包 validator 。

原理

将验证规则写在struct对字段tag里，再通过反射（ reflect ）获取struct的tag，实现数据验证。

安装

go get github.com/go-playground/validator/v10

示例

package main

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

type Users struct {
    Phone   string `form:"phone" json:"phone" validate:"required"`
    Passwd   string `form:"passwd" json:"passwd" validate:"required,max=20,min=6"`
    Code   string `form:"code" json:"code" validate:"required,len=6"`
}

func main() {

    users := &Users{
        Phone:      "1326654487",
        Passwd:       "123",
        Code:            "123456",
    }
    validate := validator.New()
    err := validate.Struct(users)
    if err != nil {
        for _, err := range err.(validator.ValidationErrors) {
            fmt.Println(err)//Key: 'Users.Passwd' Error:Field validation for 'Passwd' failed on the 'min' tag
            return
        }
    }
    return
}

验证规则

  • required ：必填
  • email：验证字符串是email格式；例："email"
  • url：这将验证字符串值包含有效的网址;例："url"
  • max：字符串最大长度；例："max=20"
  • min:字符串最小长度；例："min=6"
  • excludesall:不能包含特殊字符；例："excludesall=0x2C"//注意这里用十六进制表示。
  • len：字符长度必须等于n，或者数组、切片、map的len值为n，即包含的项目数；例："len=6"
  • eq：数字等于n，或者或者数组、切片、map的len值为n，即包含的项目数；例："eq=6"
  • ne：数字不等于n，或者或者数组、切片、map的len值不等于为n，即包含的项目数不为n，其和eq相反；例："ne=6"
  • gt：数字大于n，或者或者数组、切片、map的len值大于n，即包含的项目数大于n；例："gt=6"
  • gte：数字大于或等于n，或者或者数组、切片、map的len值大于或等于n，即包含的项目数大于或等于n；例："gte=6"
  • lt：数字小于n，或者或者数组、切片、map的len值小于n，即包含的项目数小于n；例："lt=6"
  • lte：数字小于或等于n，或者或者数组、切片、map的len值小于或等于n，即包含的项目数小于或等于n；例："lte=6"

跨字段验证

如想实现比较输入密码和确认密码是否一致等类似场景

eqfield=Field
nefield=Field
gtfield=Field
gtefield=Field
ltfield=Field
ltefield=Field
eqcsfield=Other.Field
necsfield=Other.Field
gtcsfield=Other.Field
gtecsfield=Other.Field
ltcsfield=Other.Field
ltecsfield=Other.Field

示例

type UserReg struct {
    Passwd   string `form:"passwd" json:"passwd" validate:"required,max=20,min=6"`
    Repasswd   string `form:"repasswd" json:"repasswd" validate:"required,max=20,min=6,eqfield=Passwd"`
}

示例验证了Passwd，和Repasswd值是否相等。如想了解更多类型，请参考文档 https://godoc.org/gopkg.in/go...

自定义验证类型

示例：

package main

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

type Users struct {
    Name   string `form:"name" json:"name" validate:"required,CustomValidationErrors"`//包含自定义函数
    Age   uint8 `form:"age" json:"age" validate:"required,gt=18"`
    Passwd   string `form:"passwd" json:"passwd" validate:"required,max=20,min=6"`
    Code   string `form:"code" json:"code" validate:"required,len=6"`
}

func main() {

    users := &Users{
        Name:      "admin",
        Age:        12,
        Passwd:       "123",
        Code:            "123456",
    }
    validate := validator.New()
    //注册自定义函数
    _=validate.RegisterValidation("CustomValidationErrors", CustomValidationErrors)
    err := validate.Struct(users)
    if err != nil {
        for _, err := range err.(validator.ValidationErrors) {
            fmt.Println(err)//Key: 'Users.Name' Error:Field validation for 'Name' failed on the 'CustomValidationErrors' tag
            return
        }
    }
    return
}

func CustomValidationErrors(fl validator.FieldLevel) bool {
return fl.Field().String() != "admin"
}

翻译错误信息为中文

通过以上示例我们看到，validator默认的错误提示信息类似如下

Key: 'Users.Name' Error:Field validation for 'Name' failed on the 'CustomValidationErrors' tag

显然这并不是我们想要，如想翻译成中文，或其他语言怎么办？ go-playground 上提供了很好的解决方法。

先自行安装需要的两个包

执行：

go get github.com/go-playground/universal-translator
go get github.com/go-playground/locales

示例：

package main

import (
    "fmt"
    "github.com/go-playground/locales/zh"
    ut "github.com/go-playground/universal-translator"
    "github.com/go-playground/validator/v10"
    zh_translations "github.com/go-playground/validator/v10/translations/zh"
)

type Users struct {
    Name   string `form:"name" json:"name" validate:"required"`
    Age   uint8 `form:"age" json:"age" validate:"required,gt=18"`
    Passwd   string `form:"passwd" json:"passwd" validate:"required,max=20,min=6"`
    Code   string `form:"code" json:"code" validate:"required,len=6"`
}

func main() {
    users := &Users{
        Name:      "admin",
        Age:        12,
        Passwd:       "123",
        Code:            "123456",
    }
    uni := ut.New(zh.New())
    trans, _ := uni.GetTranslator("zh")
    validate := validator.New()
    //验证器注册翻译器
    err := zh_translations.RegisterDefaultTranslations(validate, trans)
    if err!=nil {
        fmt.Println(err)
    }
    err = validate.Struct(users)
    if err != nil {
        for _, err := range err.(validator.ValidationErrors) {
            fmt.Println(err.Translate(trans))//Age必须大于18
            return
        }
    }

    return
}

输出：

Age必须大于18

至此我们发现大部分错误信息已经翻译成中文，但字段名（Age）还是没有翻译，为了将字段名翻译成中文，查了一些资料， https://www.jianshu.com/p/51b... ;，

照着做没有成功（可能有遗漏吧），最后还是翻看了一下源代码，在 https://github.com/go-playgro... ，第137行 

// RegisterTagNameFunc registers a function to get alternate names for StructFields.
//
// eg. to use the names which have been specified for JSON representations of structs, rather than normal Go field names:
//
//    validate.RegisterTagNameFunc(func(fld reflect.StructField) string {
//        name := strings.SplitN(fld.Tag.Get("json"), ",", 2)[0]
//        if name == "-" {
//            return ""
//        }
//        return name
//    })

其实原理就是注册一个函数，将struct tag里添加的中文名 作为备用名。

package main

import (
    "fmt"
    "github.com/go-playground/locales/zh"
    ut "github.com/go-playground/universal-translator"
    "github.com/go-playground/validator/v10"
    zh_translations "github.com/go-playground/validator/v10/translations/zh"
    "reflect"
)

type Users struct {
    Name   string `form:"name" json:"name" validate:"required" label:"用户名"`
    Age   uint8 `form:"age" json:"age" validate:"required,gt=18" label:"年龄"`
    Passwd   string `form:"passwd" json:"passwd" validate:"required,max=20,min=6"`
    Code   string `form:"code" json:"code" validate:"required,len=6"`
}

func main() {
    users := &Users{
        Name:      "admin",
        Age:        12,
        Passwd:       "123",
        Code:            "123456",
    }
    uni := ut.New(zh.New())
    trans, _ := uni.GetTranslator("zh")
    validate := validator.New()
    //注册一个函数，获取struct tag里自定义的label作为字段名
    validate.RegisterTagNameFunc(func(fld reflect.StructField) string {
        name:=fld.Tag.Get("label")
        return name
    })
    //注册翻译器
    err := zh_translations.RegisterDefaultTranslations(validate, trans)
    if err!=nil {
        fmt.Println(err)
    }
    err = validate.Struct(users)
    if err != nil {
        for _, err := range err.(validator.ValidationErrors) {
            fmt.Println(err.Translate(trans))//年龄必须大于18
            return
        }
    }

    return
}

输出结果：

年龄必须大于18

gin 内置的validator

gin已经支持 go-playground / validator / v10 进行验证。在 此处 查看有关标签用法的完整文档。

以下只提供了一个绑定ShouldBindWith示例，如需了解更多方法，进入 这里 。

示例

package main

import (
    "fmt"
    "github.com/go-playground/locales/zh"
    ut "github.com/go-playground/universal-translator"
    "github.com/go-playground/validator/v10"
    "net/http"
    "reflect"
    "strings"
    "time"

    "github.com/gin-gonic/gin"
    "github.com/gin-gonic/gin/binding"
    zh_translations "github.com/go-playground/validator/v10/translations/zh"
)
var trans ut.Translator
// Booking contains binded and validated data.
type Booking struct {
    CheckIn  time.Time `form:"check_in" json:"check_in" binding:"required,bookabledate" time_format:"2006-01-02" label:"输入时间"`
    CheckOut time.Time `form:"check_out" json:"check_out" binding:"required,gtfield=CheckIn" time_format:"2006-01-02" label:"输出时间"`
}

var bookableDate validator.Func = func(fl validator.FieldLevel) bool {
    date, ok := fl.Field().Interface().(time.Time)
    if ok {
        today := time.Now()
        if today.After(date) {
            return false
        }
    }
    return true
}

func main() {
    route := gin.Default()
    uni := ut.New(zh.New())
    trans, _ = uni.GetTranslator("zh")

    if v, ok := binding.Validator.Engine().(*validator.Validate); ok {
        //注册翻译器
        _= zh_translations.RegisterDefaultTranslations(v, trans)
        //注册自定义函数
        _=v.RegisterValidation("bookabledate", bookableDate)

        //注册一个函数，获取struct tag里自定义的label作为字段名
        v.RegisterTagNameFunc(func(fld reflect.StructField) string {
            name:=fld.Tag.Get("label")
            name1 := strings.SplitN(fld.Tag.Get("json"), ",", 2)[0]
            fmt.Println(name1)
            return name
        })
        //根据提供的标记注册翻译
        v.RegisterTranslation("bookabledate", trans, func(ut ut.Translator) error {
            return ut.Add("bookabledate", "{0}不能早于当前时间或{1}格式错误!", true)
        }, func(ut ut.Translator, fe validator.FieldError) string {
            t, _ := ut.T("bookabledate", fe.Field(), fe.Field())
            return t
        })

    }
    route.GET("/bookable", getBookable)
    route.Run(":8085")
}

func getBookable(c *gin.Context) {
    var b Booking
    if err := c.ShouldBindWith(&b, binding.Query); err == nil {
        c.JSON(http.StatusOK, gin.H{"message": "Booking dates are valid!"})
    } else {
        errs := err.(validator.ValidationErrors)

        fmt.Println(errs.Translate(trans))
        //for _, e := range errs {
        //    // can translate each error one at a time.
        //    fmt.Println(e.Translate(trans))
        //}
        c.JSON(http.StatusBadRequest, gin.H{"error": errs.Translate(trans)})
    }
}

运行程序，执行以下命令

$ curl "localhost:8085/bookable?check_in=2018-04-16✓_out=2018-04-16"

结果：

{"error":{"Booking.输入时间":"输入时间不能早于当前时间或输入时间格式错误!","Booking.输出时间":"输出时间必须大于CheckIn"}}

查看以上结果我们发现翻译还是不太完美，如规则中有gtfield的情况，字段（CheckIn）并没有被翻译。所以通过struct添加label的方式并不能从根本上解决字段翻译问题。为了得到想要的结果，就需要将错误信息做单独处理再输出。

先定义翻译库

var BookingTrans =map[string]string{"CheckIn":"输入时间","CheckOut":"输出时间"}

再定义翻译函数

func TransTagName(libTans,err interface{}) interface{} {
    switch err.(type) {
    case validator.ValidationErrorsTranslations:
        var errs map[string]string
        errs = make(map[string]string,0)
        for k,v:=range err.(validator.ValidationErrorsTranslations){
            for key,value:=range libTans.(map[string]string)  {
                v=strings.Replace(v,key,value,-1)
            }
            errs[k] = v
        }
        return errs
    case string:
        var errs string
        for key,value:=range libTans.(map[string]string)  {
            errs=strings.Replace(errs,key,value,-1)
        }
        return errs
    default:
        return err
    }
}

将原来翻译错误信息的地方

errs.Translate(trans)

修改为

msg:=TransTagName(BookingTrans,errs.Translate(trans))
fmt.Println(msg)

结果

{"error":{"Booking.输入时间":"输入时间不能早于当前时间或输入时间格式错误!","Booking.输出时间":"输出时间必须大于输入时间"}}

小结：

  1. gin 已经支持 validator 最新的v10。
  2. validator 数据验证顺序struct字段从上往下，单个字段规则（binding:"gt=0,lt=2`），先左后右。

参考：

https://github.com/go-playgro...

https://github.com/gin-gonic/gin

https://gitissue.com/issues/5...

https://segmentfault.com/a/11...

links

欢迎关注我们的微信公众号，每天学习 Go 知识

golang之数据验证validator

以上所述就是小编给大家介绍的《golang之数据验证validator》，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持！

查看所有标签

猜你喜欢:

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。

码农书籍
Learn Python 3 the Hard Way

Learn Python 3 the Hard Way

Zed A. Shaw / Addison / 2017-7-7 / USD 30.74

You Will Learn Python 3! Zed Shaw has perfected the world’s best system for learning Python 3. Follow it and you will succeed—just like the millions of beginners Zed has taught to date! You bring t......一起来看看 《Learn Python 3 the Hard Way》 这本书的介绍吧!

码农工具
XML 在线格式化
XML 在线格式化

在线 XML 格式化压缩工具

RGB HSV 转换
RGB HSV 转换

RGB HSV 互转工具

HSV CMYK 转换工具
HSV CMYK 转换工具

HSV CMYK互换工具

  • New
  • 文章
  • 话题
  • 教程
    • 关注 码农网 公众号
    版权所有，保留一切权利！© 2018-2025 码农网 粤ICP备17054400号-3