如何在Flask-SocketIO后台线程中正确设置Session上下文

你当前的写法存在两个本质错误，导致Session写入不生效：

• app.test_request_context() 生成的是全新的空请求上下文，和发起SocketIO连接的真实用户没有任何关联，写入这个上下文的Session数据会随着上下文销毁直接丢失，根本不会落到用户的真实会话存储里。
• 你把数据采集线程设为全局单例，这个线程不属于任何单个用户，本身就没有天然绑定的用户请求上下文，不可能直接通过全局session代理对象操作用户会话。

1. 连接阶段绑定用户会话标识 #

SocketIO的connect事件本身运行在正确的请求上下文中，你可以在这个阶段拿到当前连接用户的Session ID，和SocketIO分配的客户端SID做绑定存储，不要给后台任务传无意义的app对象。

from flask import session, request
# 存储 客户端SID -> 用户SessionID 的映射，多线程操作必须加锁
user_session_map = {}

@socketio.event
def connect():
    global data_stream_thread
    current_sid = request.sid
    # 直接获取当前连接用户的服务端Session ID
    current_session_id = session.sid
    with ds_thread_lock:
        user_session_map[current_sid] = current_session_id
        if data_stream_thread is None:
            # 后台任务不需要传app，需要时通过current_app获取即可
            data_stream_thread = socketio.start_background_task(data_stream)

2. 后台线程手动操作Session #

你已经配置了flask-session做文件系统存储，不需要依赖请求上下文，直接通过Session接口手动加载、修改、保存对应用户的会话即可。

from flask import current_app

def data_stream():
    while True:
        socketio.sleep(1)
        get_usb_connections()

def get_usb_connections():
    app = current_app._get_current_object()
    session_interface = app.session_interface
    session_cookie_name = app.config.get('SESSION_COOKIE_NAME', 'session')

    # 遍历所有在线用户，写入对应Session
    for sid, session_id in list(user_session_map.items()):
        # 构造假请求对象，提供加载Session需要的cookie信息
        class FakeRequest:
            def __init__(self):
                self.cookies = {session_cookie_name: session_id}
        fake_req = FakeRequest()
        # 加载用户已有Session
        user_session = session_interface.open_session(app, fake_req)
        if not user_session:
            continue
        # 写入需要存储的数据
        user_session['testVar'] = 'test'
        # 手动保存修改后的Session
        session_interface.save_session(app, user_session, fake_req)

    # 原有USB数据采集、emit逻辑，注意指定接收客户端避免全局广播
    # socketio.emit(usb_response, {'data': data, 'device': device, 'color': color}, to=sid)

3. 断开连接时清理缓存 #

新增断开事件监听，移除离线用户的映射关系，避免内存泄漏：

@socketio.event
def disconnect():
    current_sid = request.sid
    with ds_thread_lock:
        user_session_map.pop(current_sid, None)

• 你当前设置的socketio = SocketIO(manage_session=False)配置是正确的，交给flask-session统一管理会话即可，避免Flask-SocketIO自带的会话管理和服务端Session冲突。
• 不要在后台线程中直接导入使用全局session代理对象，这个代理只有在活跃的请求上下文中才会绑定到对应用户，后台线程无请求上下文时直接调用会抛出上下文错误。
• 如果USB连接状态是全局共享的，不需要重复写入每个用户的Session，可以把公共状态存在进程级缓存或者Redis里，路由和SocketIO事件直接读公共缓存即可，性能更好；只有用户独立的配置数据需要存在Session中。

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