31 lines
971 B
Python
31 lines
971 B
Python
import logging
|
|
from common.config.app_settings import app_settings
|
|
from beanie import init_beanie
|
|
from motor.motor_asyncio import AsyncIOMotorClient
|
|
from backend.models import backend_models
|
|
|
|
|
|
def register(app):
|
|
app.debug = "auth_mongo_debug"
|
|
app.title = "auth_mongo_name"
|
|
|
|
# Configure logging for pymongo
|
|
logging.getLogger("pymongo").setLevel(logging.WARNING) # Suppress DEBUG logs
|
|
|
|
@app.on_event("startup")
|
|
async def start_database():
|
|
await initiate_database()
|
|
|
|
|
|
async def initiate_database():
|
|
client = AsyncIOMotorClient(
|
|
app_settings.MONGODB_URI,
|
|
serverSelectionTimeoutMS=60000,
|
|
minPoolSize=5, # Minimum number of connections in the pool
|
|
maxPoolSize=20, # Maximum number of connections in the pool
|
|
heartbeatFrequencyMS=20000, # Adjust heartbeat frequency to 20 seconds
|
|
)
|
|
await init_beanie(
|
|
database=client[app_settings.MONGODB_NAME], document_models=backend_models
|
|
)
|