如何用MongoDB聚合将其他集合嵌套数组的频道ID关联到当前对象

集合结构 #

Users集合 #

{
    _id: ObjectId(friend1Id),
    username: 'john',
    friends: [friendId2, friendId3]
}
{
    _id: ObjectId(friend2Id),
    username: 'lucy',
    friends: [friendId1]
}
{
    _id: ObjectId(friend3Id),
    username: 'earl',
    friends: [friendId1]
}

Servers集合 #

{
    _id: s101,
    owner: 'john',
    name: 'name',
    channels: [
        {
             _id: ObjectId('c101'),
             with: friendId2
        },
        {
             _id: ObjectId('c102'),
             with: friendId3
        },
    ]
}

需求说明 #

获取John的所有好友，并为每个好友附加其与John在服务器中共享的频道_id，期望结果如下：

[
 {
    _id: friend2Id,
    username: 'lucy',
    friends: [friendId1],
    channel: ObjectId('c101')
 },
 {
    _id: friend3Id,
    username: 'earl',
    friends: [friendId1],
    channel: ObjectId('c102')
 }
]

问题分析 #

你之前的聚合查询通过$lookup关联了服务器集合，但返回的是整个服务器对象数组，而非单独的频道ID，不符合需求。需要调整lookup的内部管道，仅提取所需的频道ID，并将数组转换为单个字段值。

正确聚合实现 #

let friends = await Users.aggregate([
    // 匹配John的所有好友
    { $match: { friends: userId } },
    // 关联Servers集合，查找对应频道
    {
        $lookup: {
            from: 'Servers',
            let: { friend_id: '$_id' },
            pipeline: [
                // 仅匹配John拥有的服务器
                { $match: { owner: 'john' } },
                // 展开channels数组
                { $unwind: '$channels' },
                // 匹配当前好友对应的频道
                { $match: { $expr: { $eq: ['$channels.with', '$$friend_id'] } } },
                // 只保留频道ID字段
                { $project: { _id: 0, channel_id: '$channels._id' } }
            ],
            as: 'channel_info'
        }
    },
    // 将数组形式的channel_info转为单个字段
    {
        $addFields: {
            channel: { $arrayElemAt: ['$channel_info.channel_id', 0] }
        }
    },
    // 移除临时字段
    { $project: { channel_info: 0 } }
]).toArray();

关键修改说明 #

1. 限定服务器范围：在lookup内部管道中加入{ $match: { owner: 'john' } }，确保只查找John拥有的服务器，过滤无关数据。
2. 精准提取频道ID：通过$project仅保留channels._id并命名为channel_id，剔除不需要的服务器字段。
3. 数组转单值：使用$arrayElemAt从channel_info数组中取出唯一的频道ID，赋值给channel字段。
4. 清理临时字段：最后用$project删除channel_info临时数组，让结果结构更简洁。

注意：如果Servers集合中channels.with存储的是字符串类型而非ObjectId，需在聚合开头添加类型转换步骤：

{ $addFields: { friend_id_str: { $toString: '$_id' } } }

并将lookup的let改为{ friend_id: '$friend_id_str' }，内部$match条件调整为{ $eq: ['$channels.with', '$$friend_id'] }。

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