当前位置：首页 > news >正文

python实现websocket

news 2025/8/29 22:15:30

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。它允许客户端和服务器之间进行实时数据传输，而不是像 HTTP 协议那样，每次请求都需要建立新的连接。WebSocket 协议最初是由 HTML5 定义的，旨在提供一种更有效的替代方案，以解决 HTTP 协议在实时通信方面的限制。

以下是 WebSocket 和 HTTP 之间的一些主要区别：

连接方式：
- HTTP 协议是基于请求 - 响应模式的，客户端发送请求，服务器返回响应，然后连接关闭。每次请求都需要建立新的连接。
- WebSocket 协议是全双工通信的，客户端和服务器之间建立一次连接后，可以在单个连接上进行双向实时通信，而无需重新建立连接。
实时性：
- 由于 HTTP 是基于请求 - 响应模式的，它不太适合实时通信。在 HTTP 中，服务器不能主动向客户端发送消息；它只能响应客户端的请求。
- WebSocket 允许服务器主动向客户端发送消息，实现了实时通信，适用于需要及时更新数据的应用，如在线游戏、实时聊天和股票市场数据等。
协议开销：
- HTTP 协议每次请求都需要携带完整的 HTTP 头信息，这会增加一定的开销。
- WebSocket 协议通过建立长连接来避免重复的握手过程和头信息传输，减少了协议开销，提高了数据传输的效率。

要在Python中实现WebSocket，可以使用第三方库websockets来处理WebSocket的连接和通讯。

首先，可以使用pip安装websockets库：

pip install websockets

然后，可以编写WebSocket服务器的代码示例:

import asyncio
import websocketsasync def echo(websocket, path):async for message in websocket:await websocket.send(message)start_server = websockets.serve(echo, "localhost", 8765)asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

在上面的代码中，定义了一个echo函数来处理接收到的消息，并通过websockets.serve来创建一个WebSocket服务器。最后通过asyncio.get_event_loop().run_forever()来启动服务器。

接着，可以编写WebSocket客户端的代码示例:

import asyncio
import websocketsasync def send_message():async with websockets.connect('ws://localhost:8765') as websocket:message = input("Enter message: ")await websocket.send(message)response = await websocket.recv()print("Received: ", response)asyncio.get_event_loop().run_until_complete(send_message())