等待由同步代码设置的 Python 异步事件 - 使用 FastAPI

Question

我有一个运行同步后台任务的异步 web 服务（使用 FastAPI）。

开始代码的想法是：

FastAPI 异步代码接收任务请求。

此请求有一个事件标志

它将请求添加到优先队列，然后等待，直到设置了任务的事件或达到超时。

同步后台线程工作者完成队列中的所有任务并设置事件。

如何以异步代码可以可靠地等待它的方式传递事件？

代码示例：

import asyncio

from fastapi import FastAPI
import time
import uvicorn
from random import randint

from threading import Thread
import asyncio
from asyncio.locks import Event

from queue import PriorityQueue, Empty
global_pq = PriorityQueue()
global_task_count = 0

def background_thread():
    while True:
        time.sleep(0.5)
        try:
            task: MyTask
            task = global_pq.get(block=None)
            print(f'background {time.time()} {task}')
            task.complete()
        except Empty:
            print(f'background {time.time()} {None}')


bg = Thread(target=background_thread)
bg.setDaemon(True)
bg.start()

class MyTask:
    def __init__(self, no):
        self.event = Event()
        self.priority = randint(1, 5)
        self.no = no

    def __lt__(self, other):
        return self.priority < other.priority

    def __repr__(self):
        return f'Task No {self.no}, Priority {self.priority}'

    def complete(self):
        self.event._loop.call_soon_threadsafe(self.event.set)
        # self.event.set(True)

app = FastAPI()


@app.get("/")
async def root():

    global global_task_count
    global_task_count += 1
    mytask = MyTask(no=global_task_count)
    global_pq.put((mytask.priority, mytask))
    with asyncio.wait_for(mytask.event.wait(), timeout=5.0):
        pass
    return {"count": global_task_count}


if __name__ == '__main__':
    uvicorn.run(app, host='0.0.0.0', port=9000)