如何在聚合管道阶段计算连续点间的公里距离

我来帮你搞定这个连续坐标点的距离计算问题！核心思路是先把文档按时间戳排序确保坐标点的时间顺序，再用窗口函数获取前一个点的坐标，最后用MongoDB的球面距离工具计算公里数。以下是完整的解决方案：

完整聚合管道代码 #

db.yourCollectionName.aggregate([
  // 第一步：按时间戳升序排序，确保坐标点按时间先后排列
  {
    $sort: { "firstDoc.ts": 1 }
  },
  // 第二步：用窗口函数获取前一个文档的坐标点
  {
    $setWindowFields: {
      partitionBy: null, // 整个集合作为一个统一分区
      sortBy: { "firstDoc.ts": 1 },
      output: {
        previousCoordinate: {
          $lag: "$firstDoc.coordinate",
          outputMissing: null // 第一个文档没有前序点，设为null
        },
        previousTime: {
          $lag: "$firstDoc.ts",
          outputMissing: null
        }
      }
    }
  },
  // 第三步：过滤掉没有前序坐标的文档（也就是第一个文档）
  {
    $match: {
      previousCoordinate: { $ne: null }
    }
  },
  // 第四步：计算两点间的公里距离
  {
    $addFields: {
      distanceKm: {
        $divide: [
          {
            $distanceSphere: [
              "$previousCoordinate",
              "$firstDoc.coordinate"
            ]
          },
          1000 // 将米转换为公里
        ]
      }
    }
  },
  // 第五步：整理输出字段，让结果更清晰
  {
    $project: {
      _id: 0,
      fromPoint: "$previousCoordinate",
      toPoint: "$firstDoc.coordinate",
      distanceKm: { $round: ["$distanceKm", 2] }, // 保留两位小数
      timePeriod: {
        start: "$previousTime",
        end: "$firstDoc.ts"
      }
    }
  }
])

关键步骤详解 #

• $sort阶段：这一步是基础！必须按firstDoc.ts排序，才能保证后续拿到的是时间上连续的坐标点，不然计算出来的距离毫无意义。
• $setWindowFields阶段：用$lag函数“拉取”上一个文档的坐标和时间，partitionBy: null确保所有文档都在同一个排序序列里，不会被拆分。
• $match阶段：过滤掉第一个文档（它没有前序坐标，没法计算距离），剩下的三个文档正好对应你需要的三组连续点计算。
• $addFields阶段：$distanceSphere是MongoDB专门用来计算球面两点距离的函数，返回单位是米，除以1000就转换成公里了，用$round可以让结果更整洁。
• $project阶段：把输出字段整理成一目了然的格式，方便你查看每一段距离对应的起点、终点和时间区间。

注意事项 #

如果你的coordinate数组是[纬度, 经度]的格式，记得先调整顺序！MongoDB地理空间函数要求坐标格式是[经度, 纬度]，可以用$reverseArray来转换，比如在$addFields里把坐标改成$reverseArray("$previousCoordinate")。

内容的提问来源于stack exchange，提问作者Cyrus the Great