g-dZddlmZmZddlmZgdZGddeZGdd eZGd d eZ d S) z Tools for Advisory Locks )abstractmethodabstractproperty)Element)ALock ExclusiveLock ShareLockceZdZdZdZdZdjfdZedZ edZ e dZ d e fd Zd Zd Zd ZdZedZdZdS)ra Advisory Lock class for managing the acquisition and release of a sequence of PostgreSQL advisory locks. ALock()'s are fairly consistent with threading.RLock()'s. They can be acquired multiple times, and they must be released the same number of times for the lock to actually be released. A notably difference is that ALock's manage a sequence of lock identifiers. This means that a given ALock() may represent multiple advisory locks. )database identifiersALOCKz {1} [{0}]c#>Kd||j|jfVdSN)statemode)selfheadfmts F/granary/edev/quantum/lib/python3.11/site-packages/postgresql/alock.py_e_metaszALock._e_metass/ ggdj$),,,,,,,,cdS)z! The mode of the lock class. Nrs rrz ALock.mode#rcdS)a Implemented by subclasses to return the statements to try, acquire, and release the advisory lock. Returns a triple of callables where each callable takes two arguments, the lock-id pairs, and then the int8 lock-ids. ``(try, acquire, release)``. Nrrs r__select_statements__zALock.__select_statements__)rrc:d|D}d|D}||fS)NcPg|]#}|jturt|nddg$Sr) __class__intlist.0xs r z1ALock._split_lock_identifiers..7sBk$$47774+rc2g|]}|jtur|ndSr)rr r"s rr%z1ALock._split_lock_identifiers..;s8    s11   rr)idseqid_pairsidss r_split_lock_identifierszALock._split_lock_identifiers4sJ (      # CrTc|jdkrt|}|r!||j|jn||j|j}dt ||jD}||||jkr3||j | |dS|jdz|_dS)z9 Acquire the locks using the configured identifiers. rc(g|]\}}|d |S)rr)r#didlockids rr%z!ALock.acquire..Ps5 GRwqzrFrT) _countbool_acquire _id_pairs_ids_tryzipr reverse_releaser*)rblockinglenwaitr release_seqs racquirez ALock.acquireAs [A x..4 MM$.$),,,, $.$),,Aq$"233K s;33t/0000T]D00==>> E a$+ rc.||Sr)r=rs r __enter__zALock.__enter__as,,... +rc|jdkrtd|jjsE|jdkr:|t |jt |j|jdz |_dS)z9 Release the locks using the configured identifiers. rzcannot release un-acquired lockrN)r/ RuntimeErrorr closedr7reversedr2r3rs rreleasez ALock.releaseesq [1__ 7 8 88  @$+//==$.))8DI+>+>??? a$+++rc.|dSr)rD)rtypvaltbs r__exit__zALock.__exit__rs,,.....rc2|jdko |jj S)z Whether the locks have been acquired. This method is sensitive to the connection's state. If the connection is closed, it will return False. r)r/r rBrs rlockedz ALock.lockedus +/ 9DM$8 89rc2|rdndS)NrKunlocked)rKrs rrz ALock.state|s[[]] 2 2rcd|_|x|_|_||_||\|_|_|\|_|_ |_ dS)zj Initialize the lock object to manage a sequence of advisory locks for use with the given database. rN) r/ connectionr r r*r2r3rr4r1r7)rr r s r__init__zALock.__init__s_ $+$,,$/DM $"::;GG$.$),0,F,F,H,H)$)T]DMMMrN)__name__ __module__ __qualname____doc__ _e_factors_e_labelformatrrrr staticmethodr*r9r=r?rDrIrKpropertyrrPrrrrrs  +   ----    , ##@    ::: 33 (3 I I I I IrrceZdZdZdZdS)r sharecf|jjj|jjj|jjjfSr)r systry_advisory_sharedacquire_advisory_sharedrelease_advisory_sharedrs rrzShareLock.__select_statements__s-=(=,=, rNrQrRrSrrrrrr r s(rr ceZdZdZdZdS)r exclusivecf|jjj|jjj|jjjfSr)r r]try_advisory_exclusiveacquire_advisory_exclusiverelease_advisory_exclusivers rrz#ExclusiveLock.__select_statements__s-=+=/=/ rNrarrrrrs(rrN) rTabcrrpython.elementr__all__rr rrrrrks10000000######    yIyIyIyIyIGyIyIyIvEr