MongoDB聚合管道中实现数组交集判断并生成匹配字段的方法

需求回顾 #

现有文档集合，部分文档包含teamsIds数组字段，示例如下：

{
  "id": 1,
  "teamsIds": ["team_a", "team_b"]
}

需要编写聚合管道（先执行$match阶段），最终返回格式为：

{
  "id": 1,
  "matchingFilter": true
}

其中matchingFilter为true的条件是：文档的teamsIds数组与指定的团队列表（如["team_c", "team_a"]）存在任意交集，且不能使用$elemMatch。

最简实现方案 #

可以通过$setIntersection结合$ne操作符实现，无需依赖$elemMatch，具体管道如下：

db.collection.aggregate([
  // 第一步：先筛选出teamsIds与目标列表有交集的文档，减少后续处理量
  {
    $match: {
      teamsIds: { $in: ["team_c", "team_a"] }
    }
  },
  // 第二步：生成matchingFilter字段
  {
    $project: {
      id: 1,
      matchingFilter: {
        $ne: [
          { $setIntersection: ["$teamsIds", ["team_c", "team_a"]] },
          []
        ]
      }
    }
  }
])

逻辑说明 #

1. $match阶段：使用$in操作符筛选出teamsIds包含目标列表中任意团队的文档，提前过滤掉无关数据，提升管道效率。
2. $project阶段：
   • $setIntersection：计算当前文档teamsIds与目标团队列表的交集数组。
   • $ne：判断交集数组是否为空数组——若不为空（即存在匹配项），则返回true，否则返回false。

替代方案（用$anyElementTrue+$map） #

如果偏好另一种写法，也可以通过$map遍历目标团队列表，再用$anyElementTrue判断是否存在匹配项：

{
  $project: {
    id: 1,
    matchingFilter: {
      $anyElementTrue: {
        $map: {
          input: ["team_c", "team_a"],
          as: "team",
          in: { $in: ["$$team", "$teamsIds"] }
        }
      }
    }
  }
}

不过相比之下，$setIntersection的写法更简洁直观。

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