FastAPI中怎么使用WebSocket

在 FastAPI 中使用 WebSocket 非常简单。首先需要导入 WebSocket 类和 WebSocketDisconnect 异常类，然后在路由函数中添加一个 WebSocket 参数来处理 WebSocket 连接。

以下是一个简单的例子：

from fastapi import FastAPI, WebSocket

app = FastAPI()

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        await websocket.send_text(f"Message text was: {data}")

在这个例子中，我们创建了一个 WebSocket 端点 /ws，当有客户端连接时，会调用 websocket_endpoint 函数处理连接。在函数中我们首先调用 await websocket.accept() 来接受连接，并开始一个无限循环来接收和发送消息。

当客户端发送消息时，我们通过 await websocket.receive_text() 方法来接收消息，并通过 await websocket.send_text() 方法来发送消息给客户端。

在 FastAPI 中使用 WebSocket 很容易，你可以根据自己的需求来处理 WebSocket 连接和消息。