초기 커밋

app/db/mongodb.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from beanie import init_beanie
+from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
+
+from app.core.config import settings
+
+mongo_client: AsyncIOMotorClient | None = None
+mongo_db: AsyncIOMotorDatabase | None = None
+
+
+async def init_mongodb() -> None:
+    global mongo_client, mongo_db
+
+    mongo_client = AsyncIOMotorClient(settings.MONGODB_URL)
+    mongo_db = mongo_client[settings.MONGODB_DATABASE]
+
+    from app.models.mongodb.analytics_result import AnalyticsResult
+    from app.models.mongodb.device_log import DeviceLog
+    from app.models.mongodb.notification import Notification
+    from app.models.mongodb.telemetry import TelemetryData
+
+    await init_beanie(
+        database=mongo_db,
+        document_models=[DeviceLog, TelemetryData, AnalyticsResult, Notification],
+    )
+
+
+async def close_mongodb() -> None:
+    global mongo_client
+    if mongo_client:
+        mongo_client.close()
+
+
+def get_mongo_db() -> AsyncIOMotorDatabase:
+    assert mongo_db is not None, "MongoDB not initialized"
+    return mongo_db