深入理解WebSocket协议：实现实时通信的利器

源码分享
https://docs.qq.com/sheet/DUHNQdlRUVUp5Vll2?tab=BB08J2

WebSocket是一种在单个TCP连接上进行全双工通信的协议。它允许服务器实时地向客户端推送信息，是构建聊天应用、实时游戏、股票行情应用等需要快速、实时通信的web应用的理想选择。在这篇技术博客中，我们将探讨WebSocket协议的基础知识和如何在Python中使用它。

WebSocket协议概述

WebSocket协议由IETF作为RFC 6455标准发布，它提供了一个标准的方式来实现客户端和服务器之间的持久连接。与传统的HTTP请求不同，WebSocket提供了一个持续开放的连接，允许数据在任何时候从任一端传输到另一端。

WebSocket的特点

• 支持全双工通信
• 低延迟，适用于实时应用
• 通过ws（非加密）或wss（加密）协议访问
• 可以发送文本或二进制数据

在Python中使用WebSocket

在Python中，我们可以使用​​websockets​​库来创建WebSocket服务器和客户端。以下是一个简单例子的展示：

安装websockets库

确保先安装​​websockets​​库：

pip install websockets

WebSocket服务器

import asyncio
import websockets

async def echo(websocket, path):
    async for message in websocket:
        print(f"Received message: {message}")
        await websocket.send(f"Server received: {message}")

start_server = websockets.serve(echo, "localhost", 8765)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

这段代码创建了一个WebSocket服务器，它监听本地主机的8765端口。每当服务器收到消息时，它会将接收到的消息打印到控制台，并将确认消息发送回客户端。

WebSocket客户端

import asyncio
import websockets

async def hello():
    uri = "ws://localhost:8765"
    async with websockets.connect(uri) as websocket:
        await websocket.send("Hello, server!")
        response = await websocket.recv()
        print(f"Received response from server: {response}")

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

客户端连接到服务器，并发送一条消息，然后接收并打印来自服务器的响应。

WebSocket协议的工作机制

WebSocket连接通过HTTP协议开始，并通过HTTP升级机制在服务器和客户端之间建立。一旦连接升级为WebSocket连接，数据就可以在客户端和服务器之间双向流动。

使用场景

WebSocket非常适合需要实时功能的应用，如：

• 聊天应用
• 在线游戏
• 即时通知系统
• 实时数据传输（如股票行情）

结论

WebSocket是实现实时通信应用的有效协议。它提供了与传统HTTP协议相比的显著优势，包括低延迟、全双工通信和持久连接。在Python中，利用​​websockets​​库可以相对容易地实现WebSocket通信。