使用mongoDB聚合获取特定字段的列表

在 MongoDB 中，可以使用聚合管道来获取特定字段的列表。通过使用 `$match`、`$group` 和 `$project` 等聚合阶段，可以将具有特定 `uid` 值的文档分组，并从分组结果中提取相应的 `uid` 值。最终，通过使用 `$filter` 和 `$concatArrays`，可以获取一个不包含匹配 `uid` 值的唯一 `uid` 列表。

在 golang 上使用 mongo-driver。

为了保存两个实体之间的关系列表，我使用一个集合，其中每个文档都有一个“uid1”和一个“uid2”，它们是每个实体的唯一标识符。任何实体都可以是双方的。目标是找到与我选择的实体有关系的所有实体并将其放入列表中。目前正在做：

cursor, err := relationscoll.find(ctx, bson.m{"$or": []bson.m{bson.m{"uid1": uid}, bson.m{"uid2": uid}}})

    var res []bson.m
    if err = cursor.all(ctx, &res); err != nil {
        return nil, err
    }

我在变量 res 中获取了存在 uid 变量的条目列表。我想要得到的只是一个包含相应条目中 uid 的列表。因此，不要以以下形式获取某些内容：（假设我们请求与“xxxx”有关系的实体）

[[uid1:xxxx uid2:yyyy],[uid1:zzzz uid2:xxxx]]

我可以得到以下形式的东西：

[yyyy, zzzz]

这可以通过聚合来实现吗？

希望我说得清楚，如果有任何未明确解释的内容请在下面评论。

解决方案

下面的代码应该可以满足您的需要

UID := "id-1"  // Change this variable to your `match` value

matchStage := bson.D{
    {"$match", bson.D{
        {"$or", bson.A{
            bson.D{{"uid1", UID}},
            bson.D{{"uid2", UID}},
        }},
    }},
}
groupStage := bson.D{
    {"$group", bson.D{
        {"_id", nil},
        {"uids1", bson.D{
            {"$addToSet", "$uid1"},
        }},
        {"uids2", bson.D{
            {"$addToSet", "$uid2"},
        }},
    }},
}
projectStage := bson.D{
    {"$project", bson.D{
        {"uids", bson.D{
            {"$filter", bson.D{
                {"input", bson.D{
                    {"$concatArrays", bson.A{"$uids1", "$uids2"}},
                }},
                {"as", "arrayElem"},
                {"cond", bson.D{
                    {"$ne", bson.A{"$$arrayElem", UID}},
                }},
            }},
        }},
    }},
}
cursor, err := collection.Aggregate(
    ctx,
    mongo.Pipeline{matchStage, groupStage, projectStage},
    options.Aggregate().SetAllowDiskUse(true),
    )
if err != nil {
    panic(err)
}

var cursorResult []bson.M
err = cursor.All(ctx, &cursorResult)
if err != nil {
    panic(err)
}

fmt.Println("Cursor Result: ", cursorResult[0]["uids"])

今天关于《使用mongoDB聚合获取特定字段的列表》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！