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,
		}
	}
})