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

在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」關鍵字參數進一步自訂連線池大小。

以上是使用 Uvicorn 在 FastAPI 中發出並發 HTTP 請求時如何避免「ConnectionClosed」錯誤？的詳細內容。更多資訊請關注PHP中文網其他相關文章！