36 lines
1001 B
Python
36 lines
1001 B
Python
# utils/database.py
|
|
import logging
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import QueuePool
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_db_engine(database_url: str):
|
|
"""Create database engine with connection pooling"""
|
|
if database_url.startswith("sqlite"):
|
|
# SQLite configuration
|
|
engine = create_engine(
|
|
database_url, connect_args={"check_same_thread": False}, echo=False
|
|
)
|
|
else:
|
|
# PostgreSQL configuration with connection pooling
|
|
engine = create_engine(
|
|
database_url,
|
|
poolclass=QueuePool,
|
|
pool_size=10,
|
|
max_overflow=20,
|
|
pool_pre_ping=True,
|
|
echo=False,
|
|
)
|
|
|
|
logger.info(f"Database engine created for: {database_url.split('@')[0]}@...")
|
|
return engine
|
|
|
|
|
|
def get_session_local(engine):
|
|
"""Create session factory"""
|
|
return sessionmaker(autocommit=False, autoflush=False, bind=engine)
|