&e.&ddlZddlmZmZmZmZmZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZddlmZerdd lmZe eZd ed efd Zd ed efd ZGddeZdS)N) TYPE_CHECKINGDictListMutableMappingOptional)WeakKeyDictionary)configutil) get_logger) ForwardMsg) CacheStatCacheStatsProvider)HASHLIB_KWARGS) AppSessionmsgreturnc>|jdkr|j}|dtjdit }||||_|j ||jS)aoComputes and assigns the unique hash for a ForwardMsg. If the ForwardMsg already has a hash, this is a no-op. Parameters ---------- msg : ForwardMsg Returns ------- string The message's hash, returned here for convenience. (The hash will also be assigned to the ForwardMsg; callers do not need to do this.) metadata) hashr ClearFieldhashlibmd5rupdateSerializeToString hexdigestCopyFrom)rrhashers Clib/python3.11/site-packages/streamlit/runtime/forward_msg_cache.pypopulate_hash_if_neededr!s" x2~~< z"""..~.. c++--...##%% h''' 8Oct}t||_|j|j|S)aCreate a ForwardMsg that refers to the given message via its hash. The reference message will also get a copy of the source message's metadata. Parameters ---------- msg : ForwardMsg The ForwardMsg to create the reference to. Returns ------- ForwardMsg A new ForwardMsg that "points" to the original message via the ref_hash field. )r r!ref_hashrr)rref_msgs r create_reference_msgr&As;$llG.s33G cl+++ Nr"ceZdZdZGddZdZdefdZdedd d e dd fd Z d ede efdZ dedd d e de fdZddZdd d e dd fdZddZdeefdZd S)ForwardMsgCacheaA cache of ForwardMsgs. Large ForwardMsgs (e.g. those containing big DataFrame payloads) are stored in this cache. The server can choose to send a ForwardMsg's hash, rather than the message itself, to a client. Clients can then request messages from this cache via another endpoint. This cache is *not* thread safe. It's intended to only be accessed by the server thread. ceZdZdZdeefdZdefdZddde dd fd Z ddde fd Z ddde de fd Z dd Zde fdZd S)ForwardMsgCache.EntryzCache entry. Stores the cached message, and the set of AppSessions that we've sent the cached message to. rc:||_t|_dSN)rr_session_script_run_counts)selfrs r __init__zForwardMsgCache.Entry.__init__ns"DH"##  + + +r"rc*tj|Sr,r repr_r.s r __repr__zForwardMsgCache.Entry.__repr__ts:d## #r"sessionrscript_run_countNc|j|d}||kr#td|d|d|}||j|<dS)aAdds a reference to a AppSession that has referenced this Entry's message. Parameters ---------- session : AppSession script_run_count : int The session's run count at the time of the call rzNew script_run_count (z) is < prev_run_count (z). This should never happen!N)r-getLOGGERerror)r.r5r6prev_run_counts r add_session_refz%ForwardMsgCache.Entry.add_session_refwso"<@@!LLN.00 3C3C3C^^^U$2 7GD +G 4 4 4r"c||jvSr,r-r.r5s r has_session_refz%ForwardMsgCache.Entry.has_session_refsd== =r"c"||j|z S)znThe age of the given session's reference to the Entry, given a new script_run_count. r>)r.r5r6s r get_session_ref_agez)ForwardMsgCache.Entry.get_session_ref_ages$d&Eg&NN Nr"c|j|=dSr,r>r?s r remove_session_refz(ForwardMsgCache.Entry.remove_session_refs/888r"c2t|jdkS)zzTrue if this Entry has references from any AppSession. If not, it can be removed from the cache. r)lenr-r3s r has_refszForwardMsgCache.Entry.has_refss t677!; ;r"r5rrN)__name__ __module__ __qualname____doc__rr r/strr4intr<boolr@rBrDrGrr"r Entryr*fs   $ 4 $ $ $ $  $c $ $ $ $ H< H3 HSW H H H H( >< >D > > > > O' O;> O  O O O O 9 9 9 9  r"rc*tj|Sr,r1r3s r r4zForwardMsgCache.__repr__sz$r"rr5rr6NcDt||j|jd}|Xt jdrt |}nt d}||j|j<|||dS)aAdd a ForwardMsg to the cache. The cache will also record a reference to the given AppSession, so that it can track which sessions have already received each given ForwardMsg. Parameters ---------- msg : ForwardMsg session : AppSession script_run_count : int The number of times the session's script has run Nz)global.storeCachedForwardMessagesInMemory) r!rRr8rr get_optionr(rPr<)r.rr5r6entrys r add_messagezForwardMsgCache.add_messages" $$$ !!#(D11 = !LMM 4'--c22'--d33&+DM#( # g'788888r"rcN|j|d}|r|jndS)zReturn the message with the given ID if it exists in the cache. Parameters ---------- hash : str The id of the message to retrieve. Returns ------- ForwardMsg | None N)rRr8r)r.rrVs r get_messagezForwardMsgCache.get_messages, !!$--!+uyyt+r"ct||j|jd}|||sdS|||}|t tjdkS)z6Return True if a session has a reference to a message.NFglobal.maxCachedMessageAge) r!rRr8rr@rBrNr rU)r.rr5r6rVages r has_message_referencez%ForwardMsgCache.has_message_references~ $$$ !!#(D11 = 5 5g > >=5''1ABBc&+,HIIJJJJr"c|jD]K\}}||r|||s|j|=LdS)zRemove refs for all entries for the given session. This should be called when an AppSession is disconnected or closed. Parameters ---------- session : AppSession N)rRcopyitemsr@rDrG)r.r5msg_hashrVs r remove_refs_for_sessionz'ForwardMsgCache.remove_refs_for_sessions $}113399;; , ,OHe$$W-- 2((111>>## ,M(+ , ,r"ctjd}|jD]\}}||s|||}||kr[tdt|||| || s|j|=dS)a8Remove any cached messages that have expired from the given session. This should be called each time a AppSession finishes executing. Parameters ---------- session : AppSession script_run_count : int The number of times the session's script has run r[z4Removing expired entry [session=%s, hash=%s, age=%s]N) r rUrRr_r`r@rBr9debugidrDrG)r.r5r6max_agerarVr\s r "remove_expired_entries_for_sessionz2ForwardMsgCache.remove_expired_entries_for_sessions#$@AA $}113399;; 0 0OHe((11 ++G5EFFCW}} JwKK  ((111~~''0 h/! 0 0r"c8|jdS)z!Remove all entries from the cacheN)rRclearr3s r rizForwardMsgCache.clears r"c g}|jD]J\}}|tdd|j|jndK|S)NForwardMessageCacherr) category_name cache_name byte_length)rRr`appendr rByteSize)r.stats entry_hashrVs r get_statszForwardMsgCache.get_statss!#!%!4!4!6!6   J LL"7!8= 8M 2 2 4 4 4ST     r"rH)rN)rIrJrKrLrPr/rMr4r rNrWrrYrOr]rbrgrirr rsrr"r r(r(Ys|  9<9<9<9<9<9<9<9r|s_FFFFFFFFFFFFFF%%%%%%""""""""''''''555555AAAAAAAA))))))9888888 H  DjZ0NNNNN(NNNNNr"