Golang實現(xiàn)優(yōu)雅的將struct轉換為map

更新時間：2023年01月29日 10:57:53 作者：勁仔Go

在項目實踐中，有時候我們需要將struct結構體轉為map映射表，然后基于map做數(shù)據(jù)裁剪或操作。那么下面我來介紹下常用的兩種轉換方式，希望對大家有所幫助

前言

在項目實踐中，有時候我們需要將struct結構體轉為map映射表，然后基于map做數(shù)據(jù)裁剪或操作。那么下面我來介紹下常用的兩種轉換方式，以及對它們做對比，最后采用更優(yōu)雅的方式，封裝到我們的項目工程的工具包里

方式1：使用JSON序列和反序列化

使用json操作的這個方式是比較簡單的，容易想到也易實現(xiàn)。直接上代碼：

package main

import (
    "encoding/json"
    "fmt"
    "time"
)

type Person struct {
    Name    string `json:"name"`
    Address string `json:"address"`
}

func main() {
    t := time.Now().UnixNano()
    m := make(map[string]interface{})
    person := Person{
       Name:    "zhangsan",
       Address: "北京海淀",
    }
    j, _ := json.Marshal(person)
    json.Unmarshal(j, &m)
    fmt.Println(m)
    fmt.Println(fmt.Sprintf("JSON-duration:%d", time.Now().UnixNano() - t))
}

輸出結果：

map[address:北京海淀 name:zhangsan]

JSON-duration:174000

方式2：使用反射

通過反射機制，靈活的做類型轉換。具體實現(xiàn)：

package main

import (
    "fmt"
    "reflect"
    "time"
)

type Person struct {
    Name    string `json:"name"`
    Address string `json:"address"`
}

func main() {
    t := time.Now().UnixNano()
    m := make(map[string]interface{})
    person := Person{
        Name:    "zhangsan",
        Address: "北京海淀",
    }

    elem := reflect.ValueOf(&person).Elem()
    relType := elem.Type()
    for i := 0; i < relType.NumField(); i++ {
        name := relType.Field(i).Name
        m[name] = elem.Field(i).Interface()
    }

    fmt.Println(m)
    fmt.Println(fmt.Sprintf("反射-duration:%d", time.Now().UnixNano() - t))
}

輸出結果：

map[Address:北京海淀 Name:zhangsan]

反射-duration:60000

兩種方式對比

執(zhí)行效率：

使用反射的效率，明顯比使用json的效率要高，接近3倍

輸出結果：

使用json能達到預期，正常解析出結構體tag；

使用反射未能達到預期，未解析出結構體tag，字段是以結構體定義為準

封裝到工具包

基于上面兩種方式的對比，我們決定采用反射機制，并對結構體tag解析做兼容，優(yōu)雅的將struct轉換為map，并封裝到工具包中

具體實現(xiàn)，工具包代碼：

package utils

import (
    "reflect"
    "strings"
)

type IStruct interface {
    GetStructData() interface{}
}

//struct轉map
//使用反射實現(xiàn)，完美地兼容了json標簽的處理
func StructToMap(st IStruct) map[string]interface{} {
    m := make(map[string]interface{})
    in := st.GetStructData()
    val := reflect.ValueOf(in)
    if val.Kind() == reflect.Ptr {
        val = val.Elem()
    }
    if val.Kind() != reflect.Struct {
        return m
    }

    relType := val.Type()
    for i := 0; i < relType.NumField(); i++ {
        name := relType.Field(i).Name
        tag := relType.Field(i).Tag.Get("json")
        if tag != "" {
            index := strings.Index(tag, ",")
            if index == -1 {
                name = tag
            } else {
                name = tag[:index]
            }
        }
        m[name] = val.Field(i).Interface()
    }
    return m
}

測試代碼：

package main

import (
    "fmt"
    "learn-go/utils"
    "time"
)

type Person struct {
    Name    string `json:"name"`
    Address string `json:"address"`
}

//注意：必須實現(xiàn)這個方法，才能正確調用工具包轉換方法
func (p Person) GetStructData() interface{} {
    return p
}

func main() {
    t := time.Now().UnixNano()
    m := make(map[string]interface{})
    person := Person{
        Name:    "zhangsan",
        Address: "北京海淀",
    }

    m = utils.StructToMap(person)
    fmt.Println(m)
    fmt.Println(fmt.Sprintf("反射2-duration:%d", time.Now().UnixNano() - t))
}

輸出結果：