MongoDB聚合Lookup：替换数组内关联字段值的实现方案

背景 #

Stack Overflow上有很多关于MongoDB、Aggregate和Lookup的问题，但没找到直接匹配当前需求的答案。现有两个集合：

• Support集合：文档包含instructions对象数组，数组中每个对象的image属性是Images集合的id字符串。
• Images集合：文档包含id（对应Support的image字段）和实际的image值。

需求：聚合时先通过match匹配单个Support文档，再将instructions数组中每个对象的image属性替换为Lookup查询到的对应Images文档的实际image值，仅修改该属性，不新增其他字段。

问题 #

1. 如何实现上述需求？
2. 若使用Lookup管道，能否不借助let访问localField？
3. 能否将所有逻辑都放在Lookup管道内完成？
4. 实现该需求的最优方式是什么？

集合结构及尝试的聚合代码 #

Support集合 #

DB.support = [
    {
        _id: ObjectId("12345"),
        title: "Support",
        instructions: [
            {
                image: "image-id-string-one"
            },
            {
                image: "image-id-string-two"
            },
            {
                image: "image-id-string-three"
            },
        ]
    }
]

Images集合 #

DB.images = [
    {
        _id: ObjectId("11111111"),
        id:"image-id-string-one",
        image: "actual-image-one"
    },
    {
        _id: ObjectId("222222222"),
        id:"image-id-string-two",
        image: "actual-image-two"
    },
    {
        _id: ObjectId("333333333"),
        id:"image-id-string-three",
        image: "actual-image-three"
    }
]

尝试的聚合代码 #

db.collection('support').aggregate([
    {
        "$match": {
            "title": "Support"
        }
    },
    // 假设match只返回一条记录
    {
        "$unwind": "$instructions"
    },
    {
        "$lookup": {
            "from": "images",
            "localField": "instructions.image",
            "foreignField": "id",
            "as": "images"
        }
    },
    {
        "$unwind": "$images"
    },
    {
        "$set": {
            "instructions.image": "$images.image"
        }
    },
    // 移除lookup产生的images字段
    {
        "$unset": ["images"]
    }
])

期望结果 #

{
    _id: ObjectId("12345"),
    title: "Support",
    instructions: [
        {
            image: "actual-image-one"
        },
        {
            image: "actual-image-two"
        },
        {
            image: "actual-image-three"
        },
    ]
}

注：尝试了多种方式都没得到期望结果

解答 #

1. 实现需求的完整聚合代码 #

你之前的代码差最后一步——重新将instructions数组聚合回来，因为$unwind把数组拆成了多条文档，需要用$group合并：

db.collection('support').aggregate([
    { "$match": { "title": "Support" } },
    { "$unwind": "$instructions" },
    {
        "$lookup": {
            "from": "images",
            "localField": "instructions.image",
            "foreignField": "id",
            "as": "images"
        }
    },
    { "$unwind": "$images" },
    { "$set": { "instructions.image": "$images.image" } },
    { "$unset": ["images"] },
    // 关键一步：重新分组合并instructions数组
    {
        "$group": {
            "_id": "$_id",
            "title": { "$first": "$title" },
            "instructions": { "$push": "$instructions" }
        }
    }
])

另外，也可以用$lookup的管道模式，不用$unwind和$group，更高效：

db.collection('support').aggregate([
    { "$match": { "title": "Support" } },
    {
        "$lookup": {
            "from": "images",
            "let": { "instructions": "$instructions" },
            "pipeline": [
                {
                    "$match": {
                        "$expr": { "$in": ["$id", "$$instructions.image"] }
                    }
                },
                { "$project": { "_id": 0, "id": 1, "image": 1 } }
            ],
            "as": "imageMap"
        }
    },
    {
        "$set": {
            "instructions": {
                "$map": {
                    "input": "$instructions",
                    "as": "item",
                    "in": {
                        "image": {
                            "$getField": {
                                "field": "image",
                                "input": {
                                    "$first": {
                                        "$filter": {
                                            "input": "$imageMap",
                                            "cond": { "$eq": ["$$this.id", "$$item.image"] }
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
            },
            "imageMap": "$$REMOVE" // 移除临时字段
        }
    }
])

2. 使用Lookup管道能否不借助let访问localField？ #

不行。Lookup的管道模式（指定pipeline参数）下，子管道无法直接访问父文档的字段，必须通过let定义变量传递父文档字段值，子管道内用$$变量名引用。

3. 能否将所有逻辑都放在Lookup管道内完成？ #

不能。Lookup的作用是关联外部集合并返回结果，而修改原文档instructions数组的操作，必须在Lookup之后用$map/$filter等操作完成——子管道只能处理关联的Images文档，无法修改原Support文档的结构。

4. 最优实现方式 #

如果instructions数组不大，两种方案都可行；若数组较大，推荐**管道式Lookup+$map+$filter**的方案：

• 避免了$unwind和$group的拆分合并操作，减少中间文档数量，性能更优；
• 仅需一次Lookup查询所有关联的Images文档，再通过内存映射替换字段，效率更高。

如果你的MongoDB版本是5.0+，也可以用$lookup的localField/foreignField结合$mergeObjects和$map，但管道式方案兼容性更好。

内容的提问来源于stack exchange，提问作者SoEzPz