在 Uvicorn/FastAPI 中发出 HTTP 请求
处理使用 FastAPI 和 Uvicorn 构建的 HTTP 端点时,通常会从外部 API 请求数据。但是,在处理多个并发请求时,可能会出现“can't handle event type ConnectionClosed when role=SERVER and state=SEND_RESPONSE”等错误。发生这种情况是因为默认的 HTTP 客户端库“requests”在并发环境中使用时不是完全线程安全的。
要解决此问题,请考虑实现名为 httpx 的替代 HTTP 客户端库。它提供了一个异步 API,有助于避免 FastAPI 中的线程安全问题。下面的例子展示了如何在FastAPI中使用httpx:
from fastapi import FastAPI, Request, BackgroundTask from fastapi.responses import StreamingResponse, Response from contextlib import asynccontextmanager import httpx @asynccontextmanager async def lifespan(app: FastAPI): async with httpx.AsyncClient() as client: yield {'client': client} app = FastAPI(lifespan=lifespan) @app.get('/') async def home(request: Request): client = request.state.client req = client.build_request('GET', 'https://www.example.com') r = await client.send(req, stream=True) return StreamingResponse(r.aiter_raw(), background=BackgroundTask(r.aclose()))
通过利用httpx的异步API,您可以在FastAPI中更有效地处理HTTP请求,同时保持线程安全。您可以使用客户端对象上的“limits”关键字参数进一步自定义连接池大小。
免责声明: 提供的所有资源部分来自互联网,如果有侵犯您的版权或其他权益,请说明详细缘由并提供版权或权益证明然后发到邮箱:[email protected] 我们会第一时间内为您处理。
Copyright© 2022 湘ICP备2022001581号-3