对 Python Websockets 库全方位详解

news2025/4/9 5:23:37

一、WebSocket 简介

WebSocket 是一种基于 TCP 的协议，支持全双工通信（服务器和客户端可以同时发送消息），适用于实时性要求高的场景（如聊天、实时数据监控、在线游戏等）。与 HTTP 不同，WebSocket 连接一旦建立，会保持持久化，避免了 HTTP 轮询的开销。

二、Python 的 `websockets` 库

websockets 是 Python 中用于构建 WebSocket 服务器和客户端的异步库，基于 asyncio 实现，支持高并发和低延迟通信。

安装

pip install websockets

三、完整代码示例

1. WebSocket 服务器

import asyncio
import websockets
from datetime import datetime

# 保存所有活跃的客户端连接
connected_clients = set()

async def handle_client(websocket, path):
    """
    处理单个客户端连接
    """
    # 将新客户端添加到集合
    connected_clients.add(websocket)
    try:
        async for message in websocket:
            # 广播消息给所有客户端
            timestamp = datetime.now().strftime("%H:%M:%S")
            broadcast_message = f"[{timestamp}] {message}"
            await broadcast(broadcast_message)
    except websockets.exceptions.ConnectionClosed:
        print("客户端断开连接")
    finally:
        # 客户端断开后移除
        connected_clients.remove(websocket)

async def broadcast(message):
    """
    向所有连接的客户端广播消息
    """
    if connected_clients:
        await asyncio.gather(
            *[client.send(message) for client in connected_clients]
        )

async def start_server():
    """
    启动 WebSocket 服务器
    """
    async with websockets.serve(handle_client, "localhost", 8765):
        print("服务器已启动，监听端口 8765...")
        await asyncio.Future()  # 永久运行

if __name__ == "__main__":
    asyncio.run(start_server())

2. WebSocket 客户端

import asyncio
import websockets

async def client():
    """
    WebSocket 客户端实现
    """
    async with websockets.connect("ws://localhost:8765") as websocket:
        # 创建两个任务：一个接收消息，一个发送消息
        receive_task = asyncio.create_task(receive_messages(websocket))
        send_task = asyncio.create_task(send_messages(websocket))
        await asyncio.gather(receive_task, send_task)

async def receive_messages(websocket):
    """
    接收服务器消息
    """
    async for message in websocket:
        print(f"\n收到消息: {message}")

async def send_messages(websocket):
    """
    发送用户输入的消息
    """
    while True:
        message = await asyncio.get_event_loop().run_in_executor(None, input, "输入消息 (输入 'exit' 退出): ")
        if message.strip().lower() == 'exit':
            break
        await websocket.send(message)

if __name__ == "__main__":
    asyncio.run(client())

四、代码详解

服务器端

connected_clients：使用集合保存所有活跃连接，便于广播。
handle_client：
- 接受新连接并添加到集合。
- 循环接收客户端消息，收到后调用 broadcast 广播。
- 处理连接断开异常，确保客户端被移除。
broadcast：使用 asyncio.gather 并发向所有客户端发送消息。
start_server：启动服务器并永久运行。