Featured image of post golang 中使用反射实现 struct 数组根据某个 key 属性生成 map 数据

golang 中使用反射实现 struct 数组根据某个 key 属性生成 map 数据

主要点

golang reflect

方法

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
ValueOf
// ValueOf返回一个初始化为i接口保管的具体值的Value,ValueOf(nil)返回Value零值。

Type
// 返回v持有的值的类型的Type表示。

Convert
// Convert将v持有的值转换为类型为t的值,并返回该值的Value封装。如果go转换规则不支持这种转换,会panic。

FieldByName
// 返回该类型名为name的字段(的Value封装)(会查找匿名字段及其子字段),如果v的Kind不是Struct会panic;如果未找到会返回Value零值。

Interface
// 本方法返回v当前持有的值(表示为/保管在interface{}类型)

实现代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
func ArrayStructToMap(val interface{}, key string) (res map[string]interface{}, err error) {
	v := reflect.ValueOf(val)
	if v.Kind() != reflect.Slice {
		return nil, fmt.Errorf("argument must be a slice")
	}

	res = make(map[string]interface{})
	for i := 0; i < v.Len(); i++ {
		vIndex := v.Index(i)
		vType := vIndex.Type()

		st := vIndex.Convert(vType)

		// 获取列名为 key  value
		keyValue := st.FieldByName(key)
		if keyValue.IsValid() {
			var _key = strconv.FormatUint(keyValue.Uint(), 10)
			res[_key] = vIndex.Interface()
		} else {
			return nil, fmt.Errorf("未找到列名数据: %s", key)
		}
	}

	return res, nil
}