MongoDB中Aggregation Lookup结合Unwind查询失效问题求助

场景说明 #

我有Users和Courses两个MongoDB Schema，Courses的mapped_to字段存储关联的用户ID数组，需要通过aggregate、lookup和unwind操作，把每个课程对应的完整用户数据整合到课程结果中，但当前编写的聚合查询无效。

现有代码与Schema结构 #

Courses控制器中的聚合代码

const aggr = Course.aggregate([
    {
        $unwind:"$mapped_to"
    },
    {
        $lookup:{
            from:'users',
            localField:"$mapped_to",
            foreignField:"_id",
            as:"result"
        }
    },
    {
        $group:{
            _id:"$_id",
            arrayField:{$push:"$mapped_to"},
            result:{$first:"$result"}
        }
    }
]).then((res)=>{
    console.log(res)
})

Courses Schema（部分字段）

const mongoose = require('mongoose');
const schema = mongoose.Schema;
const Course = new schema({
    tag:{
        type:String,
        minlength:2,
        maxlength:20
    },
    mapped_to:{
        type:Array,
        required:true,
    },
    added_date:{
        type:String,
        required:true,
        minlength:2,
        maxlength:500
    }
})

Users Schema

const mongoose = require('mongoose');
const schema = mongoose.Schema;
const User = new schema({
    first_name: {
        type: String,
        required: true,
        minlength: 1,
        maxlength: 50
    },
    last_name: {
        type: String,
        required: true,
        minlength: 1,
        maxlength: 50
    },
    email: {
        type: String,
        required: true,
        minlength: 5,
        maxlength: 255,
        unique: true
    }
})

问题分析 #

原查询无效的核心原因有以下几点：

1. $lookup的localField参数错误：不需要加$前缀，正确写法是"mapped_to"
2. $group阶段处理result字段时使用$first，只会保留第一个关联的用户数据，无法收集所有匹配的用户
3. 未保留课程的其他字段（如tag、added_date），导致结果缺失关键信息
4. 可能存在类型不匹配：如果mapped_to中存储的是字符串ID，而Users的_id是ObjectId类型，会导致关联失败

修正后的聚合代码 #

情况1：mapped_to存储的是ObjectId类型数组 #

Course.aggregate([
    // 拆分用户ID数组，为每个ID生成一条文档
    { $unwind: "$mapped_to" },
    // 关联users集合，获取完整用户数据
    {
        $lookup: {
            from: "users",
            localField: "mapped_to", // 去掉$前缀
            foreignField: "_id",
            as: "user"
        }
    },
    // 拆分lookup返回的数组（因为每个mapped_to对应一个用户，拆分后方便后续分组）
    { $unwind: "$user" },
    // 按课程ID分组，重组用户数组并保留原课程字段
    {
        $group: {
            _id: "$_id",
            tag: { $first: "$tag" },
            added_date: { $first: "$added_date" },
            mapped_to: { $push: "$mapped_to" }, // 原ID数组
            mapped_users: { $push: "$user" } // 关联的完整用户数据数组
        }
    }
]).then(res => {
    console.log(res);
}).catch(err => {
    console.error(err);
});

情况2：mapped_to存储的是字符串类型ID #

需要先将字符串ID转为ObjectId，否则无法与Users的_id匹配：

const mongoose = require('mongoose');

Course.aggregate([
    { $unwind: "$mapped_to" },
    // 将字符串ID转为ObjectId
    {
        $addFields: {
            mapped_to_obj: { $toObjectId: "$mapped_to" }
        }
    },
    {
        $lookup: {
            from: "users",
            localField: "mapped_to_obj",
            foreignField: "_id",
            as: "user"
        }
    },
    { $unwind: "$user" },
    {
        $group: {
            _id: "$_id",
            tag: { $first: "$tag" },
            added_date: { $first: "$added_date" },
            mapped_to: { $push: "$mapped_to" },
            mapped_users: { $push: "$user" }
        }
    }
]).then(res => {
    console.log(res);
}).catch(err => {
    console.error(err);
});

代码解释 #

1. $unwind: "$mapped_to"：将课程的用户ID数组拆分为多条文档，每条对应一个用户ID
2. $lookup：关联users集合，根据用户ID匹配获取完整用户数据
3. 额外的$unwind: "$user"：因为$lookup返回的是数组（即使只有一个匹配项），拆分后方便后续分组收集
4. $group：按课程ID重新分组，用$push收集所有关联的用户数据，同时用$first保留课程的其他字段（因为同一课程的这些字段值相同）

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