在Uvicorn/FastAPI 中發出HTTP 請求
處理使用FastAPI 和Uvicorn 建構的HTTP 端點時,通常會從外部要求資料。但是,在處理多個並發請求時,可能會出現「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