当前位置：首页 > 文章列表 > Golang > Go问答 > 使用 Go 获取集合中所有键的名称

使用 Go 获取集合中所有键的名称

来源：stackoverflow 2024-04-14 15:15:38 0浏览收藏

哈喽！今天心血来潮给大家带来了《使用 Go 获取集合中所有键的名称》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

问题内容

我想获取 mongodb 集合中所有键的名称。

例如，从此：

"id": objectid("5f5a010d431c4519dcda0e3d")
            "title": "app"
            "query": ""
            "db": ""
            "widgettype": ""
            "tablename": "active_instance"
            fields:object
                user:"name",
                key:"passcode"
            "status": "active"
            "inlibrary": ""
            "createdts": 1599733804

使用“gopkg.in/mgo.v2”和“gopkg.in/mgo.v2/bson”包。

err := mongodbsession.db(dbname).c(collectionname).find(bson.m{}).one(&result)
var keyset []string
    for index, _ := range result {
        fmt.printf("%+v\n", index)
        keyset = append(keyset, index)
    }

    fmt.println(keyset)

获取输出如下

[_id title query db widgettype  status fields inlibrary createdts ]

未获取用户和密钥的子密钥。

解决方案

嵌入文档将在 result 中显示为另一个 bson.m 值，因此您必须使用递归来遍历这些值。

具体操作方法如下：

func getkeys(m bson.m) (keys []string) {
    for k, v := range m {
        keys = append(keys, k)
        if m2, ok := v.(bson.m); ok {
            keys = append(keys, getkeys(m2)...)
        }
    }
    return
}

使用示例：

m := bson.m{"id": bson.objectid("5f5a010d431c4519dcda0e3d"),
    "title":      "app",
    "query":      "",
    "db":         "",
    "widgettype": "",
    "tablename":  "active_instance",
    "fields": bson.m{
        "user": "name",
        "key":  "passcode",
    },
    "status":    "active",
    "inlibrary": "",
    "createdts": 1599733804,
}

keys := getkeys(m)
fmt.println(keys)

将输出（在 Go Playground 上尝试）：

[db widgettype createdts inlibrary _id title query tablename
  fields user key status]

如果查看结果，会发现包含 user 和 key，但无法判断它们是文档的字段还是嵌入文档的字段。

您可以选择使用嵌入文档字段本身的字段名称作为嵌入文档字段的前缀，例如获取 fields.user 和 fields.key。

您可以这样做：

func getkeys(m bson.m) (keys []string) {
    for k, v := range m {
        keys = append(keys, k)
        if m2, ok := v.(bson.m); ok {
            for _, k2 := range getkeys(m2) {
                keys = append(keys, k+"."+k2)
            }
        }
    }
    return
}

哪个会输出（在 Go Playground 上尝试一下）：

[createdts title query db status inlibrary _id widgettype tablename
    fields fields.user fields.key]

另请注意，上述解决方案不处理数组。如果您有数组，您还应该迭代它们，如果它们包含另一个数组或对象，您应该执行相同的操作（递归）。将其扩展为处理数组也是一个练习。

包“gopkg.in/mgo.v2”和“gopkg.in/mgo.v2/bson”已过时。替换为“go.mongodb.org/mongo-driver/bson”和“go.mongodb.org/mongo-driver/mongo”。必须使用递归来检索嵌套字段。假设数据库“store”具有此文档的集合“users”

{"fields": {
  "user": "alex",
  "nickname": "ferguson",
}}

从集合中检索密钥的代码是：

package main

import (
    "context"
    "fmt"
    "log"
    "time"

    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

func main() {
    ctx, cancel := context.withtimeout(context.background(), 20*time.second)
    defer cancel()
    client, err := mongo.connect(ctx, options.client().applyuri("mongodb://localhost:27017"))

    if err != nil {
        log.panicln(err)
    }
    defer client.disconnect(ctx)
    var docraw bson.raw
    res := client.database("store").collection("users").findone(context.background(), bson.d{})

    err = res.decode(&docraw)
    if err != nil {
        log.panicln(err)

    }
    elements, err := docraw.elements()
    if err != nil {
        log.panicln(err)
    }
    var keys []string
    for _, e := range elements {
        keys = append(keys, getkey(e)...)
    }
    fmt.println(keys)
}

func getkey(e bson.rawelement) []string {
    if e.value().type != bson.typeembeddeddocument {
        return []string{e.key()}
    }
    var result []string
    nested, _ := e.value().document().elements()
    for _, n := range nested {
        nkeys := getkey(n)
        for _, k := range nkeys {
            result = append(result, fmt.sprintf("%s.%s", e.key(), k))
        }
    }
    return result
}

输出：