袁聪的第一次提交,包含nl2sql,行情数据,场外申购
This commit is contained in:
@@ -4,20 +4,24 @@ import logging
|
||||
|
||||
from app.core.config import get_settings
|
||||
from app.infrastructure.db import engine
|
||||
from app.worker.offsite_mail_worker import OffsiteMailWorker
|
||||
from app.worker.runtime import WorkerRuntime
|
||||
|
||||
|
||||
async def serve(*, once: bool = False) -> None:
|
||||
settings = get_settings()
|
||||
runtime = WorkerRuntime(settings=settings)
|
||||
offsite_worker = OffsiteMailWorker(settings)
|
||||
try:
|
||||
while True:
|
||||
worked = await runtime.run_once()
|
||||
worked = await offsite_worker.run_once() or worked
|
||||
if once:
|
||||
return
|
||||
if not worked:
|
||||
await asyncio.sleep(settings.worker_poll_seconds)
|
||||
finally:
|
||||
await offsite_worker.close()
|
||||
await engine.dispose()
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user