20、实现多 WebSocket 连接与消息广播及异步 API 测试

实现多 WebSocket 连接与消息广播及异步 API 测试

多 WebSocket 连接与消息广播

WebSocket 常用于实现多客户端间的实时通信，如聊天应用。在这种场景下，多个客户端与服务器建立 WebSocket 隧道，服务器负责管理客户端连接并广播消息。

简单实现的问题

若简单地维护一个 WebSocket 连接列表并遍历广播消息，在生产环境中会很快出现问题。因为服务器部署时通常会运行多个工作进程，无法保证两个打开 WebSocket 的客户端由同一进程服务。由于连接存储在进程内存中，接收消息的进程无法将消息广播给其他进程服务的客户端。

解决方案：消息代理

为解决此问题，通常使用消息代理。消息代理接收一个程序发布的消息，并将其广播给订阅该消息的程序。常见的消息代理软件有 Apache Kafka、RabbitMQ 等。

在架构中，消息代理是唯一的，多个服务器进程连接到它进行消息的发布和订阅。

使用 broadcaster 库和 Redis 实现

• 安装库 ：

$ pip install "broadcaster[redis]"

• 代码实现 ：

# app.py
from broadcaster import Broadcast
from