Python中WebSocket连接绑定独立进程、多worker管控的实现咨询

基础配置 #

• 运行环境为配备4核CPU的服务器
• 需启动8个worker进程承载业务处理

核心逻辑规则 #

• 服务启动阶段：完成8个worker进程的全量初始化，统一交由进程管理器维护状态
• 新连接接入逻辑：
  • 新WebSocket连接建立时，先从进程管理器获取状态为可用的worker
  • 成功获取到worker后，将该worker状态修改为占用，并与当前连接绑定
  • 若无可用worker，直接向客户端返回服务器负载已满提示，拒绝本次连接
• 连接运行阶段：
  • WebSocket收到客户端新数据时，直接转发至绑定的对应worker处理
  • 所有worker进程之间无需进行任何数据交互
• 连接断开逻辑：WebSocket连接关闭后，立即将绑定的worker状态重置为可用，等待分配给后续新连接

需求逻辑示意 #

注：以下为伪代码，仅用于展示需求逻辑，并非可运行的Python代码

process_manager = ProcessManager()

# WebSocket事件回调定义
WebSocket 
    # 新连接建立时触发
    onNewConnection:
        available_process = process_manager.get_available_process()
        if available_process:
            available_process.set_state(state.Occupied)
            # 绑定当前连接与分配到的worker
            self.binded_process = available_process
        else:
            ws.write("Server is on full load")

    # 收到新消息时触发
    onNewMessage(_data):
        self.binded_process.sendData(_data)

    # 连接关闭时触发
    OnSocketClose:
        self.binded_process.set_state(state.Available)


# 主程序入口逻辑
__main__
    process_pool = ProcessPool(8)
    for i in 0..7:
        process_manager.put(i, Process)
    # 启动所有worker进程
    # 等待所有进程运行结束

内容的提问来源于stack exchange，提问作者Clean Coder