如何基于MongoDB集合匹配结果过滤JavaScript数组？

针对你的需求，我们可以通过MongoDB的聚合管道实现一次查询完成筛选，无需遍历数组逐个发起请求。以下是具体实现：

核心思路 #

通过聚合管道将输入的timestamps数组与windows集合关联，筛选出满足「存在对应device_id的窗口文档，且timestamp处于窗口时间范围内」的数组元素。

聚合查询代码 #

假设你的集合名为windows，输入数组为timestamps，执行以下聚合操作：

const timestamps = [
   { device_id: 1, timestamp: "2024-05-12T13:00:00Z"}, 
   { device_id: 3, timestamp: "2024-05-12T13:00:00Z"},
   { device_id: 4, timestamp: "2024-05-12T13:00:00Z"}
];

db.windows.aggregate([
  // 1. 过滤出输入数组中存在的device_id对应的窗口文档，减少处理数据量
  {
    $match: {
      device_id: { $in: timestamps.map(item => item.device_id.toString()) }
    }
  },
  // 2. 将输入的timestamps数组附加到每个窗口文档中
  {
    $addFields: { timestamps }
  },
  // 3. 展开timestamps数组，逐个检查每个元素
  { $unwind: "$timestamps" },
  // 4. 匹配device_id一致且timestamp处于窗口范围内的项
  {
    $match: {
      $expr: {
        $and: [
          // 统一device_id类型（集合中为字符串，输入可能是数字）
          { $eq: ["$device_id", { $toString: "$timestamps.device_id" }] },
          // timestamp大于窗口开始时间
          { $gt: [{ $toDate: "$timestamps.timestamp" }, { $toDate: "$start_time" }] },
          // 无结束时间 或 timestamp小于结束时间
          {
            $or: [
              { $not: { $exists: "$end_time" } },
              { $lt: [{ $toDate: "$timestamps.timestamp" }, { $toDate: "$end_time" }] }
            ]
          }
        ]
      }
    }
  },
  // 5. 去重：同一个device_id只保留第一个符合条件的记录
  {
    $group: {
      _id: "$timestamps.device_id",
      timestamp: { $first: "$timestamps.timestamp" }
    }
  },
  // 6. 整理成目标输出格式
  {
    $project: {
      _id: 0,
      device_id: "$_id",
      timestamp: 1
    }
  }
])

代码解释 #

1. $match阶段：快速过滤出和输入数组中device_id匹配的窗口文档，避免处理无关数据，提升查询效率。注意集合中device_id为字符串类型，需将输入的数字类型转换为字符串匹配。
2. $addFields阶段：把输入的timestamps数组添加到每个窗口文档中，为后续关联检查做准备。
3. $unwind阶段：将timestamps数组展开，每个窗口文档对应数组中的一个元素，实现逐个元素的条件检查。
4. $match阶段：用$expr实现字段间的逻辑比较，确保device_id一致，同时验证timestamp是否在窗口时间范围内（无end_time时直接满足条件）。
5. $group阶段：按device_id分组，取第一个符合条件的timestamp，避免同一个device因匹配多个窗口而重复输出。
6. $project阶段：调整输出结构，去除_id字段，得到你需要的格式。

示例验证 #

针对你提供的示例数据，执行上述查询后，会输出：

[{ device_id: 1, timestamp: "2024-05-12T13:00:00Z"}]

完全符合预期结果。

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