这是因为在Uvicorn和Starlette中,当一个HTTP请求被取消时,它会中断正在进行的所有协程。因此,在使用asyncio.sleep()等待时,如果请求被取消,协程将被中断,从而导致asyncio.sleep()立即停止并抛出CancelledError。
要解决这个问题,可以将asyncio.sleep()包装在try/except块中,以捕获CancelledError并返回。以下示例演示了如何包装asyncio.sleep()以使其在Uvicorn和Starlette中正常工作。
import asyncio
from starlette.applications import Starlette
from starlette.responses import JSONResponse
async def long_running_task():
try:
await asyncio.sleep(10)
return {"status": "done"}
except asyncio.CancelledError:
return {"status": "cancelled"}
app = Starlette()
@app.route("/")
async def homepage(request):
task = asyncio.ensure_future(long_running_task())
while not task.done():
await asyncio.sleep(0)
return JSONResponse(task.result())
在上面的示例中,我们将asyncio.sleep()包装在try/except块中以捕获CancelledError,并返回“cancelled”状态。这将确保即使请求被取消,也不会中断协程并导致asyncio.sleep()被取消。