MongoDB 聚合 - 动态`$addField`

Question

我有以下 _id: userID 格式的对象和一个 Mongoose 查询：

const obj = {
    111: "222",
    333: "444",
    555: "666"
};

Model.aggregate([
    {
        $addFields: {
            userID: ???
        }
    }
]);

我需要一种基于对象向每个文档添加userID 字段的方法。因此，假设该集合具有以下数据：

[
    {
        _id: "111",
        ...
    },
    {
        _id: "333",
        ...
    },
    {
        _id: "555",
        ...
    }
]

文档最终应该是这样的：

[
    {
        _id: "111",
        userID: "222",
        ...
    },
    {
        _id: "333",
        userID: "444",
        ...
    },
    {
        _id: "555",
        userID: "666",
        ...
    }
]

我该怎么做？

Answer 1

您需要从$objectToArray 开始，以将您的动态对象作为键和值的数组。然后您可以使用$filter 查找匹配对，$arrayElemAt 获取单个元素：

db.collection.aggregate([
    {
        $addFields: {
            userId: {
                $let: {
                    vars: { 
                        match: { 
                            $arrayElemAt: [ 
                                { 
                                    $filter: { 
                                        input: { $objectToArray: {  111: "222", 333: "444", 555: "666" } }, 
                                        cond: { $eq: [ "$$this.k", { $toString: "$_id" } ] }
                                    } 
                                },
                                0 
                            ] 
                        }
                    },
                    in: "$$match.v"
                }
            }
        }
    }
])

Mongo Playground