a e=iýã@s$dZGdd„dƒZdd„ZeƒZdS)z± WebSocket Registry Module This module provides efficient O(1) user lookup and management for WebSocket connections. Replaces O(n) set iteration with dictionary-based lookups. c@s^eZdZdZdd„Zddœdd„Zddœdd „Zed œd d „Ze dœd d„Z e dœdd„Z dS)Ú UserRegistryzl Efficient O(1) user lookup registry. Replaces O(n) set iteration with O(1) dictionary lookup. cCs i|_dS)N)Ú_users©Úself©rú:/var/www/lichun.app/lichun/ws/server/websocket_registry.pyÚ__init__szUserRegistry.__init__N)ÚreturncCst|dƒr||j|j<dS)zAdd a websocket connectionÚuserIDN)Úhasattrrr ©rÚ websocketrrrÚadds zUserRegistry.addcCs$t|dƒr |j|jvr |j|j=dS)zRemove a websocket connectionr N©r r rr rrrÚremoveszUserRegistry.remove)Úuser_idcCs |j |¡S)zGet websocket by user ID (O(1)))rÚget)rrrrrrszUserRegistry.getcCs t|jƒS)zGet number of connected users)ÚlenrrrrrÚcount!szUserRegistry.countcCst|dƒr|j|jvSdS)z Check if websocket is registeredr Frr rrrÚ __contains__%s  zUserRegistry.__contains__) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr rÚstrrÚintrÚboolrrrrrr srcCs t |¡S)z¯ Get websocket connection for a specific player ID. Args: player_id: The player's user ID Returns: WebSocket connection or None if not found )ÚUSERSr)Ú player_idrrrÚget_websocket_for_player,s rN)rrrrrrrrÚs#