<think>我们将使用Python的websockets库来创建一个简单的WebSocket服务器，并使用aiohttp来创建一个客户端。 首先，确保安装了websockets库：pip install websockets 下面是一个简单的WebSocket服务器和客户端的示例。 服务器端代码（server.py）： 该服务器将监听本地主机的8765端口。当有客户端连接时，它会等待接收消息，然后回复相同的消息（echo）。 客户端代码（client.py）： 客户端将连接到服务器，发送一条消息，然后接收并打印服务器的回复。 注意：由于WebSocket是异步的，我们使用asyncio库来处理异步操作。 我们先写服务器端： </think> 以下是一个完整的WebSocket案例，包含服务器端和客户端代码，使用Python的`websockets`库实现： ### 服务器端代码 (websocket_server.py) ```python import asyncio import websockets # 存储所有连接的客户端 connected_clients = set() async def handle_client(websocket, path): # 添加新客户端 connected_clients.add(websocket) client_ip = websocket.remote_address[0] print(f"客户端 {client_ip} 已连接，当前连接数: {len(connected_clients)}") try: # 持续接收消息 async for message in websocket: print(f"收到来自 {client_ip} 的消息: {message}") # 广播消息给所有客户端 broadcast_message = f"用户 {client_ip} 说: {message}" await broadcast(broadcast_message) except websockets.exceptions.ConnectionClosed: print(f"客户端 {client_ip} 断开连接") finally: # 移除断开连接的客户端 connected_clients.remove(websocket) print(f"客户端 {client_ip} 已移除，剩余连接: {len(connected_clients)}") async def broadcast(message): """向所有连接的客户端广播消息""" if connected_clients: await asyncio.gather( *[client.send(message) for client in connected_clients] ) async def main(): # 启动WebSocket服务器 server = await websockets.serve( handle_client, host="localhost", port=8765 ) print("WebSocket 服务器已启动，监听 ws://localhost:8765") await server.wait_closed() if __name__ == "__main__": asyncio.run(main()) ``` ### 客户端代码 (websocket_client.py) ```python import asyncio import websockets async def client(): uri = "ws://localhost:8765" async with websockets.connect(uri) as websocket: name = input("请输入你的名字: ") # 持续发送和接收消息 while True: message = input("发送消息 (输入 'exit' 退出): ") if message.lower() == 'exit': break await websocket.send(f"{name}: {message}") response = await websocket.recv() print(f"收到广播: {response}") if __name__ == "__main__": asyncio.run(client()) ``` ### 使用说明 1. **安装依赖**： ```bash pip install websockets ``` 2. **启动服务器**： ```bash python websocket_server.py ``` 3. **启动多个客户端**（新终端窗口）： ```bash python websocket_client.py ``` ### 功能说明 - **服务器端**： - 监听本地8765端口 - 维护所有连接的客户端集合 - 广播所有收到的消息给所有客户端 - 自动处理客户端连接/断开 - **客户端**： - 连接服务器并输入名称 - 发送消息到服务器 - 实时接收广播消息 ### 原理解释 1. **WebSocket协议**： - 基于TCP的全双工通信协议（RFC 6455） - 在HTTP握手后升级为持久连接 - 支持服务端主动推送数据 2. **核心组件**： - `websockets.serve()`：创建WebSocket服务器 - `async with websockets.connect()`：客户端连接 - `await websocket.send()` / `await websocket.recv()`：异步发送/接收数据 3. **广播机制**： - 服务器维护所有客户端的WebSocket对象集合 - 收到消息时遍历所有客户端并发送消息 - 使用`asyncio.gather()`实现并发广播