go的反射和断言

在go中对于一个变量,主要包含两个信息变量类型(type)和变量值(value)

可以通过reflect包在运行的时候动态获取变量信息,并能够进行操作

对于Type可以通过reflect.TypeOf()获取到变量的类型信息

reflect.TypeOf()返回的是一个Type类型,里面保存了类型的信息

通过reflect.ValueOf()来获取到变量的值部分

reflect.ValueOf()返回的是一个Value类型

可以通过Value.Kind()获取到该变量大概的类型

go中的断言可以将一个未知的类型转换为指定的类型

通过结合反射和断言,我们可以实现这样的一个功能

复制代码
// UpdateFieldMap 获取更新字段键值对,除了在empty记录的字段整数类型为0或者string类型为空默认不更新
func UpdateFieldMap(in any, empty map[string]struct{}) map[string]any {
	out := make(map[string]any)
	if empty == nil {
		empty = make(map[string]struct{})
	}

	v := reflect.ValueOf(in)
	if v.Kind() == reflect.Ptr {
		v = v.Elem()
	}
	if v.Kind() != reflect.Struct {
		panic(fmt.Errorf("ToMap only accepts structs; got %T", v))
	}

	typ := v.Type()
	for i := 0; i < v.NumField(); i++ {

		// 获取字段的值
		fieldValue := v.Field(i)
		// 获取字段的类型
		//fieldType := fieldValue.Type()

		columns := strings.Split(typ.Field(i).Tag.Get("gorm"), ";")
		column := ""
		for _, s := range columns {
			if strings.HasPrefix(s, "column:") {
				column = s[7:]
			}
		}
		switch fieldValue.Kind() {
		case reflect.Int:
			//fmt.Println(column, fieldValue.Interface().(int))
			val := fieldValue.Interface().(int)

			_, ok := empty[column]
			if val != 0 || ok {
				out[column] = val
			}
		case reflect.String:
			val := fieldValue.Interface().(string)

			_, ok := empty[column]
			if val != "" || ok {
				out[column] = val
			}
		case reflect.Struct:
			if fieldValue.Type() == reflect.TypeOf(time.Time{}) {
				// 处理 time.Time 类型
				val := fieldValue.Interface().(time.Time)
				_, ok := empty[column]
				if !val.IsZero() || ok {
					out[column] = val
				}
			}
		}
	}

	return out

}
相关推荐
明月醉窗台6 分钟前
qt使用笔记六之 Qt Creator、Qt Widgets、Qt Quick 详细解析
开发语言·笔记·qt
逍遥德6 分钟前
如何学编程之01.理论篇.如何通过阅读代码来提高自己的编程能力?
前端·后端·程序人生·重构·软件构建·代码规范
wangjialelele9 分钟前
平衡二叉搜索树:AVL树和红黑树
java·c语言·开发语言·数据结构·c++·算法·深度优先
lili-felicity17 分钟前
CANN性能调优与实战问题排查:从基础优化到排障工具落地
开发语言·人工智能
独自破碎E19 分钟前
【BISHI15】小红的夹吃棋
android·java·开发语言
进阶小白猿31 分钟前
Java技术八股学习Day33
java·开发语言·学习
MX_935942 分钟前
Spring的bean工厂后处理器和Bean后处理器
java·后端·spring
执风挽^1 小时前
Python基础编程题2
开发语言·python·算法·visual studio code
程序员泠零澪回家种桔子2 小时前
Spring AI框架全方位详解
java·人工智能·后端·spring·ai·架构
Z9fish2 小时前
sse哈工大C语言编程练习20
c语言·开发语言·算法