设为首页 收藏本站
查看: 603|回复: 0

[经验分享] golang 如何验证struct字段的数据格式

[复制链接]

尚未签到

发表于 2018-9-20 11:55:27 | 显示全部楼层 |阅读模式
package main  import (
  "fmt"
  "reflect"
  "regexp"
  "strings"
  )
  const tagName = "validate"
  //邮箱验证正则
  var mailRe = regexp.MustCompile(`\A[\w+\-.]+@[a-z\d\-]+(\.[a-z]+)*\.[a-z]+\z`)
  //验证接口
  type Validator interface {
  Validate(interface{}) (bool, error)
  }
  type DefaultValidator struct {
  }
  func (v DefaultValidator) Validate(val interface{}) (bool, error) {
  return true, nil
  }
  type StringValidator struct {
  Min int
  Max int
  }
  func (v StringValidator) Validate(val interface{}) (bool, error) {
  l := len(val.(string))
  if l == 0 {
  return false, fmt.Errorf("cannot be blank")
  }
  if l < v.Min {
  return false, fmt.Errorf(&quot;should be at least %v chars long&quot;, v.Min)
  }
  if v.Max >= v.Min && l > v.Max {
  return false, fmt.Errorf(&quot;should be less than %v chars long&quot;, v.Max)
  }
  return true, nil
  }
  type NumberValidator struct {
  Min int
  Max int
  }
  func (v NumberValidator) Validate(val interface{}) (bool, error) {
  num := val.(int)
  if num < v.Min {
  return false, fmt.Errorf(&quot;should be greater than %v&quot;, v.Min)
  }
  if v.Max >= v.Min && num > v.Max {
  return false, fmt.Errorf(&quot;should be less than %v&quot;, v.Max)
  }
  return true, nil
  }
  type EmailValidator struct {
  }
  func (v EmailValidator) Validate(val interface{}) (bool, error) {
  if !mailRe.MatchString(val.(string)) {
  return false, fmt.Errorf(&quot;is not a valid email address&quot;)
  }
  return true, nil
  }
  func getValidatorFromTag(tag string) Validator {
  args := strings.Split(tag, &quot;,&quot;)
  switch args[0] {
  case &quot;number&quot;:
  validator := NumberValidator{}
  //将structTag中的min和max解析到结构体中
  fmt.Sscanf(strings.Join(args[1:], &quot;,&quot;), &quot;min=%d,max=%d&quot;, &validator.Min, &validator.Max)
  return validator
  case &quot;string&quot;:
  validator := StringValidator{}
  fmt.Sscanf(strings.Join(args[1:], &quot;,&quot;), &quot;min=%d,max=%d&quot;, &validator.Min, &validator.Max)
  return validator
  case &quot;email&quot;:
  return EmailValidator{}
  }
  return DefaultValidator{}
  }
  func validateStruct(s interface{}) []error {
  errs := []error{}
  v := reflect.ValueOf(s)
  for i := 0; i < v.NumField(); i++ {
  //利用反射获取structTag
  tag := v.Type().Field(i).Tag.Get(tagName)
  if tag == &quot;&quot; || tag == &quot;-&quot; {
  continue
  }
  validator := getValidatorFromTag(tag)
  valid, err := validator.Validate(v.Field(i).Interface())
  if !valid && err != nil {
  errs = append(errs, fmt.Errorf(&quot;%s %s&quot;, v.Type().Field(i).Name, err.Error()))
  }
  }
  return errs
  }
  type User struct {
  Id    int    `validate:&quot;number,min=1,max=1000&quot;`
  Name  string `validate:&quot;string,min=2,max=10&quot;`
  Bio   string `validate:&quot;string&quot;`
  Email string `validate:&quot;email&quot;`
  }
  func main() {
  user := User{
  Id:    0,
  Name:  &quot;superlongstring&quot;,
  Bio:   &quot;&quot;,
  Email: &quot;foobar&quot;,
  }
  fmt.Println(&quot;Errors:&quot;)
  for i, err := range validateStruct(user) {
  fmt.Printf(&quot;\t%d. %s\n&quot;, i+1, err.Error())
  }
  }


运维网声明 1、欢迎大家加入本站运维交流群:群②:261659950 群⑤:202807635 群⑦870801961 群⑧679858003
2、本站所有主题由该帖子作者发表,该帖子作者与运维网享有帖子相关版权
3、所有作品的著作权均归原作者享有,请您和我们一样尊重他人的著作权等合法权益。如果您对作品感到满意,请购买正版
4、禁止制作、复制、发布和传播具有反动、淫秽、色情、暴力、凶杀等内容的信息,一经发现立即删除。若您因此触犯法律,一切后果自负,我们对此不承担任何责任
5、所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其内容的准确性、可靠性、正当性、安全性、合法性等负责,亦不承担任何法律责任
6、所有作品仅供您个人学习、研究或欣赏,不得用于商业或者其他用途,否则,一切后果均由您自己承担,我们对此不承担任何法律责任
7、如涉及侵犯版权等问题,请您及时通知我们,我们将立即采取措施予以解决
8、联系人Email:admin@iyunv.com 网址:www.yunweiku.com

所有资源均系网友上传或者通过网络收集,我们仅提供一个展示、介绍、观摩学习的平台,我们不对其承担任何法律责任,如涉及侵犯版权等问题,请您及时通知我们,我们将立即处理,联系人Email:kefu@iyunv.com,QQ:1061981298 本贴地址:https://www.yunweiku.com/thread-598841-1-1.html 上篇帖子: golang 性能测试pprof 下篇帖子: golang-nsq消息队列应用
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

扫码加入运维网微信交流群X

扫码加入运维网微信交流群

扫描二维码加入运维网微信交流群,最新一手资源尽在官方微信交流群!快快加入我们吧...

扫描微信二维码查看详情

客服E-mail:kefu@iyunv.com 客服QQ:1061981298


QQ群⑦:运维网交流群⑦ QQ群⑧:运维网交流群⑧ k8s群:运维网kubernetes交流群


提醒:禁止发布任何违反国家法律、法规的言论与图片等内容;本站内容均来自个人观点与网络等信息,非本站认同之观点.


本站大部分资源是网友从网上搜集分享而来,其版权均归原作者及其网站所有,我们尊重他人的合法权益,如有内容侵犯您的合法权益,请及时与我们联系进行核实删除!



合作伙伴: 青云cloud

快速回复 返回顶部 返回列表