使用 mongoDB 在不同“列”中聚合特定字段的列表

Question

在 Golang 上使用 mongo-driver。为了保存 2 个实体之间的关系列表，我使用了一个集合，其中每个文档都有一个“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]这可能与聚合有关吗？希望我很清楚，如果有任何未明确解释的内容，请在下面发表评论。

Answer 1

下面的代码应该为您解决问题

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"])