20 lines
499 B
Python
20 lines
499 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from fastapi import WebSocket, WebSocketDisconnect
|
||
|
|
|
||
|
|
from backend.services.broadcaster import Broadcaster
|
||
|
|
|
||
|
|
|
||
|
|
async def websocket_handler(ws: WebSocket, broadcaster: Broadcaster) -> None:
|
||
|
|
await ws.accept()
|
||
|
|
await broadcaster.add(ws)
|
||
|
|
try:
|
||
|
|
while True:
|
||
|
|
await ws.receive_text()
|
||
|
|
except WebSocketDisconnect:
|
||
|
|
# 客户端正常断开连接，无需打印堆栈信息
|
||
|
|
pass
|
||
|
|
finally:
|
||
|
|
await broadcaster.remove(ws)
|
||
|
|
|