首页 > 其他分享 >Go 框架 Gin使用 validator 若干实用技巧

Go 框架 Gin使用 validator 若干实用技巧

时间：2024-09-18 14:48:25浏览次数：9  
标签：实用技巧 自定义 err 校验 validator Go password string

Go 框架 Gin使用 validator 若干实用技巧

网管叨bi叨    2024年09月17日 10:39 北京  

以下文章来源于李文周 ，作者李文周

李文周.

一个北漂程序员的自我更新之旅。

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

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

基本示例

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

package main

import (
 "net/http"

 "github.com/gin-gonic/gin"
)

type SignUpParam struct {
 Age        uint8  `json:"age" binding:"gte=1,lte=130"`
 Name       string `json:"name" binding:"required"`
 Email      string `json:"email" binding:"required,email"`
 Password   string `json:"password" binding:"required"`
 RePassword string `json:"re_password" binding:"required,eqfield=Password"`
}

func main() {
 r := gin.Default()

 r.POST("/signup", func(c *gin.Context) {
  var u SignUpParam
  if 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 的检验生效了，但是错误提示的字段不是特别友好，我们可能需要将它翻译成中文。

翻译校验错误提示信息

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

package main

import (
 "fmt"
 "net/http"

 "github.com/gin-gonic/gin"
 "github.com/gin-gonic/gin/binding"
 "github.com/go-playground/locales/en"
 "github.com/go-playground/locales/zh"
 ut "github.com/go-playground/universal-translator"
 "github.com/go-playground/validator/v10"
 enTranslations "github.com/go-playground/validator/v10/translations/en"
 zhTranslations "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-Language'
  var 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:"gte=1,lte=130"`
 Name       string `json:"name" binding:"required"`
 Email      string `json:"email" binding:"required,email"`
 Password   string `json:"password" binding:"required"`
 RePassword string `json:"re_password" binding:"required,eqfield=Password"`
}

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 SignUpParam
  if err := c.ShouldBind(&u); err != nil {
   // 获取validator.ValidationErrors类型的errors
   errs, 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为必填字段"}}

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

上面的错误提示看起来是可以了，但是还是差点意思，首先是错误提示中的字段并不是请求中使用的字段，例如：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类型的errors
 errs, 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为必填字段"}}

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

自定义结构体校验方法

上面的校验还是有点小问题，就是当涉及到一些复杂的校验规则，比如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 {
  // 输出错误提示信息，最后一个参数就是传递的param
  sl.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字段的错误提示信息就符合我们预期了。

自定义字段校验方法

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

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

type SignUpParam struct {
 Age        uint8  `json:"age" binding:"gte=1,lte=130"`
 Name       string `json:"name" binding:"required"`
 Email      string `json:"email" binding:"required,email"`
 Password   string `json:"password" binding:"required"`
 RePassword string `json:"re_password" binding:"required,eqfield=Password"`
 // 需要使用自定义校验方法checkDate做参数校验的字段Date
 Date       string `json:"date" binding:"required,datetime=2006-01-02,checkDate"`
}

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

{"msg":{"date":"date的格式必须是2006-01-02"}}

针对date字段除了内置的datetime=2006-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":"[email protected]","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"}}

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

自定义翻译方法

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

// 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必须要晚于当前日期"}}

总结

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

网管叨bi叨 分享软件开发和系统架构设计基础、Go 语言和Kubernetes。 298篇原创内容 公众号

 

最后真心推荐一下我前段时间写的付费专栏，该专栏主要用一些实际的案例讲解项目业务需求分析、技术实现分析、模块划分和分层的方法论，讲明白这些底层逻辑后再教大家怎么用 UML 等工具把它们图形化表达出来。详细内容请可扫码观看，或访问：独家原创--程序员的全能画图课

图片

另外现在我在做的一个Go实战项目的专栏，整个项目也采用了这里讲的知识来实际地一步步做分析和开发的，大家可以先把这些方法论学会，后面跟着新专栏学习，看项目和代码的实践能明显提高自己做项目的架构水平。

阅读原文 阅读 1265   ​   写留言             网管叨bi叨     写留言 留言  

暂无留言

标签：实用技巧,自定义,err,校验,validator,Go,password,string
From： https://www.cnblogs.com/cheyunhua/p/18418484

相关文章

  • GO语言学习笔记之mac环境go语言配置
    此处仅为Mac的Iterm终端，关于go的环境配置，仅供参考cd~vim.zshrcGOROOT为go的安装路径，GOPATH为go安装额外的工具和具体的工具环境配置，GOPROXY将安装镜像源改为阿里云#goexportGOROOT="/usr/local/go"exportGOPATH="/Users/635458/go"exportPATH=$PATH:$GOROOT/bin:$GOPATH......
  • 高效数据移动指南 | 如何快速实现数据库 MySQL 到 MongoDB 的数据同步？
    在现代企业中，数据无处不在，贯穿于各个业务环节和系统之间。无论是跨系统的数据集成、多地域的数据协同，还是实时应用的数据同步，数据的一致性和及时性都至关重要。在数字化转型的过程中，如何确保不同系统、地域、设备之间的数据同步，成为了企业面临的重要挑战。本专题将基于实践经验，从......
  • 如何使用Go的结构体？
    在Go语言中，结构体（struct）是一种用户自定义的复合数据类型，它允许你将多个不同类型的数据项组合成一个单一的数据结构。结构体在Go中非常有用，常用于表示复杂的数据结构，如用户信息、配置设置等。定义结构体结构体可以通过struct关键字来定义。下面是一个基本的结构体定......
  • go实战全家桶优化goweb实现权限控制
    UML开源water/goweb控制端typeIrpcCheckAllowedinterface{   //测试开关、是否检查权限，方便测试可以关闭   IfCheckRes()bool   IfCheckSession()bool   //根据token获取useid的实现   RpcUserIdGetBySession(ctx*gin.Context,token......
  • Go Lang实现unix时间戳与日期格式互转
    /**golang几种post请求方式*参考:https://www.cnblogs.com/mafeng/p/7068837.html*/packagemainimport("fmt""io/ioutil""log""net/http""net/url""regexp""strings......
  • google pinying notepad++
    https://www.xitongzhijia.net/soft/199553.htmlhttps://www.techspot.com/downloads/2010-notepad.html......
  • 26.删除有序数组中的重复项 Golang实现
    题目描述：给你一个非严格递增排列的数组nums，请你原地删除重复出现的元素，使每个元素只出现一次，返回删除后数组的新长度。元素的相对顺序应该保持一致。然后返回nums中唯一元素的个数。考虑nums的唯一元素的数量为k，你需要做以下事情确保你的题解可以被通过：更改......
  • mongodb 中rs.stauts()命令参数解析
    转载请注明出处：rs.status()命令用于获取MongoDB副本集的状态信息。它提供了关于副本集中各个节点的详细信息，包括节点的健康状况、角色、选举状态等。以下是查看一个mongo集群状态返回的参数：rs0:PRIMARY>rs.status(){"set":"rs0","date":ISODa......
  • 基于python+Django的学生成绩管理系统(源码+vue+前后端分离)
    收藏关注不迷路！！......
  • 如何修改URL命名规则 让他更适合Google SEO优化
    为了使URL更符合Google的SEO优化标准，您可以遵循以下原则来修改您的URL命名规则：简洁明了：URL应该尽可能短小，避免冗长的路径。使用有意义的单词而不是数字或不相关的短语。使用关键词：在URL中包含目标关键词可以帮助搜索引擎理解页面的内容。但是避免过度堆砌关键词，这可......

赞助商

阅读排行

网站主页关于我们联系我们网站地图

本网站内容转载自其他媒体，侵权联系[admin##ips99.com]。

Copyright © 2020-2023 IPS99 版权所有 IPS99