요즘 MongoDB와 지리 공간 쿼리를 가지고 이것저것 만져보고 있습니다. 지금까지 꽤 흥미로워서 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 형식이어야 합니다. GeoJSON은 지리 정보를 표현하기 위한 JSON 표준으로, 덕분에 문서를 쿼리하기가 정말 쉬워집니다!

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