有部分入坑fastapi的同志们,可能经常会遇到一个耗时的密集型IO任务阻塞主线程的情况,导致接口单个接口占用整体的线程时间过长,从而影响到其他的接口响应。
很多人可能会说,fastapi不是一个异步的框架吗?为什么会存在阻塞这个问题?这是一个很好的问题,我之前也想过这个问题,但是后面才发现,原来是fastapi框架的接口,本身不会自动帮忙处理长耗时的任务,如生成式的请求,调用复杂的第三方api,这个过程大概需要10-20s。
假如说自己使用的http请求是直接requests库的请求,那么就会导致阻塞。
正确的做法应该是
使用一个异步执行的请求方式,例如aiohttp库,手动为请求添加上这个异步的请求方法,我就是使用这个库解决了长耗时任务阻塞线程的问题。
首先得安装好这个库:
pip install aiohttp
然后就再代码当中
from fastapi import FastAPI
import aiohttp
import asyncio
app = FastAPI()
async def long_running_task(url: str):
async with aiohttp.ClientSession() as session:
async with session.post(url, json={"load": "data"}) as response:
# 处理响应数据
data = await response.json()
return data
@app.post("/start_task/")
async def start_task():
url = "http://example.com/api" # 目标 API 的 URL
task_result = await long_running_task(url)
return task_result
这样,这个长任务就不会阻塞主线程了。