MongoDB聚合查询：将$lookup关联数据嵌入原数组对象

目前网络上关于MongoDB及其运行逻辑的公开资料十分丰富，但没有可直接参考的示例解决如下需求：按查询条件返回订单数据，同时将从products集合关联查询得到的商品信息，填充到对应订单的商品列表中。返回订单基础数据的实现难度很低，但在拆解对象数组、尤其是将关联查询结果写回数组内原有对象的环节容易遇到卡点。

涉及集合结构 #

orders 集合结构 #

{
    _id: ObjectId(...),
    products: [
        { product_id: ObjectId(...), config: { ... } },
        { product_id: ObjectId(...), config: { ... } }
    ]
}

products 集合结构 #

{
    _id: ObjectId(...),
    title: "Product 1",
    manufacturer: "Good Company"
},
{
    _id: ObjectId(...),
    title: "Product 2",
    manufacturer: "Another Company"
}

期望返回结果 #

{
    _id: ObjectId(...),
    products: [
        { product_id: ObjectId(...), config: { ... }, title: "Product 1", manufacturer: "Good Company" },
        { product_id: ObjectId(...), config: { ... }, title: "Product 2", manufacturer: "Another Company" }
    ]
}

原有代码问题 #

原有聚合查询代码如下：

db.collection('orders').aggregate([
    {
        $match: {
            // 该部分逻辑验证可用
            user_id: request.middleware.authentication.id,
            status: 'active'
        }
    },
    {
        $unwind: '$products'
    },
    {
        $lookup: {
            from: '$products',
            as: 'product_objects',
            let: { product_id: '$products.product_id' },
            pipeline: [
                {
                    $match: {
                        $expr: { $eq: [ '$$product_id', '$_id' ] }
                    }
                }
            ]
        }
    },
    {
        $unwind: {
            path: '$product_objects',
            preserveNullAndEmptyArrays: true
        }
    }
]);

原有代码存在两个核心问题：

• 仅能成功关联查询到1个商品的数据
• 关联查询结果不会写入products数组内的原有对象，而是会生成独立的product_objects新字段

正确实现方案 #

核心错误点先明确：

1. $lookup的from参数需要直接填写集合名称，不需要加$前缀，原有写法from: '$products'是导致关联结果异常的首要原因
2. 关联完成后需要合并商品字段到原有products对象，再通过分组把打散的订单商品重新组装为数组，否则会返回拆分后的多条订单记录，且字段无法合并

完整可用聚合代码如下：

db.collection('orders').aggregate([
    // 匹配目标订单，原有筛选逻辑保留
    {
        $match: {
            user_id: request.middleware.authentication.id,
            status: 'active'
        }
    },
    // 打散products数组，为单条商品关联做准备，兼容空商品列表场景
    {
        $unwind: {
            path: '$products',
            preserveNullAndEmptyArrays: true
        }
    },
    // 关联products集合，直接通过localField/foreignField关联写法更简洁
    {
        $lookup: {
            from: 'products',
            localField: 'products.product_id',
            foreignField: '_id',
            as: 'product_info'
        }
    },
    // 打散关联得到的商品信息数组，兼容商品已删除的空值场景
    {
        $unwind: {
            path: '$product_info',
            preserveNullAndEmptyArrays: true
        }
    },
    // 将关联得到的商品字段合并到原products对象中
    {
        $replaceRoot: {
            newRoot: {
                $mergeObjects: [
                    '$$ROOT',
                    { products: { $mergeObjects: ['$products', '$product_info'] } }
                ]
            }
        }
    },
    // 移除临时生成的冗余字段
    {
        $project: {
            product_info: 0
        }
    },
    // 按订单ID分组，将打散的商品重新组装为products数组
    {
        $group: {
            _id: '$_id',
            // 订单其他字段通过$first原样保留，有新增字段按相同格式补充即可
            user_id: { $first: '$user_id' },
            status: { $first: '$status' },
            products: { $push: '$products' }
        }
    }
])

补充说明 #

• 上述写法兼容MongoDB 3.2及以上所有版本，如果使用MongoDB 5.0+版本，也可以通过$lookup直接嵌套数组更新的写法减少聚合阶段，但上述写法通用性更强
• 所有$unwind阶段保留preserveNullAndEmptyArrays: true配置，可以避免商品被删除、订单商品列表为空的场景下整条订单数据丢失
• 分组阶段如果订单有除示例字段外的其他属性，需要在$group块中通过$first操作符逐一声明保留，否则对应字段会被丢弃

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