」工欲善其事,必先利其器。「—孔子《論語.錄靈公》
首頁 > 程式設計 > 使用 Uvicorn 在 FastAPI 中發出並發 HTTP 請求時如何避免「ConnectionClosed」錯誤?

使用 Uvicorn 在 FastAPI 中發出並發 HTTP 請求時如何避免「ConnectionClosed」錯誤?

發佈於2024-12-21
瀏覽:928

How Can I Avoid

在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