Geospatial queries with MongoDB

2019-07-25

使用 MongoDB 進行地理空間查詢

我最近正在把玩 MongoDB 及其 geospatial queries(地理空間查詢)。到目前為止還蠻有趣的，所以我想寫篇文章來分享我是如何搭配 PyMongo 使用它的！

首先，當然需要執行 pip install pymongo。我們首先需要一個 mongo client，這非常簡單

from pymongo import MongoClient

client = MongoClient()

這還蠻方便的——預設情況下，MongoClient 會連線到 mongodb://localhost:27017。

不過，你可能會想改成這樣做

url = os.environ.get("MONGO_URL", "mongodb://localhost:27017")
client = MongoClient(url)

我們會需要在文件欄位上建立 MongoDB 索引，才能讓 geospatial 的魔法生效。我們將建立一個 2dsphere 索引。

from pymongo import MongoClient, GEOSPHERE, DESCENDING

client = MongoClient()
db = client.foo
db.bar.create_index([("location", GEOSPHERE)])

現在只需要插入一份文件並執行查詢就可以了 :) 我們將插入一份只有 location 欄位的文件

db.bar.insert_one({
	“location”: {
		"coordinates" : [ 
			51.4982563,
			-0.0861183
    		],
    		"type" : "Point"
	}
})

location 欄位必須是 GeoJSON，這是一種用於表示地理特徵的 JSON 標準。這讓查詢文件變得超級簡單！

db.bar.find({
	"location": {
		"$near": {
			"$geometry": {"type": "Point", "coordinates": [LAT, LONG]},
			"$maxDistance": range,
		}
	}
})