前文

在开发过程中,一直需要将切片数组等转成map类型。由于golang是强类型。转换特别麻烦。正常情况是for 处理list,生成map[string]struct.每一种struct需要写下面对应的方法处理;

func Tomap(list []User) map[string]User{
	data := make(map[string]User)
	for _, row := range list {
		data[row .Name] = row 
	}
	return data
}

是否有通用的list 转map呢? 官方是没有的,需要自己写一个,使用reflect包 实现的

package test

import (
	"github.com/xiaocainiaoMX/tool/slice"
)

func main() {
	type User struct {
		Uid  int    `json:"uid"`
		Name string `json:"name"`
		Code string `json:"code"`
	}
	var Data []User
	Data = append(Data, User{Uid: 1, Name: "test1", Code: "A"})
	Data = append(Data, User{Uid: 2, Name: "test2", Code: "B"})
	res := tool.ListToMap(Data, "Name")
	fmt.Println(res)
}

包源码 https://github.com/xiaocainiaoMX/tool.git

package slice

import "reflect"

// interface{} change to map[string]interface{}
// interface{} data is []interface{}
func ListToMap(list interface{}, key string) map[string]interface{} {
	res := make(map[string]interface{})
	arr := ToSlice(list)
	for _, row := range arr {
		immutable := reflect.ValueOf(row)
		val := immutable.FieldByName(key).String()
		res[val] = row
	}
	return res
}

//interface{} change to []interface{}
func ToSlice(arr interface{}) []interface{} {
	ret := make([]interface{}, 0)
	v := reflect.ValueOf(arr)
	if v.Kind() != reflect.Slice {
		ret = append(ret, arr)
		return ret
	}
	l := v.Len()
	for i := 0; i < l; i++ {
		ret = append(ret, v.Index(i).Interface())
	}
	return ret
}