使用 Python 实现 WebSocket 服务器与客户端通信

简介

WebSocket 是一种基于 TCP 协议的通信协议，能够在客户端与服务器之间进行全双工（双向）通信。相比传统的 HTTP 协议，WebSocket 可以实现实时数据的传输，尤其适合需要实时交互的应用场景，如在线游戏、实时聊天、金融交易等。

我通过 Python 实现一个简单的 WebSocket 服务器，并使其与客户端进行通信。我们将创建两个 Python 文件：websocket.py 和 main.py，websocket.py 负责实现 WebSocket 服务器的功能，main.py 负责启动和管理服务器，以及定时向客户端发送消息。

安装 WebSocket 库

在开始之前，需要安装 websockets 库，它是 Python 中非常流行的 WebSocket 实现。

pip install websockets

websocket.py — WebSocket 服务器实现

在 websocket.py 文件中，我定义了一个 WebSocketServer 类，包含了 WebSocket 服务器的主要逻辑。

代码解析

import asyncio
import websockets

class WebSocketServer:
def __init__(self, host=\”localhost\”, port=8765):
self.host = host
self.port = port
self.clients = set() # 存储所有连接的客户端

首先，我创建了一个 WebSocketServer 类，它的构造方法初始化了服务器的主机地址 host 和端口号 port，同时维护了一个客户端集合 clients 来存储当前连接的 WebSocket 客户端。

async def handle_client(self, websocket):
# 新的客户端连接
self.clients.add(websocket)
try:
async for message in websocket:
print(f\”收到消息: {
message}\”)
# 回显消息给客户端
await websocket.send(f\”服务器已收到: {
message}\”)
except websockets.ConnectionClosed as e:
print(f\”客户端断开连接: {
e}\”)
finally: