WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议，适用于实时数据更新，如聊天应用、游戏或实时推送通知等。与传统的 HTTP 协议相比，WebSocket 可以维持持久连接，并在客户端和服务器之间快速交换数据。

下面是如何使用 WebSocket 的简单示例。这里我们将创建一个 WebSocket 客户端，连接到一个 WebSocket 服务器，以接收实时消息。

创建 WebSocket 客户端

1. 安装 websocket-client 库

首先，你需要安装 Python 的 WebSocket 客户端库，可以用 pip 安装：

pip install websocket-client

2. 编写基本的 WebSocket 客户端代码

下面是一个简单的 WebSocket 客户端示例代码，它连接到一个 WebSocket 服务器，接收消息并处理它们。

import websocket
import json

# 要连接的 WebSocket 服务器地址
WEBSOCKET_URL = "ws://your.websocket.server/url"  # 替换为你的 WebSocket 服务器地址

# 定义接收到消息的回调函数
def on_message(ws, message):
    print(f"收到消息: {message}")
    
    # 你可以在这里添加代码来处理收到的消息
    # 例如，发送到推送服务或更新数据库等

# 定义连接打开时的回调函数
def on_open(ws):
    print("连接已打开！")
    # 如果需要，可以发送一些初始消息到服务器
    # ws.send(json.dumps({"action": "subscribe", "channel": "your_channel_name"}))

# 定义连接关闭时的回调函数
def on_close(ws, close_status_code, close_msg):
    print("连接已关闭")

# 创建 WebSocket 应用
ws = websocket.WebSocketApp(
    WEBSOCKET_URL,
    on_message=on_message,
    on_open=on_open,
    on_close=on_close
)

# 运行 WebSocket 客户端
ws.run_forever()

3. 说明代码中的部分

• on_message：当接收到消息时该函数会被调用。你可以在这里处理消息，例如解析 JSON 数据并作出相应。
• on_open：连接成功后调用，可以在这里发送消息或执行其他初始化操作。
• on_close：连接关闭后调用。

4. 在服务器端实现 WebSocket

如果你还没有 WebSocket 服务器，需要创建一个。下面是一个使用 Flask-SocketIO 的简单示例：

服务器端代码示例

首先安装 Flask-SocketIO：

pip install flask-socketio

然后创建一个简单的 WebSocket 服务器：

from flask import Flask
from flask_socketio import SocketIO, emit

app = Flask(__name__)
socketio = SocketIO(app)

@app.route('/')
def index():
    return "WebSocket Server"

@socketio.on('message')
def handle_message(data):
    print(f"客户端消息: {data}")
    # 可以在这里处理接收到的消息，并选择是否回复
    emit('response', {'data': '这里是服务器的回复'})

if __name__ == '__main__':
    socketio.run(app, host='0.0.0.0', port=5000)

标签： 来源：

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。