MongoDB聚合查询如何仅返回文档中匹配条件的字段

核心实现原理 #

• 聚合管道中$match阶段仅负责筛选符合$or规则的文档，无法动态控制返回字段，需要在后续新增$project阶段实现字段动态裁剪
• 利用MongoDB系统变量$$REMOVE：当字段被赋值为$$REMOVE时，该字段会自动从最终输出文档中剔除，正好满足「命中规则就保留字段，未命中就移除」的需求
• 对$or查询中的每个匹配分支，单独做字段级的规则校验，匹配则保留字段值，不匹配则赋值为$$REMOVE

具体实现代码 #

通用逻辑说明 #

1. 保留原有$match阶段逻辑，先过滤出所有符合整体查询条件的文档，减少后续计算量
2. 新增$project阶段，对每个参与$or匹配的字段单独做匹配判断：
   • 判断字段值是否符合该字段对应的查询规则
   • 符合则返回字段本身的值
   • 不符合则赋值为$$REMOVE
3. 该阶段后可正常追加$lookup等多表关联阶段，不影响后续关联逻辑；如果关联逻辑需要用到其他字段，可将$project阶段放在所有关联处理的最后。

第一个查询场景实现 #

查询规则：name匹配正则1 或 address匹配正则bb，命中哪个字段就返回哪个字段

const users = Users.aggregate([
  {
    $match: {
      $or: [
        { name: { $regex: "1" } },
        { address: { $regex: "bb" } }
      ]
    }
  },
  {
    $project: {
      name: {
        $cond: [
          // 判断name字段是否匹配对应规则
          { $regexMatch: { input: "$name", regex: "1" } },
          "$name",
          "$$REMOVE"
        ]
      },
      address: {
        $cond: [
          // 判断address字段是否匹配对应规则
          { $regexMatch: { input: "$address", regex: "bb" } },
          "$address",
          "$$REMOVE"
        ]
      }
    }
  }
  // 后续可追加$lookup等其他聚合阶段
]);

执行后返回结果和预期完全一致：abc1、pqr1的name命中规则保留，address未命中bb规则被移除，仅返回_id和name字段。

第二个查询场景实现 #

查询规则：name匹配正则5 或 address匹配正则dd，仅需修改$project中对应字段的匹配正则即可：

const users = Users.aggregate([
  {
    $match: {
      $or: [
        { name: { $regex: "5" } },
        { address: { $regex: "dd" } }
      ]
    }
  },
  {
    $project: {
      name: {
        $cond: [
          { $regexMatch: { input: "$name", regex: "5" } },
          "$name",
          "$$REMOVE"
        ]
      },
      address: {
        $cond: [
          { $regexMatch: { input: "$address", regex: "dd" } },
          "$address",
          "$$REMOVE"
        ]
      }
    }
  }
]);

执行后所有文档的name均未匹配5被移除，address均包含dd被保留，返回结果和预期一致。

低版本MongoDB兼容方案 #

$regexMatch是MongoDB 4.2及以上版本支持的操作符，如果使用4.2以下版本，可将规则判断替换为$indexOfCP实现字符串匹配判断，以name匹配1为例，判断逻辑替换为：

// 等价于判断name字段是否包含字符串"1"，找到则返回索引位置> -1，找不到返回-1
{ $gt: [ { $indexOfCP: [ "$name", "1" ] }, -1 ] }

如果正则包含忽略大小写、通配符等复杂规则，建议升级至4.2及以上版本使用$regexMatch实现，逻辑更简洁。

如果存在文档同时命中多个字段的匹配规则，对应多个字段会同时保留返回，符合匹配逻辑预期。

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