validator库参数校验若干实用技巧

本文介绍了使用validator库做参数校验的一些十分实用的使用技巧,包括翻译校验错误提示信息、自定义提示信息的字段名称、自定义校验方法等。

validator库参数校验若干实用技巧

在web开发中一个不可避免的环节就是对请求参数进行校验,通常我们会在代码中定义与请求参数相对应的模型(结构体),借助模型绑定快捷地解析请求中的参数,例如 gin 框架中的BindShouldBind系列方法。本文就以 gin 框架的请求参数校验为例,介绍一些validator库的实用技巧。

gin框架使用github.com/go-playground/validator进行参数校验,目前已经支持github.com/go-playground/validator/v10了,我们需要在定义结构体时使用 binding tag标识相关校验规则,可以查看validator文档查看支持的所有 tag。

基本示例

首先来看gin框架内置使用validator做参数校验的基本示例。

我们使用curl发送一个POST请求测试下:

输出结果:

从最终的输出结果可以看到 validator 的检验生效了,但是错误提示的字段不是特别友好,我们可能需要将它翻译成中文。

翻译校验错误提示信息

validator库本身是支持国际化的,借助相应的语言包可以实现校验错误提示信息的自动翻译。下面的示例代码演示了如何将错误提示信息翻译成中文,翻译成其他语言的方法类似。

同样的请求再来一次:

这一次的输出结果如下:

自定义错误提示信息的字段名

上面的错误提示看起来是可以了,但是还是差点意思,首先是错误提示中的字段并不是请求中使用的字段,例如:RePassword是我们后端定义的结构体中的字段名,而请求中使用的是re_password字段。如何是错误提示中的字段使用自定义的名称,例如jsontag指定的值呢?

只需要在初始化翻译器的时候像下面一样添加一个获取json tag的自定义方法即可。

再尝试发请求,看一下效果:

可以看到现在错误提示信息中使用的就是我们结构体中jsontag设置的名称了。

但是还是有点瑕疵,那就是最终的错误提示信息中心还是有我们后端定义的结构体名称——SignUpParam,这个名称其实是不需要随错误提示返回给前端的,前端并不需要这个值。我们需要想办法把它去掉。

这里参考https://github.com/go-playground/validator/issues/633#issuecomment-654382345提供的方法,定义一个去掉结构体名称前缀的自定义方法:

我们在代码中使用上述函数将翻译后的errors做一下处理即可:

看一下最终的效果:

这一次看起来就比较符合我们预期的标准了。

自定义结构体校验方法

上面的校验还是有点小问题,就是当涉及到一些复杂的校验规则,比如re_password字段需要与password字段的值相等这样的校验规则,我们的自定义错误提示字段名称方法就不能很好解决错误提示信息中的其他字段名称了。

最后输出的错误提示信息如下:

可以看到re_password字段的提示信息中还是出现了Password这个结构体字段名称。这有点小小的遗憾,毕竟自定义字段名称的方法不能影响被当成param传入的值。

此时如果想要追求更好的提示效果,将上面的Password字段也改为和json tag一致的名称,就需要我们自定义结构体校验的方法。

例如,我们为SignUpParam自定义一个校验方法如下:

然后在初始化校验器的函数中注册该自定义校验方法即可:

最终再请求一次,看一下效果:

这一次re_password字段的错误提示信息就符合我们预期了。

自定义字段校验方法

除了上面介绍到的自定义结构体校验方法,validator还支持为某个字段自定义校验方法,并使用RegisterValidation()注册到校验器实例中。

接下来我们来为SignUpParam添加一个需要使用自定义校验方法checkDate做参数校验的字段Date

其中datetime=2006-01-02是内置的用于校验日期类参数是否满足指定格式要求的tag。 如果传入的date参数不满足2006-01-02这种格式就会提示如下错误:

针对date字段除了内置的datetime=2006-01-02提供的格式要求外,假设我们还要求该字段的时间必须是一个未来的时间(晚于当前时间),像这样针对某个字段的特殊校验需求就需要我们使用自定义字段校验方法了。

首先我们要在需要执行自定义校验的字段后面添加自定义tag,这里使用的是checkDate,注意使用英文分号分隔开。

定义好了字段及其自定义校验方法后,就需要将它们联系起来并注册到我们的校验器实例中。

这样,我们就可以对请求参数中date字段执行自定义的checkDate进行校验了。 我们发送如下请求测试一下:

此时得到的响应结果是:

这…自定义字段级别的校验方法的错误提示信息很“简单粗暴”,和我们上面的中文提示风格有出入,必须想办法搞定它呀!

自定义翻译方法

我们现在需要为自定义字段校验方法提供一个自定义的翻译方法,从而实现该字段错误提示信息的自定义显示。

定义好了相关翻译方法之后,我们在InitTrans函数中通过调用RegisterTranslation()方法来注册我们自定义的翻译方法。

这样再次尝试发送请求,就能得到想要的错误提示信息了。

总结

由于本篇博客示例代码较多,我已经把文中示例代码上传到我的github仓库——https://github.com/Q1mi/validator_demo,大家可以查看完整的示例代码。

本文总结的gin框架中validator的使用技巧同样也适用于直接使用validator库,区别仅仅在于我们配置的是gin框架中的校验器还是由validator.New()创建的校验器。同时使用validator库确实能够在一定程度上减少我们的编码量,但是它不太可能完美解决我们所有需求,所以你需要找到两者之间的平衡点。

参考链接:

https://github.com/go-playground/validator/blob/master/_examples/simple/main.go

https://github.com/go-playground/validator/blob/master/_examples/translations/main.go

https://github.com/go-playground/validator/issues/567

https://github.com/go-playground/validator/issues/633

https://github.com/go-playground/validator/issues/551

本文为转载文章,贵在分享,版权归原作者及原出处所有,如涉及版权等问题,请及时与我联系。
原文出处:李文周的博客
原文链接:https://www.liwenzhou.com/posts/Go/validator_usages/

发表评论

电子邮件地址不会被公开。 必填项已用*标注