go使用反射获取http.Request参数到结构体

1. 为什么写这个工具

公司代码使用go原生写的,没有使用框架,接收前端的代码都使用的是req.FormValue()来接收,我觉得写了太多的接收参数的重复代码

2. Go代码
go 复制代码
// ParseParams 解析前端传的参数到结构体
func ParseParams[T any](req *http.Request) (T, error) {
	var t T
	tType := reflect.TypeOf(t)
	tValue := reflect.ValueOf(&t).Elem()

	if tType.Kind() != reflect.Struct {
		return t, fmt.Errorf("parseParams err: tType.Kind() != reflect.Struct")
	}

	for i := 0; i < tType.NumField(); i++ {
		field := tType.Field(i)
		value := tValue.Field(i)
		err := setFieldValue(field, value, req)
		if err != nil {
			return t, err
		}
	}

	return t, nil
}

// 设置对应的字段值
// 仅支持string, int, int64, int32, slice
func setFieldValue(field reflect.StructField, value reflect.Value, req *http.Request) error {
	formValue := req.FormValue(field.Tag.Get("json"))
	switch field.Type.Kind() {
	case reflect.String:
		value.SetString(formValue)
	case reflect.Int, reflect.Int64, reflect.Int32:
		atoi, err := strconv.Atoi(formValue)
		if err != nil {
			return fmt.Errorf("setFieldValue err: strconv.Atoi(%s) err: %v, field.Name: %v", formValue, err, field.Name)
		}
		value.SetInt(int64(atoi))
	case reflect.Slice:
		newSlice := reflect.New(field.Type)
		if err := json.Unmarshal([]byte(formValue), newSlice.Interface()); err != nil {
			return fmt.Errorf("setFieldValue err: json.Unmarshal(%s) err: %v, field.Name: %v", formValue, err, field.Name)
		}
		value.Set(newSlice.Elem())
	default:
		return fmt.Errorf("setFieldValue err: field.Type.Kind() not support: %v, field.Name: %v", field.Type.Kind(), field.Name)
	}

	return nil
}
3.使用
go 复制代码
type TestReq struct {
	ID           int32   `json:"id"`
	Name         string  `json:"name"`
	SiteIdList   []int32 `json:"siteIdList"`
	DateTimeList []int64 `json:"dateTimeList"`
}

func TestParseParams(t *testing.T) {
	req := &http.Request{}
	req.Form = map[string][]string{
		"id":           {"1"},
		"name":         {"test"},
		"siteIdList":   {"[1,2,3]"},
		"dateTimeList": {"[1694560000, 1694560060]"},
	}
	params, err := ParseParams[TestReq](req)
	if err != nil {
		fmt.Println(err)
		return
	}

	fmt.Println(params)
}
目前只做了我常用的一些变量,后续可能会加入其他变量和字段校验
相关推荐
天若有情6731 天前
程序员原创|借鉴JS事件冒泡,根治电脑文件混乱的“冒泡整理法”
开发语言·javascript·windows·ecmascript·电脑·办公·日常
qq_589568101 天前
springbootweb案例,出现访问 http://localhost:8080/list 一直处于浏览器运转阶段
java·网络协议·http·list·springboot
特种加菲猫1 天前
继承,一场跨越时空的对话
开发语言·c++
玩转单片机与嵌入式1 天前
玩转边缘AI(TInyML):需要掌握的C++知识汇总!
开发语言·c++·人工智能
茉莉玫瑰花茶1 天前
Qt 信号与槽 [ 1 ]
开发语言·数据库·qt
AI人工智能+电脑小能手1 天前
【大白话说Java面试题】【Java基础篇】第30题:JDK动态代理和CGLIB动态代理有什么区别
java·开发语言·后端·面试·代理模式
张健11564096481 天前
临界区和同一线程上锁
java·开发语言·jvm
头发够用的程序员1 天前
C++和Python面试经典算法汇总(一)
开发语言·c++·python·算法·容器·面试
夜猫逐梦1 天前
【逆向经验】一篇文章讲透为什么CE搜不到Python游戏的内存值
开发语言·python·游戏