网站留言板带后台模板,域名注册免费申请,wordpress 淘客主题,网站开发的经济效益分析在web开发中对请求参数进行校验#xff0c;通常在代码中定义与请求参数相对应的模型#xff08;结构体#xff09;#xff0c;借助模型绑定快捷地解析请求中的参数#xff0c;例如 gin 框架中的Bind和ShouldBind系列方法。
gin框架使用github.com/go-playground/validato…在web开发中对请求参数进行校验通常在代码中定义与请求参数相对应的模型结构体借助模型绑定快捷地解析请求中的参数例如 gin 框架中的Bind和ShouldBind系列方法。
gin框架使用github.com/go-playground/validator进行参数校验目前已经支持github.com/go-playground/validator/v10了我们需要在定义结构体时使用 binding tag标识相关校验规则可以查看validator文档查看支持的所有 tag。
1 基本校验
首先来看gin框架内置使用validator做参数校验的基本示例。
package mainimport (net/httpgithub.com/gin-gonic/gin
)type SignUpParam struct {Age uint8 json:age binding:gte1,lte130Name string json:name binding:requiredEmail string json:email binding:required,emailPassword string json:password binding:requiredRePassword string json:re_password binding:required,eqfieldPassword
}func main() {r : gin.Default()r.POST(/signup, func(c *gin.Context) {var u SignUpParamif err : c.ShouldBind(u); err ! nil {c.JSON(http.StatusOK, gin.H{msg: err.Error(),})return}// 保存入库等业务逻辑代码...c.JSON(http.StatusOK, success)})_ r.Run(:8999)
}我们使用curl发送一个POST请求测试下
curl -H Content-type: application/json -X POST -d {name:q1mi,age:18,email:123.com} http://127.0.0.1:8999/signup输出结果
{msg:Key: SignUpParam.Email Error:Field validation for Email failed on the email tag\nKey: SignUpParam.Password Error:Field validation for Password failed on the required tag\nKey: SignUpParam.RePassword Error:Field validation for RePassword failed on the required tag}从结果看出 validator 的检验虽然生效了但是错误提示的字段不是特别友好我们可能需要将它翻译成中文。
2 翻译校验错误提示信息
validator库本身是支持国际化的借助相应的语言包可以实现校验错误提示信息的自动翻译。
package mainimport (fmtnet/httpgithub.com/gin-gonic/gingithub.com/gin-gonic/gin/bindinggithub.com/go-playground/locales/engithub.com/go-playground/locales/zhut github.com/go-playground/universal-translatorgithub.com/go-playground/validator/v10enTranslations github.com/go-playground/validator/v10/translations/enzhTranslations github.com/go-playground/validator/v10/translations/zh
)// 定义一个全局翻译器T
var trans ut.Translator// InitTrans 初始化翻译器
func InitTrans(locale string) (err error) {// 修改gin框架中的Validator引擎属性实现自定制if v, ok : binding.Validator.Engine().(*validator.Validate); ok {zhT : zh.New() // 中文翻译器enT : en.New() // 英文翻译器// 第一个参数是备用fallback的语言环境// 后面的参数是应该支持的语言环境支持多个// uni : ut.New(zhT, zhT) 也是可以的uni : ut.New(enT, zhT, enT)// locale 通常取决于 http 请求头的 Accept-Languagevar ok bool// 也可以使用 uni.FindTranslator(...) 传入多个locale进行查找trans, ok uni.GetTranslator(locale)if !ok {return fmt.Errorf(uni.GetTranslator(%s) failed, locale)}// 注册翻译器switch locale {case en:err enTranslations.RegisterDefaultTranslations(v, trans)case zh:err zhTranslations.RegisterDefaultTranslations(v, trans)default:err enTranslations.RegisterDefaultTranslations(v, trans)}return}return
}type SignUpParam struct {Age uint8 json:age binding:gte1,lte130Name string json:name binding:requiredEmail string json:email binding:required,emailPassword string json:password binding:requiredRePassword string json:re_password binding:required,eqfieldPassword
}func main() {if err : InitTrans(zh); err ! nil {fmt.Printf(init trans failed, err:%v\n, err)return}r : gin.Default()r.POST(/signup, func(c *gin.Context) {var u SignUpParamif err : c.ShouldBind(u); err ! nil {// 获取validator.ValidationErrors类型的errorserrs, ok : err.(validator.ValidationErrors)if !ok {// 非validator.ValidationErrors类型错误直接返回c.JSON(http.StatusOK, gin.H{msg: err.Error(),})return}// validator.ValidationErrors类型错误则进行翻译c.JSON(http.StatusOK, gin.H{msg:errs.Translate(trans),})return}// 保存入库等具体业务逻辑代码...c.JSON(http.StatusOK, success)})_ r.Run(:8999)
}同样的请求再来一次
curl -H Content-type: application/json -X POST -d {name:q1mi,age:18,email:123.com} http://127.0.0.1:8999/signup这一次的输出结果如下
{msg:{SignUpParam.Email:Email必须是一个有效的邮箱,SignUpParam.Password:Password为必填字段,SignUpParam.RePassword:RePassword为必填字段}}3 自定义错误提示信息的字段名
上面的错误提示看起来是可以了但是还是差点意思首先是错误提示中的字段并不是请求中使用的字段例如RePassword是我们后端定义的结构体中的字段名而请求中使用的是re_password字段。如何是错误提示中的字段使用自定义的名称例如jsontag指定的值呢
只需要在初始化翻译器的时候像下面一样添加一个获取json tag的自定义方法即可。
// InitTrans 初始化翻译器
func InitTrans(locale string) (err error) {// 修改gin框架中的Validator引擎属性实现自定制if v, ok : binding.Validator.Engine().(*validator.Validate); ok {// 注册一个获取json tag的自定义方法v.RegisterTagNameFunc(func(fld reflect.StructField) string {name : strings.SplitN(fld.Tag.Get(json), ,, 2)[0]if name - {return }return name})zhT : zh.New() // 中文翻译器enT : en.New() // 英文翻译器// 第一个参数是备用fallback的语言环境// 后面的参数是应该支持的语言环境支持多个// uni : ut.New(zhT, zhT) 也是可以的uni : ut.New(enT, zhT, enT)// ... liwenzhou.com ...
}再尝试发请求看一下效果
{msg:{SignUpParam.email:email必须是一个有效的邮箱,SignUpParam.password:password为必填字段,SignUpParam.re_password:re_password为必填字段}}可以看到现在错误提示信息中使用的就是我们结构体中jsontag设置的名称了。
但是还是有点瑕疵那就是最终的错误提示信息中心还是有我们后端定义的结构体名称——SignUpParam这个名称其实是不需要随错误提示返回给前端的前端并不需要这个值。我们需要想办法把它去掉。
这里参考https://github.com/go-playground/validator/issues/633#issuecomment-654382345提供的方法定义一个去掉结构体名称前缀的自定义方法
func removeTopStruct(fields map[string]string) map[string]string {res : map[string]string{}for field, err : range fields {res[field[strings.Index(field, .)1:]] err}return res
}我们在代码中使用上述函数将翻译后的errors做一下处理即可
if err : c.ShouldBind(u); err ! nil {// 获取validator.ValidationErrors类型的errorserrs, ok : err.(validator.ValidationErrors)if !ok {// 非validator.ValidationErrors类型错误直接返回c.JSON(http.StatusOK, gin.H{msg: err.Error(),})return}// validator.ValidationErrors类型错误则进行翻译// 并使用removeTopStruct函数去除字段名中的结构体名称标识c.JSON(http.StatusOK, gin.H{msg: removeTopStruct(errs.Translate(trans)),})return
}看一下最终的效果
{msg:{email:email必须是一个有效的邮箱,password:password为必填字段,re_password:re_password为必填字段}}这一次看起来就比较符合我们预期的标准了。
4 自定义结构体校验方法
上面的校验还是有点小问题就是当涉及到一些复杂的校验规则比如re_password字段需要与password字段的值相等这样的校验规则我们的自定义错误提示字段名称方法就不能很好解决错误提示信息中的其他字段名称了。
curl -H Content-type: application/json -X POST -d {name:q1mi,age:18,email:123.com,password:123,re_password:321} http://127.0.0.1:8999/signup最后输出的错误提示信息如下
{msg:{email:email必须是一个有效的邮箱,re_password:re_password必须等于Password}}可以看到re_password字段的提示信息中还是出现了Password这个结构体字段名称。这有点小小的遗憾毕竟自定义字段名称的方法不能影响被当成param传入的值。
此时如果想要追求更好的提示效果将上面的Password字段也改为和json tag一致的名称就需要我们自定义结构体校验的方法。
例如我们为SignUpParam自定义一个校验方法如下
// SignUpParamStructLevelValidation 自定义SignUpParam结构体校验函数
func SignUpParamStructLevelValidation(sl validator.StructLevel) {su : sl.Current().Interface().(SignUpParam)if su.Password ! su.RePassword {// 输出错误提示信息最后一个参数就是传递的paramsl.ReportError(su.RePassword, re_password, RePassword, eqfield, password)}
}然后在初始化校验器的函数中注册该自定义校验方法即可
func InitTrans(locale string) (err error) {// 修改gin框架中的Validator引擎属性实现自定制if v, ok : binding.Validator.Engine().(*validator.Validate); ok {// ... liwenzhou.com ...// 为SignUpParam注册自定义校验方法v.RegisterStructValidation(SignUpParamStructLevelValidation, SignUpParam{})zhT : zh.New() // 中文翻译器enT : en.New() // 英文翻译器// ... liwenzhou.com ...
}最终再请求一次看一下效果
{msg:{email:email必须是一个有效的邮箱,re_password:re_password必须等于password}}这一次re_password字段的错误提示信息就符合我们预期了。
5 自定义字段校验方法
除了上面介绍到的自定义结构体校验方法validator还支持为某个字段自定义校验方法并使用RegisterValidation()注册到校验器实例中。
接下来我们来为SignUpParam添加一个需要使用自定义校验方法checkDate做参数校验的字段Date。
type SignUpParam struct {Age uint8 json:age binding:gte1,lte130Name string json:name binding:requiredEmail string json:email binding:required,emailPassword string json:password binding:requiredRePassword string json:re_password binding:required,eqfieldPassword// 需要使用自定义校验方法checkDate做参数校验的字段DateDate string json:date binding:required,datetime2006-01-02,checkDate
}其中datetime2006-01-02是内置的用于校验日期类参数是否满足指定格式要求的tag。 如果传入的date参数不满足2006-01-02这种格式就会提示如下错误
{msg:{date:date的格式必须是2006-01-02}}针对date字段除了内置的datetime2006-01-02提供的格式要求外假设我们还要求该字段的时间必须是一个未来的时间晚于当前时间像这样针对某个字段的特殊校验需求就需要我们使用自定义字段校验方法了。
首先我们要在需要执行自定义校验的字段后面添加自定义tag这里使用的是checkDate注意使用英文分号分隔开。
// customFunc 自定义字段级别校验方法
func customFunc(fl validator.FieldLevel) bool {date, err : time.Parse(2006-01-02, fl.Field().String())if err ! nil {return false}if date.Before(time.Now()) {return false}return true
}定义好了字段及其自定义校验方法后就需要将它们联系起来并注册到我们的校验器实例中。
// 在校验器注册自定义的校验方法
if err : v.RegisterValidation(checkDate, customFunc); err ! nil {return err
}这样我们就可以对请求参数中date字段执行自定义的checkDate进行校验了。 我们发送如下请求测试一下
curl -H Content-type: application/json -X POST -d {name:q1mi,age:18,email:123qq.com,password:123, re_password: 123, date:2020-01-02} http://127.0.0.1:8999/signup此时得到的响应结果是
{msg:{date:Key: SignUpParam.date Error:Field validation for date failed on the checkDate tag}}这…自定义字段级别的校验方法的错误提示信息很“简单粗暴”和我们上面的中文提示风格有出入必须想办法搞定它呀
6 自定义翻译方法
我们现在需要为自定义字段校验方法提供一个自定义的翻译方法从而实现该字段错误提示信息的自定义显示。
// registerTranslator 为自定义字段添加翻译功能
func registerTranslator(tag string, msg string) validator.RegisterTranslationsFunc {return func(trans ut.Translator) error {if err : trans.Add(tag, msg, false); err ! nil {return err}return nil}
}// translate 自定义字段的翻译方法
func translate(trans ut.Translator, fe validator.FieldError) string {msg, err : trans.T(fe.Tag(), fe.Field())if err ! nil {panic(fe.(error).Error())}return msg
}定义好了相关翻译方法之后我们在InitTrans函数中通过调用RegisterTranslation()方法来注册我们自定义的翻译方法。
// InitTrans 初始化翻译器
func InitTrans(locale string) (err error) {// ...liwenzhou.com...// 注册翻译器switch locale {case en:err enTranslations.RegisterDefaultTranslations(v, trans)case zh:err zhTranslations.RegisterDefaultTranslations(v, trans)default:err enTranslations.RegisterDefaultTranslations(v, trans)}if err ! nil {return err}// 注意因为这里会使用到trans实例// 所以这一步注册要放到trans初始化的后面if err : v.RegisterTranslation(checkDate,trans,registerTranslator(checkDate, {0}必须要晚于当前日期),translate,); err ! nil {return err}return}return
}这样再次尝试发送请求就能得到想要的错误提示信息了。
{msg:{date:date必须要晚于当前日期}}参考文章
https://www.fansimao.com/937560.html
