在 Golang 中向嵌套的 map[string]interface{} 添加属性

Question

我正在处理map[string]interface{}类型的数据。 它可以在 (map[string]interface{}) 类型中包含无限数量的嵌套对象。

编辑：此数据来自 mongodb。 我不能在这里真正应用 golang 的结构，因为属性因文档而异。 我要做的就是获取嵌套最深的 object，为其添加一个新属性，并确保之后更新整个数据 object。

data["person"] = map[string]interface{}{
    "peter": map[string]interface{}{
        "scores": map[string]interface{}{
            "calculus": 88,
            "algebra":  99,
            "golang":   89,
        },
    },
}

这些数据来自远程 API，我不知道里面的属性。 我要添加的只是在最后一个 object 中添加新属性（在本例中为“分数”），让我们说使用这个新属性（“物理”），数据看起来像这样

data["person"] = map[string]interface{}{
    "peter": map[string]interface{}{
        "scores": map[string]interface{}{
            "calculus": 88,
            "algebra":  99,
            "golang":   89,
            "physics":  95,
        },
    },
}

我不确定如何将该属性添加到最后一个 object。

我进行了递归类型检查，并且能够获取每个字段并打印其值。 但是因为地图不是参考的，所以当我到达 map 的值不是复杂类型时，我无法向原始 map 添加值。

package main

import "fmt"

func main() {

    data := make(map[string]interface{})
    data["person"] = map[string]interface{}{
        "peter": map[string]interface{}{
            "scores": map[string]interface{}{
                "calculus": 88,
                "algebra":  99,
                "golang":   89,
            },
        },
    }

    parseMap(data)
}


func parseMap(aMap map[string]interface{}) interface{} {
    var retVal interface{}

    for _, val := range aMap {
        switch val.(type) {
        case map[string]interface{}:
            retVal = parseMap(val.(map[string]interface{}))
        //case []interface{}:
        //  retVal = parseArray(val.([]interface{}))
        default:
            //here i would have done aMap["physics"] = 95 if I could access the original map by reference, but that is not possible

            retVal = aMap

        }
    }

    return retVal
}

Answer 1

尽量避免使用原始map[string]interface{}类型。 Go encoding/json文件可以很好地处理字符串键控映射，希望远程 API 对您正在处理的内容有某种规范。 （例如，您知道您期望person顶级键和在层次结构中特定点的scores 。）

我假设远程 API 是 JSON-over-HTTP。 您可能 model 其结构为

type Input struct {
    Person map[string]Person `json:"person"`
}

type Person struct {
    Scores map[string]int `json:"scores"`
}

将json.Unmarshal()数据编入此结构后，您可以直接设置

data.Person["peter"].Scores["physics"] = 95

然后json.Marshal()再次得到结果。 https://play.golang.org/p/qoAVFodSvK2有一个完整的例子。

如果您真的想直接操作map[string]interface{}结构，我建议将每个“级别”拆分为单独的 function 调用

func ParseTopLevel(data map[string]interface{}) {
    switch peter := data["peter"].(type) {
    case map[string]interface{}:
        ParsePeter(peter)
    }
}

map类型是通过引用传递的，因此当您到达堆栈底部时，您可以直接设置scores["physics"] = 95 。 （在您的原始代码中，如果您不能按照您的建议直接设置aMap["physics"] ，我会感到惊讶，尽管设置的内容相当不精确；比较https://play.golang.org/p/ VuTjcjezwwU .)

Answer 2

根据对问题的评论，目标是在嵌套最深的 map 中设置一个值。

使用以下 function 在最大嵌套级别找到 map。 如果在最大嵌套级别上存在多个 map，则此 function 将返回这些映射中的任意一个。

func findDeepest(m map[string]interface{}) (int, map[string]interface{}) {
    depth := 0
    candidate := m
    for _, v := range m {
        if v, ok := v.(map[string]interface{}); ok {
            d, c := findDeepest(v)
            if d+1 > depth {
                depth = d + 1
                candidate = c
            }
        }
    }
    return depth, candidate
}

像这样使用它在深度嵌套的 map 中设置一个值：

_, m := findDeepest(data)
m["physics"] = 95

在操场上运行它。