Go语言中利用反射实现Map键类型通用转换以支持JSON序列化

package main

import (
    "fmt"
    "encoding/json"
)

type ClassA struct {
    Id   int    `json:"id"`
    Name string `json:"name"`
}

// 为ClassA编写的转换函数
func TransformMapClassA(mapOfIntToClassA map[int]*ClassA) map[string]*ClassA {
    mapOfStringToClassA := make(map[string]*ClassA)
    for id, obj := range mapOfIntToClassA {
        mapOfStringToClassA[fmt.Sprintf("%d", id)] = obj // 使用id作为键
    }
    return mapOfStringToClassA
}

func main() {
    data := map[int]*ClassA{
        1: {Id: 101, Name: "Item A"},
        2: {Id: 102, Name: "Item B"},
    }

    // 转换并序列化
    transformedData := TransformMapClassA(data)
    jsonData, err := json.MarshalIndent(transformedData, "", "  ")
    if err != nil {
        fmt.Println("Error marshalling JSON:", err)
        return
    }
    fmt.Println(string(jsonData))
    // 预期输出:
    // {
    //   "1": {
    //     "id": 101,
    //     "name": "Item A"
    //   },
    //   "2": {
    //     "id": 102,
    //     "name": "Item B"
    //   }
    // }
}

package main

import (
    "encoding/json"
    "errors"
    "fmt"
    "reflect"
)

// TransformMap 是一个通用函数，用于将任意Map的键转换为字符串，
// 并返回 map[string]interface{} 类型。
// 它适用于键类型可以被 fmt.Sprint 转换为字符串的Map。
func TransformMap(m interface{}) (map[string]interface{}, error) {
    // 获取m的反射值
    v := reflect.ValueOf(m)

    // 检查m是否为Map类型
    if v.Kind() != reflect.Map {
        return nil, errors.New("输入参数必须是Map类型")
    }

    // 初始化结果Map，预分配容量以优化性能
    result := make(map[string]interface{}, v.Len())

    // 获取Map的所有键
    keys := v.MapKeys()
    for _, k := range keys {
        // 将键转换为其接口值，然后使用 fmt.Sprint 转换为字符串
        stringKey := fmt.Sprint(k.Interface())
        // 获取对应键的值，并将其转换为接口值
        value := v.MapIndex(k).Interface()
        result[stringKey] = value
    }

    return result, nil
}

// 示例结构体
type Product struct {
    SKU  string  `json:"sku"`
    Name string  `json:"name"`
    Price float64 `json:"price"`
}

func main() {
    // 示例数据：map[int]*Product
    productsById := map[int]*Product{
        101: {SKU: "P001", Name: "Laptop", Price: 1200.00},
        102: {SKU: "P002", Name: "Mouse", Price: 25.50},
    }

    // 示例数据：map[uint]string
    statusMap := map[uint]string{
        1: "Active",
        2: "Pending",
        3: "Inactive",
    }

    // 使用TransformMap转换 productsById
    transformedProducts, err := TransformMap(productsById)
    if err != nil {
        fmt.Println("Error transforming products map:", err)
        return
    }

    // 将转换后的Map序列化为JSON
    jsonProducts, err := json.MarshalIndent(transformedProducts, "", "  ")
    if err != nil {
        fmt.Println("Error marshalling products JSON:", err)
        return
    }
    fmt.Println("Transformed Products JSON:")
    fmt.Println(string(jsonProducts))
    fmt.Println("--------------------")

    // 使用TransformMap转换 statusMap
    transformedStatus, err := TransformMap(statusMap)
    if err != nil {
        fmt.Println("Error transforming status map:", err)
        return
    }

    // 将转换后的Map序列化为JSON
    jsonStatus, err := json.MarshalIndent(transformedStatus, "", "  ")
    if err != nil {
        fmt.Println("Error marshalling status JSON:", err)
        return
    }
    fmt.Println("Transformed Status JSON:")
    fmt.Println(string(jsonStatus))
}