如何在MongoDB聚合中使用$lookup为嵌套对象补充关联字段

你可以通过拆分数组-关联查询-合并字段-重组数组的聚合流程实现需求，完整的聚合管道代码如下（Pymongo语法）：

pipeline = [
    # 第一步：拆分contacts数组，将每个联系人转为独立文档
    {"$unwind": "$contacts"},
    # 第二步：关联users集合，查询对应用户信息
    {
        "$lookup": {
            "from": "users",
            "localField": "contacts.user_id",
            "foreignField": "_id",
            "as": "user_info"
        }
    },
    # 拆分user_info数组（$lookup默认返回数组，匹配唯一用户的情况下只有一个元素）
    {"$unwind": "$user_info"},
    # 第三步：关联locations集合，查询对应地点信息
    {
        "$lookup": {
            "from": "locations",
            "localField": "contacts.location_id",
            "foreignField": "_id",
            "as": "location_info"
        }
    },
    # 拆分location_info数组
    {"$unwind": "$location_info"},
    # 第四步：合并字段到contacts对象中
    {
        "$addFields": {
            "contacts.user_name": "$user_info.name",
            "contacts.user_age": "$user_info.age",
            "contacts.location_name": "$location_info.name"
        }
    },
    # 第五步：按原文档_id分组，将处理后的contacts元素重组为数组
    {
        "$group": {
            "_id": "$_id",
            "contacts": {"$push": "$contacts"}
        }
    }
]

# 执行查询
result = list(db.contacts.aggregate(pipeline))

核心逻辑说明 #

• $unwind用来拆解嵌套数组，让数组内的每个对象都能作为独立文档参与关联查询，避免出现独立的关联结果数组
• 两次$lookup分别关联用户和地点信息后，直接通过$addFields把需要的字段写入原contacts对象的对应属性
• 最后用$group把分散的单条联系人数据重新合并为数组，还原原始文档结构

如果需要保留没有匹配到用户/地点的联系人项，可以给$unwind添加"preserveNullAndEmptyArrays": True参数即可。

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