Tweak workers

main
Thomas Sileo 2022-08-14 22:44:30 +02:00
parent 7d9ced7740
commit 6c3fe894b5
2 changed files with 6 additions and 3 deletions

View File

@ -12,12 +12,14 @@ from app.config import DB_PATH
from app.config import SQLALCHEMY_DATABASE_URL
engine = create_engine(
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False, "timeout": 15}
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
DATABASE_URL = f"sqlite+aiosqlite:///{DB_PATH}"
async_engine = create_async_engine(DATABASE_URL, future=True, echo=False)
async_engine = create_async_engine(
DATABASE_URL, future=True, echo=False, connect_args={"timeout": 15}
)
async_session = sessionmaker(async_engine, class_=AsyncSession, expire_on_commit=False)
Base: Any = declarative_base()

View File

@ -30,8 +30,9 @@ class Worker(Generic[T]):
next_message = await self.get_next_message(db_session)
if next_message:
await self.process_message(db_session, next_message)
await asyncio.sleep(0.5)
else:
await asyncio.sleep(1)
await asyncio.sleep(2)
async def _until_stopped(self) -> None:
await self._stop_event.wait()