fastapi websocket 如何捕捉客户端关闭事件

在 FastAPI 中，您可以使用 Starlette 的 WebSocket 类来处理 WebSocket 连接。要捕获客户端关闭事件，您可以使用 `WebSocket.disconnect()` 方法。当 WebSocket 连接关闭时，这个方法会被调用。您可以在这个方法中编写代码来处理关闭事件。

以下是一个示例代码：

from fastapi import FastAPI, WebSocket

app = FastAPI()

class ConnectionManager:
    def __init__(self):
        self.active_connections = []

    async def connect(self, websocket: WebSocket):
        await websocket.accept()
        self.active_connections.append(websocket)

    def disconnect(self, websocket: WebSocket):
        self.active_connections.remove(websocket)

manager = ConnectionManager()

@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await manager.connect(websocket)
    try:
        while True:
            data = await websocket.receive_text()
            # 处理接收到的数据
    except WebSocketDisconnect:
        manager.disconnect(websocket)
        # 处理客户端关闭事件

在上面的示例代码中，我们创建了一个 `ConnectionManager` 类来管理所有的 WebSocket 连接。在 `websocket_endpoint` 函数中，我们使用 `manager.connect()` 方法来添加新的 WebSocket 连接。在 `try` 块中，我们等待接收客户端发送的数据。如果客户端关闭了连接，`WebSocketDisconnect` 异常会被抛出，我们可以在 `except` 块中使用 `manager.disconnect()` 方法来移除关闭的 WebSocket 连接。这样，我们就可以捕获客户端关闭事件了。