ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-07-14 14:35:13 +04:00
parent 18bd83413b
commit f402957f05
1 changed files with 0 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
backend/open_webui/socket/main.py
View file

@ -107,7 +107,6 @@ if WEBSOCKET_MANAGER == "redis":
# TODO: Implement Yjs document management with Redis # TODO: Implement Yjs document management with Redis
DOCUMENTS = {} DOCUMENTS = {}
DOCUMENT_USERS = {}
clean_up_lock = RedisLock( clean_up_lock = RedisLock(
redis_url=WEBSOCKET_REDIS_URL, redis_url=WEBSOCKET_REDIS_URL,
@ -124,7 +123,6 @@ else:
USAGE_POOL = {} USAGE_POOL = {}
DOCUMENTS = {} # document_id -> Y.YDoc instance DOCUMENTS = {} # document_id -> Y.YDoc instance
DOCUMENT_USERS = {} # document_id -> set of user sids
aquire_func = release_func = renew_func = lambda: True aquire_func = release_func = renew_func = lambda: True
@ -377,11 +375,9 @@ async def yjs_document_join(sid, data):
"ydoc": Y.Doc(), # Create actual Yjs document "ydoc": Y.Doc(), # Create actual Yjs document
"users": set(), "users": set(),
} }
DOCUMENT_USERS[document_id] = set()
# Add user to document # Add user to document
DOCUMENTS[document_id]["users"].add(sid) DOCUMENTS[document_id]["users"].add(sid)
DOCUMENT_USERS[document_id].add(sid)
# Join Socket.IO room # Join Socket.IO room
await sio.enter_room(sid, f"doc_{document_id}") await sio.enter_room(sid, f"doc_{document_id}")
@ -506,9 +502,6 @@ async def yjs_document_leave(sid, data):
if document_id in DOCUMENTS: if document_id in DOCUMENTS:
DOCUMENTS[document_id]["users"].discard(sid) DOCUMENTS[document_id]["users"].discard(sid)
if document_id in DOCUMENT_USERS:
DOCUMENT_USERS[document_id].discard(sid)
# Leave Socket.IO room # Leave Socket.IO room
await sio.leave_room(sid, f"doc_{document_id}") await sio.leave_room(sid, f"doc_{document_id}")
@ -523,7 +516,6 @@ async def yjs_document_leave(sid, data):
# If no users left, clean up the document # If no users left, clean up the document
log.info(f"Cleaning up document {document_id} as no users are left") log.info(f"Cleaning up document {document_id} as no users are left")
del DOCUMENTS[document_id] del DOCUMENTS[document_id]
del DOCUMENT_USERS[document_id]
except Exception as e: except Exception as e:
log.error(f"Error in yjs_document_leave: {e}") log.error(f"Error in yjs_document_leave: {e}")