fix: Server is busy (#3878) (#3895)

This commit is contained in:
Jan Cheng 2024-09-24 21:49:30 +08:00 committed by GitHub
commit a4ae2b52a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -324,7 +324,7 @@ async def build_flow(
client_consumed_queue: asyncio.Queue,
event_manager: "EventManager",
) -> None:
build_task = asyncio.create_task(await asyncio.to_thread(_build_vertex, vertex_id, graph, event_manager))
build_task = asyncio.create_task(asyncio.to_thread(asyncio.run, _build_vertex(vertex_id, graph, event_manager)))
try:
await build_task
except asyncio.CancelledError as exc:
@ -359,7 +359,7 @@ async def build_flow(
async def event_generator(event_manager: EventManager, client_consumed_queue: asyncio.Queue) -> None:
if not data:
# using another thread since the DB query is I/O bound
vertices_task = asyncio.create_task(await asyncio.to_thread(build_graph_and_get_order))
vertices_task = asyncio.create_task(asyncio.to_thread(asyncio.run, build_graph_and_get_order()))
try:
await vertices_task
except asyncio.CancelledError: