golang 反射,泛型参数类型约束,泛型结构体约束,单元测试综合示例

golang中有关反射, 泛型约束和单元测试的综合案例,废话不多说直接上代码

文件名:modifydemo/modify.go

Go 复制代码
package modifydemo

import (
	"fmt"
	"reflect"
)

type User struct {
	UserId int    `json:"uid"`
	Name   string `json:"name"`
}
type Student struct {
	Name   string `json:"name"`
	UserId int    `json:"uid"`
}

// 定义一个自定义的类型接口 用来约束要修改的新值的范围,只能是这里支持的值
type MyType interface {
	int | int8 | int32 | int64 | uint | uint32 | uint64 | float32 | float64 | ~string | bool
}

// 反射修改结构体演示函数,
// 这里的第1个和第3个参数都使用泛型进行了约束,
// 即 参数obj只能是 User或者Student且必须是指针
// 参数 newVal 只能是上面的MyType接口中定义的类型之一
func ModifyStructDemo[T MyType, M ~*S, S User | Student](obj M, name string, newVal T) (err error) {
	//捕获panic异常 单元测试时不要捕获, 如果是生产环境的话就需要捕获这里可能的panic异常,避免影响后续程序的运行
	// defer func() {
	// 	if err := recover(); err != nil {
	// 		fmt.Println(err)
	// 	}
	// }()

	// 这个地方需要注意区分各个rv (reflect.Value), 因为ValueOf(), Elem(), FieldByName()他们返的对象都是 rv,
	// 但是他们各自对应的实际对象又都不同, 一定要分清楚,特别是名称不能乱用,否则就 会造成混乱!
	rv := reflect.ValueOf(obj)                       // 要修改对象的 rv
	rvElemFieldByName := rv.Elem().FieldByName(name) // 这时要修改对象上的字段的 rv

	// 如果字段name对应的rv不合法,这返回异常
	if !rvElemFieldByName.IsValid() {
		return fmt.Errorf("字段 %v 在 %T 中不存在", name, obj)
	}

	rvNewVal := reflect.ValueOf(newVal) // 这个地方是只需要获取新值对应的rv就行,不要和前面的对象关联,否则原来的rv就会被改变

	// 获取要修改的字段的reflect.Value.Kind() 类别用于switch中对不同的值调用rvf对应的方法进行修改
	newValKind := rvNewVal.Kind()
	// 根据要修改的值的类型进行匹配
	switch newValKind {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		nv := reflect.ValueOf(newVal).Int() // 这里通过反射将所有上面的int类型全部转换为int64类型
		rvElemFieldByName.SetInt(nv)        // 修改为新值
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
		nv := reflect.ValueOf(newVal).Uint() // 这里通过反射将所有上面的uint类型全部转换为uint64类型
		rvElemFieldByName.SetUint(nv)        // 修改为新值
	case reflect.Float32, reflect.Float64:
		nv := reflect.ValueOf(newVal).Float()
		rvElemFieldByName.SetFloat(nv)
	case reflect.String:
		nv := reflect.ValueOf(newVal).String()
		rvElemFieldByName.SetString(nv)
	case reflect.Bool:
		nv := reflect.ValueOf(newVal).Bool()
		rvElemFieldByName.SetBool(nv)
	default:
		err = fmt.Errorf("要设置的值类型 %T 不被支持", newVal)
	}
	return
}

单元测试 测试用例文件名:modifydemo/modify_test.go

Go 复制代码
package modifydemo

import (
	"testing"
)

func TestModifyStructDemo(t *testing.T) {
	user := &User{Name: "alex", UserId: 1}
	t.Logf("修改前的用户信息: %v", user)
	err := ModifyStructDemo(user, "Name", "p")
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("修改后的用户信息: %v", user)

	stu := &Student{Name: "小明", UserId: 18}
	t.Logf("修改前的stu信息: %v", stu)
	err2 := ModifyStructDemo(stu, "Name", "小张")
	if err2 != nil {
		t.Fatal(err2)
	}
	t.Logf("修改后的stu信息: %v", stu)
}

运行单元测试 go test -v

相关推荐
杨荧17 分钟前
【JAVA毕业设计】基于Vue和SpringBoot的服装商城系统学科竞赛管理系统
java·开发语言·vue.js·spring boot·spring cloud·java-ee·kafka
白子寰24 分钟前
【C++打怪之路Lv14】- “多态“篇
开发语言·c++
王俊山IT36 分钟前
C++学习笔记----10、模块、头文件及各种主题(一)---- 模块(5)
开发语言·c++·笔记·学习
为将者,自当识天晓地。38 分钟前
c++多线程
java·开发语言
小政爱学习!40 分钟前
封装axios、环境变量、api解耦、解决跨域、全局组件注入
开发语言·前端·javascript
k09331 小时前
sourceTree回滚版本到某次提交
开发语言·前端·javascript
神奇夜光杯1 小时前
Python酷库之旅-第三方库Pandas(202)
开发语言·人工智能·python·excel·pandas·标准库及第三方库·学习与成长
Themberfue1 小时前
Java多线程详解⑤(全程干货!!!)线程安全问题 || 锁 || synchronized
java·开发语言·线程·多线程·synchronized·
plmm烟酒僧1 小时前
Windows下QT调用MinGW编译的OpenCV
开发语言·windows·qt·opencv
测试界的酸菜鱼1 小时前
Python 大数据展示屏实例
大数据·开发语言·python