From ee00470ced35dadcfee01d4b94ed01001a2dd7f1 Mon Sep 17 00:00:00 2001 From: Wu Clan Date: Sun, 15 Mar 2026 23:11:02 +0800 Subject: [PATCH] Fix socket auth --- backend/common/socketio/server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/common/socketio/server.py b/backend/common/socketio/server.py index 6e237878..64d62968 100644 --- a/backend/common/socketio/server.py +++ b/backend/common/socketio/server.py @@ -1,7 +1,10 @@ import urllib.parse +import uuid import socketio +from starlette_context import request_cycle_context + from backend.common.log import log from backend.common.security.jwt import jwt_authentication from backend.core.conf import settings @@ -38,7 +41,8 @@ async def connect(sid, environ, auth) -> bool: return True try: - await jwt_authentication(token) + with request_cycle_context({settings.TRACE_ID_REQUEST_HEADER_KEY: uuid.uuid4().hex}): + await jwt_authentication(token) except Exception as e: log.info(f'WebSocket 连接失败：{e!s}') return False