将join room提前到on_connect

This commit is contained in:
CakeCN
2026-01-08 20:50:56 +08:00
parent 09103abf38
commit 5b8180b48e
2 changed files with 10 additions and 2 deletions

View File

@@ -63,6 +63,13 @@ class VSCodeNamespace(Namespace):
class AgentNamespace(Namespace): class AgentNamespace(Namespace):
def on_connect(self):
# 获取前端连接时传递的 query 参数
user_uuid = request.args.get('user_uuid')
folder = request.args.get('folder')
print(f'Agent client connected with user_uuid: {user_uuid}, folder: {folder}')
join_room(user_id, namespace='/agent')
def _load_learning_progress(self, chatmanager, user_id, course_id, chapter_name, lesson_name, continue_learn): def _load_learning_progress(self, chatmanager, user_id, course_id, chapter_name, lesson_name, continue_learn):
"""加载并恢复用户学习进度 """加载并恢复用户学习进度
@@ -120,7 +127,6 @@ class AgentNamespace(Namespace):
chapter_name = data['chapter_name'] chapter_name = data['chapter_name']
user_id = uuid2username[user_uuid] user_id = uuid2username[user_uuid]
session['user_uuid'] = user_uuid session['user_uuid'] = user_uuid
join_room(user_uuid, namespace='/agent') # 将该用户加入以user_id为名的room
print(f'User connected with session user_uuid: {user_uuid}') print(f'User connected with session user_uuid: {user_uuid}')
user_uuid2chatmanager = ex["user_uuid2chatmanager"] user_uuid2chatmanager = ex["user_uuid2chatmanager"]

View File

@@ -75,6 +75,8 @@ document.addEventListener('DOMContentLoaded', function () {
folder: data.folder folder: data.folder
} }
}); });
console.log('Connecting to WebSocket server at wss://${host}/agent');
console.log(data);
// 初始化聊天消息管理器 // 初始化聊天消息管理器
chatManager = new ChatMessageManager(); chatManager = new ChatMessageManager();