a %i@sTdZddlmZddlmZmZddlZddlZddlZe e Z GdddZ dS)zz LRU cache for player records with auto-save on eviction. Prevents unbounded memory growth by evicting inactive players. ) OrderedDict)OptionalCallableNc@seZdZdZdeedddZeddd Zed d d d Z eed ddZ d dddZ d dddZ edddZ edddZedddZedddZd S) PlayerCacheaW LRU cache for player records. Features: - Automatic eviction of least-recently-used players - Never evicts connected players - Auto-save on eviction (optional) - Memory usage tracking Usage: cache = PlayerCache(max_size=100) cache.set("user123", player) player = cache.get("user123") dT)max_size auto_savecCs4t|_||_||_d|_td|d|dS)z Initialize player cache. Args: max_size: Maximum number of players to keep in memory auto_save: Whether to auto-save evicted players Nz"PlayerCache initialized: max_size=z , auto_save=)r_cache _max_size _auto_saveon_evictloggerinfo)selfrrr-/var/www/lichun.app/lichun/ws/player_cache.py__init__ s zPlayerCache.__init__)user_idcCs$||jvr |j||j|SdS)z Get player from cache (marks as recently used). Args: user_id: User ID Returns: playerClass instance or None N)r move_to_endrrrrrget3s   zPlayerCache.getN)rreturncCsJ||jvr$|j|||j|<dSt|j|jkr<|||j|<dS)z{ Add player to cache. Args: user_id: User ID player: playerClass instance N)r rlenr _evict_lrurrplayerrrrsetCs   zPlayerCache.setcCs||jvr|j|=dSdS)z Remove player from cache. Args: user_id: User ID Returns: True if removed, False if not found TF)r rrrrremoveXs zPlayerCache.remove)rcCst|jD]b\}}|jdkrtd|d|jjd|jrL| ||j rbt | ||j|=dSqt d|jddS)z3Evict least recently used player (unless connected) disconnectedzEvicting player from cache: z ()NzPlayerCache full (z) with all connected players)listr items connectionr rc firstnamer _save_playerr asyncio create_taskwarningr rrrrrgs  zPlayerCache._evict_lruc Cs`ddlm}zt||Wn<tyZ}z$td|jd|WYd}~n d}~00dS)zSave player to database (async)r) saveGameAsynczFailed to save evicted player z: N) functionsr)r&r' Exceptionr errorid)rrr)errrr%}s  zPlayerCache._save_playercCs t|jS)zGet current cache size)rr rrrrsizeszPlayerCache.sizecCst|j|jkS)zCheck if cache is full)rr r r/rrris_fullszPlayerCache.is_fullcCs@|js dStt|j}t|}|t|jd}|dS)zo Estimate memory usage of cached players. Returns: Estimated memory in MB g i)r nextitervaluessys getsizeofr)rZ sample_playerZ sample_sizeZestimated_totalrrrestimate_memory_mbs  zPlayerCache.estimate_memory_mbcCsTtdd|jD}t|j|}t|j|j|||t|j|jddS)zGet cache statisticscss|]}|jdkrdVqdS) connectedN)r").0prrr z(PlayerCache.get_stats..r)r0rr9r memory_mbZ utilization)sumr r5rr r8)rr9rrrr get_statsszPlayerCache.get_stats)rT)__name__ __module__ __qualname____doc__intboolrstrrrrrr%r0r1floatr8dictrArrrrrs   r) rE collectionsrtypingrrr6loggingr& getLoggerrBr rrrrrs