前文
在开发过程中,一直需要将切片数组等转成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
}