首页 > 其他分享 >golang接口请求结构体验证器Validator实现

golang接口请求结构体验证器Validator实现

时间：2024-06-06 15:11:17浏览次数：12  
标签：case reflect return string val 接口 golang Validator value

一、前提：认识reflect.TypeOf及reflect.ValueOf

TypeOf:

  • 动态的获取从函数接口中传进去的变量的类型，如果为空则返回值为 nil（获取类型对象）

  • 可以从该方法获取的对象中拿到字段的所属类型，字段名，以及该字段是否是匿名字段等信息。

  • 还可以获取到与该字段进行绑定的 tag。

ValueOf：

  • 获取从函数入口传进去变量的值（获取值对象）该方法获取到的对象是对应的数据信息区域（具体的值可以通过 Filed(index) 直接拿到对应位置的值）FieldByName（字段名）

  • 可以通过该函数拿到成员属性的详细信息（返回值与ValueOf一样）

  • 注：指针类型经过以上函数拿到的还是指针类型，结构体还是结构体类型。

Elem()方法：

  • 该方法只接受指针类型与接口类型，如果不是这两种类型就会抛出异常（相当于对指针进行取元素）

示例：

package main

import (
	"fmt"
	"reflect"
)

// 定义一个结构体
type stu struct {
	id   string
	name string
	age  int
}

func main() {
	//对结构体进行初始化
	var stu1 stu
	stu1.age = 20
	stu1.id = "hahaha"
	stu1.name = "小明"
	sTructType := reflect.TypeOf(stu1)
	sTructValue := reflect.ValueOf(stu1)

	// 1. 类型
	fmt.Println("类型:", sTructType)
	// 2. 结构体值
	fmt.Println("值是:", sTructValue)
	// 3. 获取属性个数
	fmt.Println(sTructType.NumField())
	// 4. 获取指定位置属性的详细类型
	fmt.Println(sTructType.Field(0))
	// 5. 直接获取到结构体名
	fmt.Println("类型是:", sTructType.Name())
	// 6. 获取该类型最初的基本
	fmt.Println(sTructType.Kind())

	//打印结果
	//  1. 类型: main.stu
	//  2. 值是: {hahaha 小明 20}
	//  3. 3
	//  4. {id main string  0 [0] false}
	//  Field输出以上结果含义为
	//	| Type      Type      // field type字段名
	//	| Tag       StructTag // field tag string所在包
	//	| Offset    uintptr   // offset within struct, in bytes//基类型名称
	//	| Index     []int     // index sequence for Type.FieldByIndex//下标
	//	| Anonymous bool      // is an embedded field//是不是匿名字段
	//  5. 类型是: stu
	//  6. struct
}

二、实现请求结构体验证器

1. 对reqStruct设置规则

示例，登录接口请求结构体规则：

LoginVerify = Rules{"CaptchaId": {NotEmpty()}, "Username": {NotEmpty()}, "Password": {NotEmpty()}}

2. validator实现

1. Rules结构，RulesMap结构
type Rules map[string][]string

type RulesMap map[string]Rules
2. 规则名称定义
//@description: 非空 不能为其对应类型的0值
func NotEmpty() string {
	return "notEmpty"
}

// @description: 正则校验 校验输入项是否满足正则表达式
func RegexpMatch(rule string) string {
	return "regexp=" + rule
}

//@description: 小于入参(<) 如果为string array Slice则为长度比较 如果是 int uint float 则为数值比较
func Lt(mark string) string {
	return "lt=" + mark
}

//@description: 等于入参(==) 如果为string array Slice则为长度比较 如果是 int uint float 则为数值比较
func Eq(mark string) string {
	return "eq=" + mark
}

//@description: 不等于入参(!=)  如果为string array Slice则为长度比较 如果是 int uint float 则为数值比较
func Ne(mark string) string {
	return "ne=" + mark
}

