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

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

article 2024/11/14 5:18:57

源码分享
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())

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