This commit is contained in:
2026-02-22 00:05:04 +08:00
parent 6ad5256df9
commit ed8ba1ae7d
5 changed files with 54 additions and 40 deletions

View File

@@ -1,9 +1,11 @@
from sqlmodel import Session, create_engine
from sqlmodel.ext.asyncio.session import AsyncSession
from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
from app.core.config import settings
connect_args = {"check_same_thread": False} # SQLite-specific argument for multithreading
engine = create_engine(settings.database_url, echo=True, connect_args=connect_args)
engine = create_async_engine(settings.database_url, echo=True, future=True)
def get_session():
with Session(engine) as session:
AsyncSessionLocal = async_sessionmaker(bind=engine, class_=AsyncSession, expire_on_commit=False)
async def get_session():
async with AsyncSessionLocal() as session:
yield session