//@description: 大于入参(>) 如果为string array Slice则为长度比较 如果是 int uint float 则为数值比较
func Gt(mark string) string {
	return "gt=" + mark
}
3. 校验方法
func Verify(st interface{}, roleMap Rules) (err error) {
	compareMap := map[string]bool{
		"lt": true,
		"le": true,
		"eq": true,
		"ne": true,
		"ge": true,
		"gt": true,
	}

	sTructType := reflect.TypeOf(st)
	sTructValue := reflect.ValueOf(st) // 获取reflect.Type类型

	kd := sTructValue.Kind() // 获取到st对应的类别
	if kd != reflect.Struct {
		return errors.New("expect struct")
	}
	num := sTructValue.NumField()
	// 遍历结构体的所有字段
	for i := 0; i < num; i++ {
		tagVal := sTructType.Field(i)
		val := sTructValue.Field(i)
		if tagVal.Type.Kind() == reflect.Struct {
			if err = Verify(val.Interface(), roleMap); err != nil {
				return err
			}
		}
		if len(roleMap[tagVal.Name]) > 0 {
			for _, v := range roleMap[tagVal.Name] {
				switch {
				case v == "notEmpty":
					if isBlank(val) {
						return errors.New(tagVal.Name + "值不能为空")
					}
				case strings.Split(v, "=")[0] == "regexp":
					if !regexpMatch(strings.Split(v, "=")[1], val.String()) {
						return errors.New(tagVal.Name + "格式校验不通过")
					}
				case compareMap[strings.Split(v, "=")[0]]:
					if !compareVerify(val, v) {
						return errors.New(tagVal.Name + "长度或值不在合法范围," + v)
					}
				}
			}
		}
	}
	return nil
}
4. 检验各类方法
//@description: 非空校验
func isBlank(value reflect.Value) bool {
	switch value.Kind() {
	case reflect.String, reflect.Slice:
		return value.Len() == 0
	case reflect.Bool:
		return !value.Bool()
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		return value.Int() == 0
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		return value.Uint() == 0
	case reflect.Float32, reflect.Float64:
		return value.Float() == 0
	case reflect.Interface, reflect.Ptr:
		return value.IsNil()
	}
	return reflect.DeepEqual(value.Interface(), reflect.Zero(value.Type()).Interface())
}

//@description: 正则校验
func regexpMatch(rule, matchStr string) bool {
	return regexp.MustCompile(rule).MatchString(matchStr)
}

//@description: 长度和数字的校验方法 根据类型自动校验
func compareVerify(value reflect.Value, VerifyStr string) bool {
	switch value.Kind() {
	case reflect.String:
		return compare(len([]rune(value.String())), VerifyStr)
	case reflect.Slice, reflect.Array:
		return compare(value.Len(), VerifyStr)
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		return compare(value.Uint(), VerifyStr)
	case reflect.Float32, reflect.Float64:
		return compare(value.Float(), VerifyStr)
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		return compare(value.Int(), VerifyStr)
	default:
		return false
	}
}

//@description: 比较函数
func compare(value interface{}, VerifyStr string) bool {
	VerifyStrArr := strings.Split(VerifyStr, "=")
	val := reflect.ValueOf(value)
	switch val.Kind() {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		VInt, VErr := strconv.ParseInt(VerifyStrArr[1], 10, 64)
		if VErr != nil {
			return false
		}
		switch {
		case VerifyStrArr[0] == "lt":
			return val.Int() < VInt
		case VerifyStrArr[0] == "le":
			return val.Int() <= VInt
		case VerifyStrArr[0] == "eq":
			return val.Int() == VInt
		case VerifyStrArr[0] == "ne":
			return val.Int() != VInt
		case VerifyStrArr[0] == "ge":
			return val.Int() >= VInt
		case VerifyStrArr[0] == "gt":
			return val.Int() > VInt
		default:
			return false
		}
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		VInt, VErr := strconv.Atoi(VerifyStrArr[1])
		if VErr != nil {
			return false
		}
		switch {
		case VerifyStrArr[0] == "lt":
			return val.Uint() < uint64(VInt)
		case VerifyStrArr[0] == "le":
			return val.Uint() <= uint64(VInt)
		case VerifyStrArr[0] == "eq":
			return val.Uint() == uint64(VInt)
		case VerifyStrArr[0] == "ne":
			return val.Uint() != uint64(VInt)
		case VerifyStrArr[0] == "ge":
			return val.Uint() >= uint64(VInt)
		case VerifyStrArr[0] == "gt":
			return val.Uint() > uint64(VInt)
		default:
			return false
		}
	case reflect.Float32, reflect.Float64:
		VFloat, VErr := strconv.ParseFloat(VerifyStrArr[1], 64)
		if VErr != nil {
			return false
		}
		switch {
		case VerifyStrArr[0] == "lt":
			return val.Float() < VFloat
		case VerifyStrArr[0] == "le":
			return val.Float() <= VFloat
		case VerifyStrArr[0] == "eq":
			return val.Float() == VFloat
		case VerifyStrArr[0] == "ne":
			return val.Float() != VFloat
		case VerifyStrArr[0] == "ge":
			return val.Float() >= VFloat
		case VerifyStrArr[0] == "gt":
			return val.Float() > VFloat
		default:
			return false
		}
	default:
		return false
	}
}

