【WebSocket教程】第10节：WebSocket 服务端搭建（Python）

原创于 2025-08-28 13:00:00 发布 · 1.3k 阅读

29 ·

CC 4.0 BY-SA版权

文章标签：

#websocket #python #网络协议 #前端 #javascript #教程

Websocket技术栈专栏收录该内容

21 篇文章

订阅专栏

🖥️第10节：WebSocket 服务端搭建（Python）

💬 引言

🏢 哈喽，大家好！我是老曹。在前几节课中，我们分别学习了如何使用 Node.js 和 Java 搭建 WebSocket 服务端。今天，我们将切换到 Python 生态系统，使用 Flask 框架结合 Flask-SocketIO 库来构建一个实时通信服务。

Flask-SocketIO 是一个功能强大的库，它使得在 Flask 应用中集成 WebSocket 变得非常简单。通过本节课的学习，你将掌握如何快速搭建一个基于 Python 的 WebSocket 服务端，并实现消息广播、点对点通信等核心功能。

🎯 学习目标

掌握 Flask 和 Flask-SocketIO 的基本用法
学会创建 WebSocket 服务端并处理连接事件
实现消息广播和点对点通信
理解 Flask-SocketIO 的配置与优化
构建可扩展的 WebSocket 服务端架构

🛠️ 1. 准备工作

🔧1.1 安装依赖

首先，我们需要安装 Flask 和 Flask-SocketIO 库。可以通过以下命令快速安装：

pip install flask flask-socketio

💡 关键点：

Flask 是一个轻量级的 Web 框架。
Flask-SocketIO 提供了对 WebSocket 的支持，并简化了实时通信的开发。

📂1.2 创建项目结构

为了保持代码清晰，我们可以按照以下结构组织项目：

websocket-server/
├── app.py          # 主应用文件
├── requirements.txt # 依赖列表
└── templates/      # HTML 模板文件夹
    └── index.html  # 前端页面

🖥️2. 配置 WebSocket 服务端

🧩2.1 初始化 Flask 应用

在 app.py 中初始化 Flask 应用并集成 Flask-SocketIO。

示例代码：

from flask import Flask, render_template
from flask_socketio import SocketIO, send, emit

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
socketio = SocketIO(app)

@app.route('/')
def index():
    return render_template('index.html')

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

📌 注意：

SECRET_KEY 用于加密会话数据，生产环境中应设置为随机字符串。
socketio.run() 启动应用并启用 WebSocket 支持。

🔄3. 实现 WebSocket 处理器

🧰3.1 监听连接事件

Flask-SocketIO 提供了装饰器来监听连接、断开和消息事件。

示例代码：

@socketio.on('connect')
def handle_connect():
    print('新客户端连接')

@socketio.on('disconnect')
def handle_disconnect():
    print('客户端断开连接')

@socketio.on('message')
def handle_message(message):
    print(f'收到消息: {message}')
    send(message, broadcast=True)  # 广播消息给所有客户端

💡 关键点：

connect 和 disconnect 事件分别在客户端连接和断开时触发。
message 事件用于接收客户端发送的消息。

📢 4. 消息广播与点对点通信

🔄4.1 消息广播的实现

广播消息是指将一条消息发送给所有连接的客户端。以下是改进后的广播逻辑：

from flask_socketio import join_room, leave_room

users = {}  # 用户会话字典

@socketio.on('message')
def handle_message(data):
    sender_id = data.get('sender')
    content = data.get('content')

    print(f'收到消息: {content} (来自 {sender_id})')

    # 广播消息
    emit('message', {
        'sender': sender_id,
        'content': content,
        'timestamp': get_current_timestamp()
    }, broadcast=True)

def get_current_timestamp():
    from datetime import datetime
    return int(datetime.now().timestamp() * 1000)

📌 优化点：

使用 emit 方法发送消息，支持自定义事件名称。
broadcast=True 表示将消息广播给所有客户端。

📩 4.2 点对点通信的实现

点对点通信是指将消息发送给特定的客户端。为了实现这一点，我们可以使用 Flask-SocketIO 提供的房间机制。

示例代码：

@socketio.on('join')
def on_join(data):
    user_id = data['user_id']
    room = data['room']
    join_room(room)
    users[user_id] = room
    print(f'{user_id} 加入房间 {room}')

@socketio.on('leave')
def on_leave(data):
    user_id = data['user_id']
    room = users.pop(user_id, None)
    if room:
        leave_room(room)
        print(f'{user_id} 离开房间 {room}')

@socketio.on('private_message')
def handle_private_message(data):
    sender_id = data.get('sender')
    receiver_id = data.get('receiver')
    content = data.get('content')

    room = users.get(receiver_id)
    if room:
        emit('private_message', {
            'sender': sender_id,
            'content': content,
            'timestamp': get_current_timestamp()
        }, room=room)
    else:
        print(f'目标用户 {receiver_id} 不在线')

📌 注意：

join_room 和 leave_room 方法用于管理房间。
通过 room 参数指定消息的目标房间。

🚀5. 配置与优化

🔒5.1 跨域支持

在实际项目中，WebSocket 通常需要支持跨域访问。可以通过以下方式配置：

socketio = SocketIO(app, cors_allowed_origins="*")

📌 注意：

生产环境中应严格限制允许的来源，避免安全风险。

🧮5.2 性能优化

在高并发场景下，可以通过以下方式优化性能：

异步处理：使用异步方法处理消息。
负载均衡：部署多个服务端实例，并使用消息队列同步状态。
Redis 支持：启用 Redis 作为消息代理，支持水平扩展。

示例代码（启用 Redis）：

from flask_socketio import SocketIO

socketio = SocketIO(app, message_queue='redis://localhost:6379')

📊6. 实战案例分析

📢6.1 实时聊天室系统

在实时聊天室系统中，服务端需要支持多人在线聊天。以下是完整的实现步骤：

用户身份管理：为每个客户端分配唯一 ID。
消息路由：支持广播和点对点通信。
在线状态管理：显示当前在线用户列表。

示例代码（在线状态管理）：

online_users = set()

@socketio.on('connect')
def handle_connect():
    user_id = request.sid
    online_users.add(user_id)
    emit('update_online_users', list(online_users), broadcast=True)

@socketio.on('disconnect')
def handle_disconnect():
    user_id = request.sid
    online_users.discard(user_id)
    emit('update_online_users', list(online_users), broadcast=True)