python实现websocket

时间：2024-03-19 16:03:04

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

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

  1. 连接方式：

    • HTTP 协议是基于请求 - 响应模式的，客户端发送请求，服务器返回响应，然后连接关闭。每次请求都需要建立新的连接。
    • WebSocket 协议是全双工通信的，客户端和服务器之间建立一次连接后，可以在单个连接上进行双向实时通信，而无需重新建立连接。

  2. 实时性：

    • 由于 HTTP 是基于请求 - 响应模式的，它不太适合实时通信。在 HTTP 中，服务器不能主动向客户端发送消息；它只能响应客户端的请求。
    • WebSocket 允许服务器主动向客户端发送消息，实现了实时通信，适用于需要及时更新数据的应用，如在线游戏、实时聊天和股票市场数据等。

  3. 协议开销：

    • HTTP 协议每次请求都需要携带完整的 HTTP 头信息，这会增加一定的开销。
    • WebSocket 协议通过建立长连接来避免重复的握手过程和头信息传输，减少了协议开销，提高了数据传输的效率。

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

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

pip install websockets

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

import asyncio
import websockets

async 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 websockets

async 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())

在上面的代码中，定义了一个send_message函数来与WebSocket服务器建立连接，并发送和接收消息。

通过以上两段代码，可以实现一个简单的WebSocket通讯。您可以根据具体的需求来修改和扩展这些代码。

相关文章