app/database.py

@@ -3,9 +3,13 @@ from sqlmodel.ext.asyncio.session import AsyncSession
 from sqlalchemy.ext.asyncio import create_async_engine
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy import text
+from sqlalchemy.exc import DBAPIError, OperationalError
 import os
+import logging
 from dotenv import load_dotenv
 
+logger = logging.getLogger(__name__)
+
 load_dotenv()
 
 DB_USER = os.getenv("DB_USER")
@@ -21,7 +25,10 @@ engine = create_async_engine(
     echo=True,
     future=True,
     pool_pre_ping=True,
-    pool_recycle=300,
+    pool_recycle=180,
+    pool_size=5,
+    max_overflow=10,
+    pool_timeout=30,
 )
 
 async def init_db():
@@ -55,9 +62,21 @@ async def init_db():
             except Exception:
                 pass  # Already applied
 
-async def get_session() -> AsyncSession:
-    async_session = sessionmaker(
+_async_session_factory = sessionmaker(
     engine, class_=AsyncSession, expire_on_commit=False
-    )
-    async with async_session() as session:
+)
+
+_MAX_RETRIES = 2
+
+async def get_session() -> AsyncSession:
+    for attempt in range(_MAX_RETRIES):
+        try:
+            async with _async_session_factory() as session:
                 yield session
+                return
+        except (DBAPIError, OperationalError, ConnectionResetError, OSError) as e:
+            if attempt < _MAX_RETRIES - 1:
+                logger.warning("Database connection error (attempt %d/%d): %s", attempt + 1, _MAX_RETRIES, e)
+                await engine.dispose()
+                continue
+            raise