получить документ с помощью $ box с mongodb и добавить поле расстояния для каждой записи с указанием c координатной точки - PullRequest
Новая
       30

получить документ с помощью $ box с mongodb и добавить поле расстояния для каждой записи с указанием c координатной точки

0 голосов
/ 27 февраля 2020

Случай

Я сохранил координаты местоположения в коллекции MongoDB с индексом 2dsphere. Я хочу получить записи, которые входят в указанные c границы на карте. У меня есть одна точка координат, которая фиксирована широтой и долготой. Я использовал метод $geoWithin с $box в агрегации MongoDB. Я могу получить данные правильно.

Проблема

Мне нужно добавить поле расстояния в каждой записи из одной фиксированной широты и долготы, которые у меня есть. 

{
  "$match": {
    "location": {
      "$geoWithin": {
        "$box": [
          [
            0.024719919885622943,
            51.54643953472475
          ],
          [
            -0.1589577534542408,
            51.47239969138267
          ]
        ]
      }
    }
  }
}

Теперь у меня есть координаты, подобные массиву [-0.0649729793707321, 51.50160291888072]

Мне нужно расстояние для каждой записи от указанной выше точки. Как я могу найти это поле расстояния с агрегацией MongoDB?

образец документа для записи в базе данных 

{
    "_id" : ObjectId("5e342ac2e0dcfa65273751ea"),
    "location" : {
        "type" : "Point",
        "coordinates" : [ 
            -0.123507, 
            51.5083228
        ]
    },
    "status" : 1,
    "price" : {
        "value" : 27,
        "type" : "day",
        "base" : 3.38
    },
    "name" : "Initial Space with Wormup",
    "venuePolicy" : "Venue Policies",
    "updatedAt" : ISODate("2020-02-20T12:28:16.024Z"),
    "createdAt" : ISODate("2020-01-31T13:25:22.600Z"),
    "dimension" : {
        "width" : 100
    },
    "description" : "<p>Hi,&nbsp;</p><p>please find attached document for testing purpose",
}

1 Ответ

1 голос
/ 27 февраля 2020

Это должно работать: 

db.collection.aggregate([
   {
      "$match": {
         "location": {
            "$geoWithin": {
               "$box": [
                  [0.024719919885622943, 51.54643953472475],
                  [-0.1589577534542408, 51.47239969138267]
               ]
            }
         }
      }
   },
   { $set: { lon: -0.0649729793707321 } },
   { $set: { lat: 51.50160291888072 } },
   {
      $set: {
         distance: {
            $let: {
               vars: {
                  dlon: { $degreesToRadians: { $subtract: [{ $arrayElemAt: ["$location.coordinates", 0] }, "$lon"] } },
                  dlat: { $degreesToRadians: { $subtract: [{ $arrayElemAt: ["$location.coordinates", 1] }, "$lat"] } },
                  lat1: { $degreesToRadians: { $arrayElemAt: ["$location.coordinates", 1] } },
                  lat2: { $degreesToRadians: "$lat" }
               },
               in: {
                  // Haversine formula: sin²(dLat / 2) + sin²(dLon / 2) * cos(lat1) * cos(lat2);
                  $add: [
                     { $pow: [{ $sin: { $divide: ["$$dlat", 2] } }, 2] },
                     { $multiply: [{ $pow: [{ $sin: { $divide: ["$$dlon", 2] } }, 2] }, { $cos: "$$lat1" }, { $cos: "$$lat2" }] }
                  ]
               }
            }
         }
      }
   },
   {
      $set: {
         distance: {
            // Distance in Meters given by "6372.8 * 1000"
            $multiply: [6372.8, 1000, 2, { $asin: { $sqrt: "$distance" } }]
         }
      }
   },
])

Просто примечание, $box используется только для Устаревшие пары координат . Для правильного запроса (т. Е. Правильного использования 2dsphere index) вы должны использовать $geometry: 

db.collection.createIndex({ location: "2dsphere" })
db.collection.find(
   {
      "location": {
         "$geoWithin": {
            "$geometry": {
               type: "Polygon",
               coordinates: [[
                  [-0.1589577534542408, 51.47239969138267],
                  [-0.1589577534542408, 51.54643953472475],
                  [0.024719919885622943, 51.54643953472475],
                  [0.024719919885622943, 51.47239969138267],
                  [-0.1589577534542408, 51.47239969138267]
               ]]
            }
         }
      }
   }
).explain().queryPlanner.winningPlan.inputStage.indexName

--> location_2dsphere // which means index is used


db.collection.find(
   {
      "location": {
         "$geoWithin": {
            "$box": [
               [0.024719919885622943, 51.54643953472475],
               [-0.1589577534542408, 51.47239969138267]
            ]
         }
      }
   }
).explain().queryPlanner.winningPlan.stage

--> COLLSCAN // which means full collection scan

или правильное использование устаревших координат: 

db.collection.createIndex({ "location.coordinates": "2d" })
db.collection.find(
   {
      "location.coordinates": { // <- note this difference
         "$geoWithin": {
            "$box": [
               [0.024719919885622943, 51.54643953472475],
               [-0.1589577534542408, 51.47239969138267]
            ]
         }
      }
   }
).explain().queryPlanner.winningPlan.inputStage.indexName

--> location.coordinates_2d
...