[golang]結構體(slice)如何轉map(slice)

需求:有一個切片,其元素是不固定類型的結構體,如何轉換爲元素爲map類型的切片。

以下例子是通過反射reflect的方法來完成這個轉換過程。

package main

import (
    "fmt"
    "reflect"
)

type Student struct {
    Name string `json:"name"`
    Age  uint   `json:"age"`
}

type Teacher struct {
    Name string `json:"name"`
    Gender  uint   `json:"gender"`
}

// StructSliceToMapSlice : struct切片轉爲map切片
func StructSliceToMapSlice(source interface{}) {
    // 判斷,interface轉爲[]interface{}
    v := reflect.ValueOf(source)
    if v.Kind() != reflect.Slice {
        panic("ERROR: Unknown type, slice expected.")
    }
    l := v.Len()
    ret := make([]interface{}, l)
    for i := 0; i < l; i++ {
        ret[i] = v.Index(i).Interface()
    }

    // 轉換之後的結果變量
    res := make([]map[string]interface{}, 0)

    // 通過遍歷,每次迭代將struct轉爲map
    for _, elem := range ret {
        data := make(map[string]interface{})
        objT := reflect.TypeOf(elem)
        objV := reflect.ValueOf(elem)
        for i := 0; i < objT.NumField(); i++ {
            data[objT.Field(i).Name] = objV.Field(i).Interface()
        }
        res = append(res, data)
    }

    fmt.Printf("==== Convert Result ====\n%+v\n", res)
}

func main() {
    var people []interface{}

    // 切片添加Student元素
    stu1 := Student{
        Name: "Ame",
        Age:  18,
    }
    stu2 := Student{
        Name: "Maybe",
        Age:  20,
    }
    people = append(people, stu1)
    people = append(people, stu2)

    // 切片添加Teacher元素
    tech := Teacher{
        Name: "Rotk",
        Gender:  0,
    }
    people = append(people, tech)

    StructSliceToMapSlice(people)
}

示例結果:

==== Convert Result ====
[map[Name:Ame Age:10] map[Age:20 Name:Maybe] map[Name:Rotk Gender:0]]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章