MongoDB使用geoNear时如何保留无地理位置字段的文档并排序

当使用$geoNear时，所有没有location字段（未纳入2dsphere索引）的文档会被自动过滤。要实现获取最近地理位置结果的同时保留无地理位置文档，并将这类文档排在结果末尾，可以通过MongoDB的聚合管道$facet和$concatArrays来实现，具体方案如下：

实现思路 #

• 用$facet拆分聚合逻辑为两个分支：
  • 分支1：处理带location字段的文档，通过$geoNear计算距离并按距离排序
  • 分支2：筛选出不带location字段的文档
• 用$concatArrays合并两个分支的结果，确保带地理位置的文档在前，无地理位置的文档在后

修改后的聚合代码示例 #

.collection('users')
.aggregate([
  {
    $facet: {
      // 处理有location的文档，计算距离并排序（geoNear默认按距离升序排列）
      withLocation: [
        {             
          $geoNear: {
             near: { type: "Point", coordinates: [userLocArray[0], userLocArray[1]]},
             distanceField: "dist.calculated",
             spherical: true,
             // maxDistance: matchDistance, // 按需启用最大距离过滤
          }
        }
      ],
      // 筛选无location的文档
      withoutLocation: [
        { $match: { location: { $exists: false } } }
      ]
    }
  },
  // 合并结果集：有地理位置的在前，无地理位置的在后
  {
    $project: {
      combinedResults: { $concatArrays: ["$withLocation", "$withoutLocation"] }
    }
  },
  // 展开合并后的数组为单个文档（按需选择，若业务需要直接用数组可省略）
  { $unwind: "$combinedResults" },
  // 恢复为原始文档结构（按需选择）
  { $replaceRoot: { newRoot: "$combinedResults" } }
])

文档示例（中文翻译） #

db={
  users: [
    {
      _id: "abc",
      name: "abc",
      group: 1,
      location: {
        type: "Point",
        coordinates: [53.23, 67.12]
      },
      calculatedDist: 112
    },
    {
      _id: "xyz",
      name: "xyyy",
      group: 1,
      calculatedDist: 13
    },
    {
      _id: "123",
      name: "yyy",
      group: 1,
      location: {
        type: "Point",
        coordinates: [54.23, 67.12]
      },
      calculatedDist: 13
    },
    {
      _id: "rrr",
      name: "tttt",
      group: 1,
      location: {
        type: "Point",
        coordinates: [51.23, 64.12]
      },
      calculatedDist: 14
    },
    {
      _id: "mmm",
      name: "mmmm",
      group: 1,
      calculatedDist: 14
    }
  ]
}

关键说明 #

• $facet支持在一个聚合流程中并行处理多组文档，互不干扰
• $geoNear默认已按距离从近到远排序结果，无需额外添加$sort（如需降序可自行补充）
• 无地理位置文档的筛选逻辑通过$match: { location: { $exists: false } }实现，确保不会遗漏这类数据

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