Spaces:
Sleeping
Sleeping
| from sqlalchemy import create_engine, event | |
| from sqlalchemy.orm import sessionmaker | |
| from ..config import settings | |
| from .models import Base | |
| _connect_args = {"check_same_thread": False} if settings.db_url.startswith("sqlite") else {} | |
| engine = create_engine( | |
| settings.db_url, | |
| connect_args=_connect_args, | |
| echo=False, | |
| future=True, | |
| pool_pre_ping=True, | |
| ) | |
| if settings.db_url.startswith("sqlite"): | |
| def _tune_sqlite(dbapi_connection, _): | |
| """ | |
| Apply SQLite tuning pragmas on every new connection: | |
| * WAL journaling — better concurrent read/write, dramatically reduces | |
| ``database is locked`` errors when multiple requests hit history. | |
| * synchronous=NORMAL — safe with WAL, ~2x faster commits than FULL. | |
| * foreign_keys=ON — needed for our ON DELETE CASCADE relationships. | |
| * temp_store=MEMORY — temp B-trees in RAM, faster ordering / grouping. | |
| * cache_size=-20000 — 20 MB page cache (negative = KiB), big help for | |
| history pagination workload. | |
| """ | |
| cursor = dbapi_connection.cursor() | |
| cursor.execute("PRAGMA journal_mode=WAL") | |
| cursor.execute("PRAGMA synchronous=NORMAL") | |
| cursor.execute("PRAGMA foreign_keys=ON") | |
| cursor.execute("PRAGMA temp_store=MEMORY") | |
| cursor.execute("PRAGMA cache_size=-20000") | |
| cursor.close() | |
| SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False, expire_on_commit=False) | |
| def init_db() -> None: | |
| Base.metadata.create_all(engine) | |