WebSocket 入门教程

news2024/11/23 15:37:48

什么是 WebSocket？

WebSocket 是一种通信协议，它在单个 TCP 连接上提供全双工通信。与传统的 HTTP 不同，WebSocket 允许服务器主动向客户端推送数据，而不仅仅是客户端请求数据。这使得 WebSocket 非常适用于需要低延迟和实时通信的应用，例如在线聊天、游戏和实时数据流。
在这里插入图片描述

WebSocket 的工作原理

WebSocket 的通信过程可以分为以下几个步骤：

握手：客户端向服务器发送一个 HTTP 请求，要求升级到 WebSocket 协议。
建立连接：服务器接受请求并返回响应，连接建立后，客户端和服务器可以在同一连接上互相发送消息。
传输数据：数据传输过程中，客户端和服务器可以随时发送和接收消息，通信是全双工的。
关闭连接：任意一方可以随时关闭连接。

WebSocket 的优势

实时性：WebSocket 提供低延迟的实时通信，适用于需要即时响应的应用。
节省带宽：相比 HTTP 轮询，WebSocket 减少了不必要的网络开销，节省带宽。
全双工通信：允许客户端和服务器同时发送和接收数据，通信效率更高。

WebSocket 示例

为了更好地理解 WebSocket 的工作原理，我们将通过一个简单的示例来展示如何使用 WebSocket 进行通信。我们将实现一个基本的 WebSocket 服务器和客户端。

安装 WebSocket 库

在开始之前，我们需要安装 Python 的 websockets 库。可以通过以下命令安装：

pip install websockets

实现 WebSocket 服务器

首先，我们来实现一个简单的 WebSocket 服务器。服务器将接受客户端的连接，并能够接收和发送消息。

# server.py
import asyncio
import websockets

async def handle_connection(websocket, path):
    print("客户端已连接")
    try:
        while True:
            message = await websocket.recv()
            print(f"收到消息: {message}")
            response = f"服务器收到: {message}"
            await websocket.send(response)
            print(f"发送消息: {response}")
    except websockets.ConnectionClosed:
        print("客户端已断开连接")

async def start_server():
    server = await websockets.serve(handle_connection, "localhost", 8765)
    print("WebSocket 服务器已启动，监听端口 8765")
    await server.wait_closed()

if __name__ == "__main__":
    asyncio.run(start_server())

实现 WebSocket 客户端

接下来，我们来实现一个简单的 WebSocket 客户端。客户端将连接到服务器，并发送和接收消息。

# client.py
import asyncio
import websockets

async def communicate():
    uri = "ws://localhost:8765"
    async with websockets.connect(uri) as websocket:
        print("已连接到服务器")
        while True:
            message = input("请输入要发送的消息: ")
            await websocket.send(message)
            print(f"发送消息: {message}")
            response = await websocket.recv()
            print(f"收到消息: {response}")

if __name__ == "__main__":
    asyncio.run(communicate())