使用 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: