youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

ruff format

Browse Source
This commit is contained in:
hjlarry
2025-07-17 15:37:13 +08:00
parent 0d7d27ec0b
commit 37440e9416
4 changed files with 30 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
api/controllers/console/app/online_user.py
View File

@ -1,20 +1,21 @@
import json
from flask_login import current_user, login_required
from flask import request
from flask_login import current_user, login_required
from extensions.ext_socketio import ext_socketio
from extensions.ext_redis import redis_client
from extensions.ext_socketio import ext_socketio
@ext_socketio.on('user_connect')
@ext_socketio.on("user_connect")
@login_required
def handle_user_connect(data):
"""
Handle user connect event, check login and get user info.
"""
sid = request.sid
workflow_id = data.get('workflow_id')
workflow_id = data.get("workflow_id")
old_info_json = redis_client.hget(f"workflow_online_users:{workflow_id}", current_user.id)
if old_info_json:
@ -27,25 +28,16 @@ def handle_user_connect(data):
"user_id": current_user.id,
"username": getattr(current_user, "username", ""),
"avatar": getattr(current_user, "avatar", ""),
"sid": sid
}
"sid": sid,
}
redis_client.hset(
f"workflow_online_users:{workflow_id}",
current_user.id,
json.dumps(user_info)
)
redis_client.hset(f"workflow_online_users:{workflow_id}", current_user.id, json.dumps(user_info))
redis_client.set(
f"ws_sid_map:{sid}",
json.dumps({
"workflow_id": workflow_id,
"user_id": current_user.id
})
)
return {'msg': 'connected', 'user_id': current_user.id, 'sid': sid}
redis_client.set(f"ws_sid_map:{sid}", json.dumps({"workflow_id": workflow_id, "user_id": current_user.id}))
return {"msg": "connected", "user_id": current_user.id, "sid": sid}
@ext_socketio.on('disconnect')
@ext_socketio.on("disconnect")
def handle_disconnect():
"""
Handle user disconnect event, remove user from workflow's online user list.
@ -55,6 +47,6 @@ def handle_disconnect():
if mapping:
data = json.loads(mapping)
workflow_id = data["workflow_id"]
user_id = data["user_id"]
user_id = data["user_id"]
redis_client.hdel(f"workflow_online_users:{workflow_id}", user_id)
redis_client.delete(f"ws_sid_map:{sid}")
redis_client.delete(f"ws_sid_map:{sid}")