Go中查找并删除嵌套的json对象

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《Go中查找并删除嵌套的json对象》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

我有一个 Kubernetes Pod 的 json 文档，这是一个示例： https://github.com/itaysk/kubectl-neat/blob/master/test/fixtures/pod-1-raw.json

我想遍历 spec.containers[i].volumeMounts 并删除 .name 以 "default-token-" 开头的那些volumeMount 对象。请注意，containers 和 volumeMounts 都是数组。

使用jq，我花了1分钟写了这1行：try del(.spec.containers[].volumeMounts[] | select(.name |startswith("default-token-")))。我正在尝试用 Go 重写它。

在寻找一个好的 json 库时，我选择了 gjson/sjson。 由于 sjson 不支持数组访问器（# 语法），并且 gjson 不支持获取结果路径，因此我寻找解决方法。

我尝试使用 Result.Index 直接从字节切片中删除结果，并成功，但对于我编写的查询（spec.containers.#.volumeMounts.#(name%\"default-token-*\" )|0) 索引始终为 0（我尝试了它的不同变体，结果相同）。

所以目前我有一些代码 25 行代码，使用 gjson 获取 spec.containers.#.volumeMounts 并迭代它的结构，最终使用 sjson.Delete 进行删除。 它有效，但感觉比我预期的要复杂得多。

在 Go 中是否有更好的方法来做到这一点？如果需要，我愿意切换 json 库。

编辑：我宁愿避免使用类型化架构，因为我可能需要在不同类型上执行此操作，对于某些类型我没有完整的架构。

（还删除了有关我当前实现的一些分散注意力的细节）

解决方案

这里最简单的事情是将 json 解析为一个对象，使用该对象，然后序列化回 json。

kubernetes 提供了一个 go 客户端库，它定义了 v1.Pod struct，您可以使用 stdlib 编码/json 解组：

// import "k8s.io/api/core/v1"
var pod v1.pod
if err := json.unmarshal(podbody, &pod); err != nil {
    log.fatalf("parsing pod json: %s", err)
}

从那里您可以阅读 pod.spec.containers 及其 volumemounts：

// modify.
for c := range pod.spec.containers {
    container := &pod.spec.containers[c]
    for i, vol := range container.volumemounts {
        if strings.hasprefix(vol.name, "default-token-") {
            // remove the volumemount at index i.
            container.volumemounts = append(container.volumemounts[:i], container.volumemounts[i+1:]...)
        }
    }
}

https://play.golang.org/p/3r5-XKIazhK

如果您担心丢失输入中可能出现的一些任意 json，您可能希望定义 var pod map[string]interface{}，然后将其中的每个属性类型转换为 spec，好的:= pod["spec"].(map[string]interface{})、containers, ok := spec["containers"].([]map[string]interface) 等等。

希望有帮助。

ps。 “删除”是在 https://github.com/golang/go/wiki/SliceTricks#delete 之后

要采取与以前完全不同的方法，您可以创建一个

type root struct {
    fields struct {
        spec *spec `json:"spec,omitempty"`
    }
    other map[string]interface{}
}

使用自定义 unmarshaljson 解组到字段和其他字段，以及自定义 marshaljson 在返回 json.marshal(other) 之前设置 other["spec"] = json.rawmessage(spec.marshaljson())：

func (v *root) unmarshaljson(b []byte) error {
    if err := json.unmarshal(b, &v.fields); err != nil {
        return err
    }
    if v.other == nil {
        v.other = make(map[string]interface{})
    }
    if err := json.unmarshal(b, &v.other); err != nil {
        return err
    }
    return nil
}

func (v *root) marshaljson() ([]byte, error) {
    var err error
    if v.other["spec"], err = rawmarshal(v.fields.spec); err != nil {
        return nil, err
    }
    return json.marshal(v.other)
}

func rawmarshal(v interface{}) (json.rawmessage, error) {
    b, err := json.marshal(v)
    if err != nil {
        return nil, err
    }
    return json.rawmessage(b), nil
}

然后，您可以通过 .spec.containers.volumemounts 一直定义这些类型，并拥有一个 container.marshaljson ，它会丢弃我们不喜欢的 volumemounts：

func (v *Container) MarshalJSON() ([]byte, error) {
    mounts := v.fields.VolumeMounts
    for i, mount := range mounts {
        if strings.HasPrefix(mount.fields.Name, "default-token-") {
            mounts = append(mounts[:i], mounts[i+1:]...)
        }
    }

    var err error
    if v.other["volumeMounts"], err = rawMarshal(mounts); err != nil {
        return nil, err
    }
    return json.Marshal(v.other)
}

完整演示示例：https://play.golang.org/p/k1603cchwC7

我不会这样做。

到这里，我们也就讲完了《Go中查找并删除嵌套的json对象》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！