如何对MongoDB find方法返回的每个文档执行aggregate查询

核心问题说明 #

你当前的实现思路存在两个明显问题：

1. MongoDB Node.js 驱动中，find()返回的游标调用forEach()后会直接消费游标，该方法返回值为Promise（异步逻辑），无法再链式调用toArray()，你的原有写法会直接抛出语法错误。
2. 对find返回的每一份文档单独发起aggregate请求，会产生大量不必要的数据库交互，性能损耗极高，该需求完全可以通过一次聚合查询合并所有逻辑实现。

最优实现方案（推荐） #

直接通过一次聚合查询合并你的两个查询条件即可，注意$search是MongoDB Atlas Search的专属stage，必须放在聚合pipeline的第一位，我们可以通过compound语法同时合并全文搜索和名称正则匹配条件：

app.get('/', async (req, res) => {
  try {
    const result = await db.collection('subs').aggregate([
      {
        $search: {
          index: 'default',
          compound: {
            // 全文搜索匹配所有字段包含night的文档
            must: [
              {
                text: {
                  query: 'night',
                  path: { wildcard: '*' }
                }
              }
            ],
            // 过滤name字段符合dexter正则的文档，不影响搜索算分
            filter: [
              {
                regex: {
                  query: 'dexter',
                  path: 'name',
                  allowAnalyzedField: true,
                  options: 'i'
                }
              }
            ]
          }
        }
      },
      // 等价于原find逻辑中的projection
      {
        $project: { _id: 0, content: 0 }
      }
    ]).toArray();

    res.json({
      length: result.length,
      body: { result }
    });
  } catch (err) {
    throw err;
  }
});

特殊场景实现（不推荐，仅满足强需求） #

如果你的业务场景确实必须先执行find再对每个单独文档执行aggregate查询，可参考以下异步处理逻辑：

app.get('/', async (req, res) => {
  try {
    // 先执行find查询拿到所有符合条件的文档，这里需要保留_id用于后续单文档匹配
    const findResults = await db.collection('subs').find(
      { name: { $regex: 'dexter', $options: 'i' } }
    ).toArray();

    // 批量处理每个文档的aggregate查询
    const processedResults = await Promise.all(
      findResults.map(async (doc) => {
        const searchRes = await db.collection('subs').aggregate([
          // 匹配当前单文档
          { $match: { _id: doc._id } },
          {
            $search: {
              index: 'default',
              text: {
                query: 'night',
                path: { wildcard: '*' }
              }
            }
          },
          { $project: { _id: 0, content: 0 } }
        ]).toArray();
        // 合并原文档与搜索结果返回
        return {
          ...doc,
          content: undefined,
          _id: undefined,
          searchResult: searchRes
        };
      })
    );

    res.json({
      length: processedResults.length,
      body: { result: processedResults }
    });
  } catch (err) {
    throw err;
  }
});

注意事项 #

• 你原有代码中的正则选项写为$i，正确写法为i，不需要加$符号，原有写法会导致正则不生效。
• $search为MongoDB Atlas专属功能，自建MongoDB服务未部署Atlas Search的场景下无法使用该stage。

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