요즘 MongoDB와 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,
		}
	}
})