Mongoose如何实现跨Schema关联查询，合并用户权限校验与项目查询

完全可以将两次数据库查询合并为MongoDB层面的单次查询，使用聚合操作（aggregate）即可实现，相比原有两次查询的写法可以减少一次数据库IO开销，大幅提升查询效率。

最优实现方案（聚合查询，真正单次数据库请求） #

import { Types } from 'mongoose';

export async function findProjectForUser(user: User, id: string) {
  // 字符串ID转MongoDB ObjectId类型，避免类型不匹配导致匹配失败
  const targetProjectId = new Types.ObjectId(id);
  const userId = new Types.ObjectId(user.id);

  const queryResult = await UserModel.aggregate([
    // 匹配目标用户，过滤不存在的用户
    { $match: { _id: userId } },
    // 过滤无项目权限的用户
    { $match: { projectIds: targetProjectId } },
    // 关联project集合拉取目标项目详情
    {
      $lookup: {
        from: 'project',
        localField: 'projectIds',
        foreignField: '_id',
        pipeline: [{ $match: { _id: targetProjectId } }],
        as: 'targetProject'
      }
    },
    // 拆分项目数组为单个对象
    { $unwind: '$targetProject' },
    // 直接返回项目内容作为最终结果
    { $replaceRoot: { newRoot: '$targetProject' } }
  ]);

  // 结果为空等价于用户不存在/无权限，和原有逻辑完全对齐
  return queryResult.length ? queryResult[0] : undefined;
}

优势说明 #

• 所有逻辑都在数据库层面完成，仅产生1次数据库网络IO，数据量越大性能提升越明显
• 不需要将用户全量的projectIds拉到应用层做遍历判断，减少数据传输和内存开销
• 上层错误处理逻辑不需要做任何修改，返回值规则和原有实现完全一致

简化写法（populate，本质两次查询） #

如果你更看重代码简洁性，对性能要求没有极致高，也可以用populate实现，写法更简单，但本质是Mongoose内部帮你做了两次数据库查询：

export async function findProjectForUser(user: User, id: string) {
  const dbUser = await UserModel.findById(user.id).populate({
    path: 'projectIds',
    match: { _id: id },
    select: 'name description createdAt updatedAt'
  });

  if (!dbUser || !dbUser.projectIds?.length) return undefined;
  return dbUser.projectIds[0];
}

注意事项 #

• 所有需要参与匹配的ID都要先转为Types.ObjectId类型，字符串类型的ID无法和MongoDB默认存储的ObjectId匹配
• 如果使用低版本Mongoose（<5.0）不支持$lookup的pipeline参数，可以把项目匹配逻辑放到$lookup之后的$match阶段即可。

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