Geospatial queries with MongoDB

2019-07-25

使用 MongoDB 进行地理空间查询

我目前正在体验 MongoDB 及其geospatial queries(地理空间查询)。目前来看挺有意思的，所以我想写点东西来展示我是如何通过 PyMongo 使用它的！

首先，显然需要执行 pip install pymongo。我们首先需要一个 Mongo 客户端，这很简单

from pymongo import MongoClient

client = MongoClient()

这很简洁——默认情况下，MongoClient 会连接到 mongodb://localhost:27017。

不过，你可能更想改成这样做

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

我们需要在文档字段上创建一个 MongoDB 索引，才能让地理空间的魔法生效。我们将创建一个 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,
		}
	}
})