mirror of
https://github.com/joaovitoriasilva/endurain.git
synced 2026-05-03 03:00:41 -04:00
1d7e0262a1
Attempt fix for #393 - Replaces manual creation and closing of SQLAlchemy database sessions with context manager usage (with SessionLocal() as db) across multiple modules. This improves resource management and reduces boilerplate, ensuring sessions are properly closed even in case of exceptions. Also doubles DB connection pool size.
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import os
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker, declarative_base
|
|
from sqlalchemy.engine.url import URL
|
|
|
|
import core.config as core_config
|
|
|
|
# Define the database connection URL using environment variables
|
|
db_url = URL.create(
|
|
drivername="postgresql+psycopg",
|
|
username=os.environ.get("DB_USER", "endurain"),
|
|
password=core_config.read_secret("DB_PASSWORD"),
|
|
host=os.environ.get("DB_HOST", "postgres"),
|
|
port=os.environ.get("DB_PORT", "5432"),
|
|
database=os.environ.get("DB_DATABASE", "endurain"),
|
|
)
|
|
|
|
# Create the SQLAlchemy engine
|
|
engine = create_engine(
|
|
db_url,
|
|
pool_size=20,
|
|
max_overflow=40,
|
|
pool_timeout=180,
|
|
pool_recycle=3600,
|
|
pool_pre_ping=True,
|
|
)
|
|
|
|
# Create a session factory
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
# Create a base class for declarative models
|
|
Base = declarative_base()
|
|
|
|
|
|
def get_db():
|
|
"""
|
|
Yields a new SQLAlchemy database session.
|
|
|
|
This generator function creates a new database session using SessionLocal,
|
|
yields it for use in database operations, and ensures the session is properly
|
|
closed after use. Intended for use as a dependency in FastAPI routes or other
|
|
contexts where session management is required.
|
|
|
|
Yields:
|
|
Session: An active SQLAlchemy database session.
|
|
|
|
Example:
|
|
with get_db() as db:
|
|
# use db session here
|
|
"""
|
|
# Create a new database session and return it
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# Yield the database session
|
|
yield db
|
|
finally:
|
|
# Close the database session
|
|
db.close()
|