UniqAI
/
open-webui
mirror of https://github.com/open-webui/open-webui
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: active users

This commit is contained in:
Timothy J. Baek 2024-06-07 21:38:09 -07:00
parent f78e6a5fba
commit d7d8896e43
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
backend/apps/socket/main.py
View File

@ -134,13 +134,17 @@ async def remove_after_timeout(sid, model_id):
@sio.event @sio.event
async def disconnect(sid): async def disconnect(sid):
if sid in USER_POOL: if sid in SESSION_POOL:
disconnected_user = SESSION_POOL.pop(sid) user_id = SESSION_POOL[sid]
USER_POOL[disconnected_user].remove(sid) del SESSION_POOL[sid]
if len(USER_POOL[disconnected_user]) == 0:
del USER_POOL[disconnected_user]
print(f"user {disconnected_user} disconnected with session ID {sid}") USER_POOL[user_id].remove(sid)
if len(USER_POOL[user_id]) == 0:
del USER_POOL[user_id]
print(f"user {user_id} disconnected with session ID {sid}")
print(USER_POOL)
await sio.emit("user-count", {"count": len(USER_POOL)}) await sio.emit("user-count", {"count": len(USER_POOL)})
else: else: