&`icddlZddlmZmZddlmZmZddlmZddl m Z m Z m Z m Z edddgZed d dgZGd d e ZGddeZdS)N)ABCMetaabstractmethod) defaultdict namedtuple)Lock)DictListOptionalTuple StoreStatussuccessversionVersionedValuevalueceZdZdZe ddedeeeefdeeedee de f dZ e dded ed ed ee d e de f dZ ededeeeeee ffe ffdZededeedeeeeee ffe ffdZede fdZdS)StorageaInterface for a storage backend that stores the state of nodes in the cluster. The storage is thread-safe. The storage is versioned, which means that each successful stage change to the storage will bump the version number. The version number can be used to implement optimistic concurrency control. Each entry in the storage table is also versioned. The version number of an entry is the last version number when the entry is updated. Ntablemutationdeletionexpected_storage_versionreturnc td)avBatch update the storage table. This method is atomic. Args: table: The name of the table. mutation: A dictionary of key-value pairs to be updated. deletion: A list of keys to be deleted. expected_storage_version: The expected storage version. The update will fail if the version does not match the current storage version. Returns: StoreStatus: A tuple of (success, version). If the update is successful, returns (True, new_version). Otherwise, returns (False, current_version). z$batch_update() has to be implementedNotImplementedError)selfrrrrs ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/v2/instance_manager/storage.py batch_updatezStorage.batch_updates.""HIIIFkeyrexpected_entry_version insert_onlyc td)aUpdate a single entry in the storage table. Args: table: The name of the table. key: The key of the entry. value: The value of the entry. expected_entry_version: The expected version of the entry. The update will fail if the version does not match the current version of the entry. insert_only: If True, the update will fail if the entry already exists. Returns: StoreStatus: A tuple of (success, version). If the update is successful, returns (True, new_version). Otherwise, returns (False, current_version). zupdate() has to be implementedr)rrrrr r!s rupdatezStorage.update1s2""BCCCrc td)Nzget_all() has to be implementedrrrs rget_allzStorage.get_allLs!"CDDDrkeysc td)a^Get a list of entries from the storage table. Args: table: The name of the table. keys: A list of keys to be retrieved. If the list is empty, all entries in the table will be returned. Returns: Tuple[Dict[str, VersionedValue], int]: A tuple of (entries, storage_version). The entries is a dictionary of (key, (value, entry_version)) pairs. The entry_version is the version of the entry when it was last updated. The storage_version is the current storage version. zget() has to be implementedr)rrr's rgetz Storage.getPs$""?@@@rc td)ziGet the current storage version. Returns: int: The current storage version. z#get_version() has to be implementedrrs r get_versionzStorage.get_versionds""GHHHrNNN)NF)__name__ __module__ __qualname____doc__rstrr rr intr rboolr#r r&r)r,rrrr s  .2(,26 JJJ4S>*J49% J #+3- J  JJJ^J0 15! DDDD D !) D  D DDD^D4ESEU4U38_0D+Es+J%KEEE^EAA $S A tCsCx()3. /AAA^A&ISIII^IIIrr) metaclassc$eZdZdZdZ ddedeeefdeedee de f d Z dded ed ed ee dee de de fdZ dedeeeefe ffdZdedeedeeeefe ffdZde fdZdS)InMemoryStoragez\An in-memory implementation of the Storage interface. This implementation is not durablecld|_tt|_t |_dS)Nr)_versionrdict_tablesr_lockr+s r__init__zInMemoryStorage.__init__rs' "4(( VV rNrrrexpected_versionrc|r|ni}|r|ng}j5|,|jkr!tdjcdddSxjdz c_fd|D}j|||D]#}j||d$tdjcdddS#1swxYwYdS)NFcBi|]\}}|t|jSr5)rr:).0rrrs r z0InMemoryStorage.batch_update..s;,,,C^E4=99,,,rT)r=r:r itemsr<r#pop)rrrrr?key_value_pairs_with_version deleted_keys` rrzInMemoryStorage.batch_updatewsx (/88R'/88R Z 4 4+0@DM0Q0Q"5$-88 4 4 4 4 4 4 4 4 MMQ MM,,,,"*.."2"2,,, ( L  & &'C D D D' ; ;  U#'' T::::tT]33 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s"CB CC"CFrrr rr!c.|j5|,||jkr!td|jcdddS|r0||j|vr!td|jcdddS|j||d\}}|'||kr!td|jcdddS|xjdz c_t ||j|j||<td|jcdddS#1swxYwYdS)NF)NrAT)r=r:r r<r)r) rrrrr rr!_rs rr#zInMemoryStorage.updatesZ 4 4(4, =="5$-88 4 4 4 4 4 4 4 4  9sdl5&999"5$-88 4 4 4 4 4 4 4 4e,00jAAJAw%16LPW6W6W"5$-88 4 4 4 4 4 4 4 4 MMQ MM'5eT]'K'KDL  $tT]33 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s$"D %D )AD 6AD  DDc|j5tj|j||jfcdddS#1swxYwYdSN)r=copydeepcopyr<r:r%s rr&zInMemoryStorage.get_alls Z G GM$,u"566 F G G G G G G G G G G G G G G G G G Gs &;??r'c|s||S|j5i}|D]5}||j|ivr|j||||<6t ||jcdddS#1swxYwYdSrM)r&r=r<r)r r:)rrr'resultrs rr)zInMemoryStorage.gets '<<&& & Z 6 6F ; ;$,**5"5555"&,u"5c":F3Kvt}55  6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6sAA;;A?A?c|jSrM)r:r+s rr,zInMemoryStorage.get_versions }rr-)NNF)r.r/r0r1r>r2rr r r3r rr4r#r rr&r)r,r5rrr8r8ns$("*. 444sCx.4s) 4 #3- 4  444461526!4444 4 !) 4 #+3- 44 44440GSGU4^0C+Dc+I%JGGGG66DI6%S.=P8QSV8V2W6666Srr8)rNabcrr collectionsrr threadingrtypingrr r r r rrr8r5rrrWs! ''''''''////////............jI(>?? ,w .BCC`I`I`I`I`I`I`I`I`IFFFFFFgFFFFFr