feat: add DB connection pooling with configurable pool_size, overflow, recycle

2 files changed, 20 insertions, 2 deletions

diff --git a/shared/src/shared/config.py b/shared/src/shared/config.py
index b6ccebd..b6b9d69 100644
--- a/shared/src/shared/config.py
+++ b/shared/src/shared/config.py
@@ -9,6 +9,9 @@ class Settings(BaseSettings):
     alpaca_paper: bool = True  # Use paper trading by default
     redis_url: str = "redis://localhost:6379"
     database_url: str = "postgresql://trading:trading@localhost:5432/trading"
+    db_pool_size: int = 20
+    db_max_overflow: int = 10
+    db_pool_recycle: int = 3600
     log_level: str = "INFO"
     risk_max_position_size: float = 0.1
     risk_stop_loss_pct: float = 5.0
diff --git a/shared/src/shared/db.py b/shared/src/shared/db.py
index 9cc8686..e7cad92 100644
--- a/shared/src/shared/db.py
+++ b/shared/src/shared/db.py
@@ -36,9 +36,24 @@ class Database:
         self._engine = None
         self._session_factory = None
 
-    async def connect(self) -> None:
+    async def connect(
+        self,
+        pool_size: int = 20,
+        max_overflow: int = 10,
+        pool_recycle: int = 3600,
+    ) -> None:
         """Create the async engine, session factory, and all tables."""
-        self._engine = create_async_engine(self._database_url)
+        if self._database_url.startswith("sqlite"):
+            # SQLite doesn't support pooling options
+            self._engine = create_async_engine(self._database_url)
+        else:
+            self._engine = create_async_engine(
+                self._database_url,
+                pool_pre_ping=True,
+                pool_size=pool_size,
+                max_overflow=max_overflow,
+                pool_recycle=pool_recycle,
+            )
         self._session_factory = async_sessionmaker(self._engine, expire_on_commit=False)
         async with self._engine.begin() as conn:
             await conn.run_sync(Base.metadata.create_all)