Python WebSocket 服务端|极客教程

Python WebSocket 服务端

WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，它和 HTTP 协议不同之处在于，WebSocket 协议可以实现服务端向客户端推送数据，而不是仅由客户端向服务端发起请求。在本文中，我们将详细介绍如何使用 Python 来创建一个 WebSocket 服务端。

WebSocket 简介

WebSocket 是一种基于 TCP 协议的通信协议，它允许客户端和服务端之间建立一条持久的连接，并进行全双工通信。WebSocket 协议允许服务端主动向客户端推送数据，而不需要客户端主动发起请求。

WebSocket 协议的特点包括：

建立在 TCP 协议之上，使用端口 80 或 443；
与 HTTP 协议不同，可以支持全双工通信；
可以在客户端和服务端之间建立一个持久的连接；
支持跨域通信。

由于 WebSocket 是一种相对新的协议，所以目前仍在不断发展之中。在前端开发中，我们通常会使用 WebSocket 来实现实时通信功能，例如在线聊天、数据更新推送等。

Python WebSocket 服务端搭建

在 Python 中，我们可以使用第三方库 websockets 来快速搭建一个 WebSocket 服务端。首先，我们需要安装该库：

pip install websockets

接着，我们可以编写一个简单的 WebSocket 服务端示例：

# server.py
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 服务，并监听在 localhost 的 8765 端口上。

运行 WebSocket 服务端

为了运行我们的 WebSocket 服务端，我们只需要在命令行中运行上述代码：

python server.py

这样就启动了一个 WebSocket 服务端，并监听在 localhost:8765 地址上。接下来，我们可以使用任何支持 WebSocket 协议的客户端来连接到这个服务端，并进行通信。

示例代码

下面是一个简单的 Python 客户端示例，用来连接到我们刚刚搭建的 WebSocket 服务端，并发送和接收消息：

# client.py
import asyncio
import websockets

async def hello():
    uri = "ws://localhost:8765"
    async with websockets.connect(uri) as websocket:
        await websocket.send("Hello, WebSocket!")
        response = await websocket.recv()
        print(response)

asyncio.get_event_loop().run_until_complete(hello())

在上面的示例中，我们定义了一个 hello 函数，用来连接到 WebSocket 服务端，并发送消息 “Hello, WebSocket!”，然后接收服务端返回的消息并打印出来。

接着，我们可以在命令行中运行上述代码：