MongoDB聚合$lookup中$geoWithin空间关联的问题及解决方法

集合结构与索引准备 #

现有两个存储地理空间数据的集合，且已为两个集合的location字段创建2DSphere地理空间索引：

• addresses集合（存储地址点位）：

{
    "id_addr" : "id1",
    "location" : {
        "type" : "Point",
        "coordinates" : [4.00,45.00]
    }
}

• zones集合（存储区域多边形）：

{
    "id_zone" : "zone1",
    "location" : {
        "type" : "Polygon",
        "coordinates" : [[[3.00,40.00],[5.00,40.00],[5.00,45.00],[3.00,45.00], [3.00,40.00]]]
    }
}

基础空间查询验证 #

使用$geoWithin可直接查询指定多边形内的地址点位，查询语句执行正常：

db.adresses.find({
    "location" : { 
        $geoWithin : { 
            "$geometry" : { 
                "type" : "Polygon", 
                "coordinates" : [[[3.00,40.00],[5.00,40.00],[5.00,45.00],[3.00,45.00], [3.00,40.00]]]
            }
        }
    }
})

$lookup关联查询的问题与报错 #

尝试在zones集合的聚合中用$lookup关联addresses、筛选区域内地址时，两次写法均报错：

1. 第一种写法报错：unknown geo specifier: $geometry: "$$zonePolygon"

$lookup{
  from: "adresses",
  let : {zonePolygon:"$location"},
  as: "zones",
  pipeline : [
    {
      $match : {
        location : {
          $geoWithin : {
            $geometry : "$$zonePolygon"
          }
        }       
      }
    }
  ]
}

2. 调整写法后报错："Polygon coordinates must be an array, instead got type string"

$lookup : {
  from: "addresses",
  let : {zoneCoordinates:"$location.coordinates"},
  as: "zones",
  pipeline : [
    {
      $match : {
        location : {
          $geoWithin : {
            $geometry : {
              type :"Polygon",
              coordinates : "$$zoneCoordinates"
            }
          }
        }       
      }
    }
  ]
}

问题根源 #

$lookup的let变量仅能在$match的$expr算子中使用，但MongoDB不支持在$expr内使用$geoWithin，导致无法直接通过变量传递地理空间参数。

解决方案：$geoNear实现关联 #

通过在addresses集合的聚合查询中使用$geoNear算子解决：当点位被多边形包含时，两者的空间距离为0，以此筛选关联区域。具体$lookup阶段写法如下：

{
  from: "polygons", 
  let : {pointGeometry : "$location"}, // 基于点位的location字段声明变量
  as: "zones",
  pipeline : [
    {
      $geoNear : {
        near : "$$pointGeometry", // 在$geoNear中可直接使用变量
        distanceField : "distance", // 存储点与多边形的距离
        spherical : false,
        maxDistance : 0 // 筛选距离为0的多边形（即包含该点的区域）
      }
    },
    {
      $project : {
        id_zone : 1, // 仅保留id_zone字段
      }
    }
  ]
}

该方案可成功将每个地址点位与包含它的区域关联，后续可通过$unwind等算子进一步处理结果。

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