标签：case,reflect,return,string,val,接口,golang,Validator,value
From： https://www.cnblogs.com/brandonv/p/18235120

相关文章

  • SendGrid发送邮件时如何调用API接口群发？
    SendGrid发送邮件模板如何定制？邮件发送限制有哪些？SendGrid发送邮件是一种方便快捷的方式，可以在应用程序或网站中轻松地发送大量邮件。通过调用SendGrid的API接口，您可以实现群发邮件，无论是通知用户、发送营销邮件还是其他目的，都能够高效完成。SendGrid发送邮件：调用接口通过S......
  • kube-platform平台可视化的第一个接口-namespace列表
    目录概述实践代码启动概述  此文完成kube-platform平台的第一个接口namespace列表返回。  kube-platform从平台搭建至完成第一个接口，至此基本框架就已成型，在此对几篇文章做整理。1.kube-plaform-gin框架使用2.kube-plaform-viper框架使用kube-plaform-cl......
  • Java中所有的集合可以分为两大类：接口和实现类。
     接口：Collection：是所有集合的根接口，定义了一组操作集合的基本方法，如添加、删除、遍历等。List：是有序的、可重复的集合，继承自Collection接口。Set：是无序的、不可重复的集合，继承自Collection接口。Queue：是队列接口，用于存储按一定顺序访问的元素。Deque：是双端队列接口，可以......
  • Python实现【监控远程仓库代码提交，如果有提交就自动执行需要监控测试的接口，以确保新提
    一、代码如下importgitfromdel_folderimportdel_folderimporttimefromsend_Dmessageimportsend_messagefromsend_giftimportsend_gift#设置远程仓库路径remote_url='xxx'#本地仓库路径local_path='xxx'#webhook地址和密钥webhook_url="x......
  • Qt QLineEdit设置数据的输入范围QIntValidator和QDoubleValidator
    在日常开发过程中QLineEdit作为输入框，有时要限制输入的内容，比哪，考试分数为1-100，这个时候就使用QIntValidator作为限制范围，而如何输入的是带小数的呢，那么使用QDoubleValidator可以吗，下面请看具体的示例。1.限制输入的最初使用的是正则表达式1QRegExpregIntExp("^(([1-9][0......
  • 如何用Golang写msf插件模块
    最近有空在看msf，发现msf里面有模块的源码是golang的，去翻了翻wiki，wiki上面的编写日期是2018.12.13，搜了下国内，好像没有这方面的文章，那就自己跟着做做记个笔记首先第一步自然是安装go，官方wiki上测试是在1.11.2通过，建议使用version>=1.11.2的go，怎么安装go我不再赘述注意事项......
  • 强！推荐一款开源接口自动化测试平台：AutoMeter-API ！
    在当今软件开发的快速迭代中，接口自动化测试已成为确保代码质量和服务稳定性的关键步骤。随着微服务架构和分布式系统的广泛应用，对接口自动化测试平台的需求也日益增长。今天，我将为大家推荐一款强大的开源接口自动化测试平台:AutoMeter-API。1、介绍AutoMeter-API是一款专为测......
  • 网络编程相关接口
    网络编程相关接口//创建套接字intsocket(intdomain,inttype,intprotocol);/************************************************参数 @domain 网络协议一般使用AF_INET（IPV4）* @type 传输协议* SOCK_STREAM tcp协议* SOCK_DGRAM udp协议* @protocol默认......
  • golang 可变参数用法, handlers ...HandlerFunc
     handlers...HandlerFunc这是什么写法，与group.handle()第三个参数是[]handlerFunc是什么关系呢？下面是gin中的用法：routergroup.go//GETisashortcutforrouter.Handle("GET",path,handle).func(group*RouterGroup)GET(relativePathstring,handlers...Ha......
  • eladmin-mp 低代码生成后台页面和服务端java代码接口调研
    概述总体这个项目前端使用vue2，后端使用springbootgit地址https://github.com/elunez/eladmin-mp预览地址https://eladmin.vip/demo/#/sys-tools/generator数据表配置低代码下载生成后的前端代码示例生成的后端代码示例：这里可以配置生成自动放在项目的目录......

赞助商

阅读排行

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

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

Copyright © 2020-2023 IPS99 版权所有 IPS99