快捷導(dǎo)航

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

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

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

前言

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

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

使用json操作的這個(gè)方式是比較簡單的，容易想到也易實(shí)現(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))
}

輸出結(jié)果：

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

JSON-duration:174000

方式2：使用反射

通過反射機(jī)制，靈活的做類型轉(zhuǎn)換。具體實(shí)現(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))
}

輸出結(jié)果：

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

反射-duration:60000

兩種方式對比

執(zhí)行效率：

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

輸出結(jié)果：

使用json能達(dá)到預(yù)期，正常解析出結(jié)構(gòu)體tag；

使用反射未能達(dá)到預(yù)期，未解析出結(jié)構(gòu)體tag，字段是以結(jié)構(gòu)體定義為準(zhǔn)

封裝到工具包

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

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

package utils

import (
    "reflect"
    "strings"
)

type IStruct interface {
    GetStructData() interface{}
}

//struct轉(zhuǎn)map
//使用反射實(shí)現(xiàn)，完美地兼容了json標(biāo)簽的處理
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"`
}

//注意：必須實(shí)現(xiàn)這個(gè)方法，才能正確調(diào)用工具包轉(zhuǎ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))
}

輸出結(jié)果：

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

反射2-duration:65000

結(jié)論：

執(zhí)行效率高的同時(shí)，輸出結(jié)果也符合預(yù)期，能正確的解析出結(jié)構(gòu)體tag

到此這篇關(guān)于Golang實(shí)現(xiàn)優(yōu)雅的將struct轉(zhuǎn)換為map的文章就介紹到這了,更多相關(guān)Golang struct轉(zhuǎn)map內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: