golang之数据验证validator的实现
作者：无风的雨 发布时间：2024-04-26 17:36:03
前言
在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: 必须等于 Field 的值；
nefield=Field: 必须不等于 Field 的值；
gtfield=Field: 必须大于 Field 的值；
gtefield=Field: 必须大于等于 Field 的值；
ltfield=Field: 必须小于 Field 的值；
ltefield=Field: 必须小于等于 Field 的值；
eqcsfield=Other.Field: 必须等于 struct Other 中 Field 的值；
necsfield=Other.Field: 必须不等于 struct Other 中 Field 的值；
gtcsfield=Other.Field: 必须大于 struct Other 中 Field 的值；
gtecsfield=Other.Field: 必须大于等于 struct Other 中 Field 的值；
ltcsfield=Other.Field: 必须小于 struct Other 中 Field 的值；
ltecsfield=Other.Field: 必须小于等于 struct 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-playground/validator.v10
自定义验证类型
示例：
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上提供了很好的解决方法。
先自行安装需要的两个包
https://github.com/go-playground/locales
https://github.com/go-playground/universal-translator
执行：
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.aspxhome.com/article/197866.htm>，
照着做没有成功（可能有遗漏吧），最后还是翻看了一下源代码，在<https://github.com/go-playground/validator/blob/master/validator_instance.go，第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")
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&check_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-playground/validator
https://github.com/gin-gonic/gin
https://gitissue.com/issues/5d06a73965d56f73569b825f
https://segmentfault.com/a/1190000022527284
来源：https://blog.csdn.net/guyan0319/article/details/105918559/
0
投稿猜你喜欢
最终效果前言这是最近在学qt这个东西，然后又学会了调用api，然后就想了用pyqt5做一个GUI界面，后期也可以打包分享给其他人使用，所以就- 函数先定义函数，后调用一、定义函数：1、简单的规则：函数代码块以 def 关键词开头，后接函数标识符名称和圆括号&nbs
- 一： 安装命令jupyter： pip install jupyter如果缺少依赖，缺啥装啥二： 运行 jupyter notebook首先
- 之前我们曾经分享过：Python获取某一日期是“星期几”的6种方法！实际上，在我们使用Python处
- 一、Tensorlow结构import tensorflow as tfimport numpy as np#创建数据x_data = np
- dispy，是用asyncoro实现的分布式并行计算框架。框架也是非常精简，只有4个组件，在其源码文件夹下可以找到：dispy.py (cl
- Python 变量类型变量存储在内存中的值。这就意味着在创建变量时会在内存中开辟一个空间。基于变量的数据类型，解释器会分配指定内存，并决定什
- 一、停止数据库服务打开sqlserver自带的配置管理器，停止数据库服务。二、打开cmd窗口，执行命令(以管理员运行)以单用户模式启动sql
- 本文实例讲述了JS求解三元一次方程组值的方法。分享给大家供大家参考，具体如下：// 求用js 码一段代码求 三元一次方程组的值！// a1
- Python 模块EasyGui详细介绍前言：在Windows想用Python开发一些简单的界面，所以找到了很容易上手的EasyGui库。下
- 本文实例讲述了Python机器学习库scikit-learn安装与基本使用。分享给大家供大家参考，具体如下：引言scikit-learn是P
- 前言做一个简单的日志数据库功能不需要特别繁琐主要就是记录普通日志和错误日志（INFO,ERROR）用数据库作为日志有好处也有坏处相比于文本来
- pip install命令用于安装扩展库，由于安装扩展库需要从国外网站下载，速度较慢，可以使用-i选项设置临时使用国内的镜像网站。命令格式：
- 前言本笔记通过记录 数据包 在网络中的生命履历来引出一些网络基础知识，如：MAC、ARP、IP、子网掩码、网关、集线器、交换机、路由器这些概
- 目录1.celery异步消息队列介绍celery应用举例Celery有以下优点Celery 特性2.工作原理 *****Celery 扮演生
- 首先来描述下环境，在机器上有很多个JAVA程序，我们在每个JAVA程序里都配置了一个启动|停止|重启的脚本举个例子：我们现在要同时运行这些脚
- 一，什么是JSON文件JSON和XML都是互联网上数据交换的主要载体。在JSON出现之前，大家一直用XML来传递数据。因为XML是一种纯文本
- 1.先停止mysqld.exe的进程2.打开cmd进入到你mysql的bin目录下输入此命令：mysqld --skip-grant-tab
- 背景有时候爬虫爬过的url需要进行指纹核对，比如Scrapy就是进行指纹核对，如果是指纹重复则不再爬取。当然在入库的时候我还是需要做一次核对
- Beautiful Soup 库一般被称为bs4库，支持Python3，是我们写爬虫非常好的第三方库。因用起来十分的简便流畅。所以也被人叫做
