B 4\@sddlmZmZmZddlmZmZddlmZddl Z ddl Z ddl m Z ddl Z ddlmZddlmZdd lmZe eZGd d d eZGd d d eZdS))print_functiondivisionabsolute_import) defaultdictdeque) timedeltaN)gen)_get_global_client) log_errors) get_workerc@s2eZdZdZddZejd ddZd ddZdS) LockExtensionz An extension for the scheduler to manage Locks This adds the following routes to the scheduler * lock_acquire * lock_release cCs@||_tt|_t|_|jj|j|j d||jj d<dS)N) lock_acquire lock_releaselocks) schedulerrreventsdictidsZhandlersupdateacquirerelease extensions)selfrr/lib/python3.7/site-packages/distributed/lock.py__init__s    zLockExtension.__init__Nc cstt|trt|}||jkr*d}nx||jkrtj}|j| || }|dk rrt t |d|}z0y |VWnt jk rd}PYnXd}Wd|j|}||kstXq,W|r||jkst||j|<t |WdQRXdS)NT)ZsecondsF)r isinstancelisttuplertornadorZEventrappendwaitrZ with_timeoutr TimeoutErrorpopleftAssertionErrorReturn) rstreamnameidtimeoutresultZeventZfutureZevent2rrrr%s.      zLockExtension.acquirec Csvtft|trt|}|j||kr2td|j|=|j|r`|jj |j|dj n|j|=WdQRXdS)Nz#This lock has not yet been acquiredr) r rrrrget ValueErrorrrZloopZ add_callbackset)rr'r(r)rrrrBs  zLockExtension.release)NNNN)NNN) __name__ __module__ __qualname____doc__rr coroutinerrrrrrr s   r c@sheZdZdZdddZdddZdd Zd d Zd d ZddZ e j ddZ e j ddZ ddZdS)Locka Distributed Centralized Lock Parameters ---------- name: string Name of the lock to acquire. Choosing the same name allows two disconnected processes to coordinate a lock. Examples -------- >>> lock = Lock('x') # doctest: +SKIP >>> lock.acquire(timeout=1) # doctest: +SKIP >>> # do things with protected resource >>> lock.release() # doctest: +SKIP NcCs>|ptptj|_|p$dtj|_tj|_d|_dS)Nzlock-F) r r clientuuidZuuid4hexr(r)_locked)rr(r5rrrr_s z Lock.__init__TcCs@|s|dk rtdd}|jj|jjj|j|j|d}d|_|S)a Acquire the lock Parameters ---------- blocking : bool, optional If false, don't wait on the lock in the scheduler at all. timeout : number, optional Seconds to wait on the lock in the scheduler. This does not include local coroutine time, network transfer time, etc.. It is forbidden to specify a timeout when blocking is false. Examples -------- >>> lock = Lock('x') # doctest: +SKIP >>> lock.acquire(timeout=1) # doctest: +SKIP Returns ------- True or False whether or not it sucessfully acquired the lock Nz/can't specify a timeout for a non-blocking callr)r(r)r*T)r-r5syncrrr(r)r8)rZblockingr*r+rrrresz Lock.acquirecCs6|std|jj|jjj|j|jd}d|_|S)z& Release the lock if already acquired zLock is not yet acquired)r(r)F) lockedr-r5r9rrr(r)r8)rr+rrrrs z Lock.releasecCs|jS)N)r8)rrrrr:sz Lock.lockedcCs ||S)N)r)rrrr __enter__szLock.__enter__cOs |dS)N)r)rargskwargsrrr__exit__sz Lock.__exit__ccs|Vt|dS)N)rrr&)rrrr __aenter__s zLock.__aenter__cos|VdS)N)r)rr<r=rrr __aexit__szLock.__aexit__cCs t|jffS)N)r4r()rrrr __reduce__szLock.__reduce__)NN)TN)r/r0r1r2rrrr:r;r>rr3r?r@rArrrrr4Os   r4)Z __future__rrr collectionsrrZdatetimerZloggingr6r rZ tornado.locksr5r Zutilsr Zworkerr Z getLoggerr/Zloggerobjectr r4rrrrs      =