Root Zanli
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
opt
/
alt
/
python27
/
lib
/
python2.7
/
site-packages
/
redis
/
Filename :
lock.pyc
back
Copy
� ���fc @ sd d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d e f d � � YZ d S( i����N( t LockErrort LockNotOwnedError( t dummyt Lockc B s� e Z d Z d Z d Z d Z d Z d Z d Z d d e d e d � Z d � Z d � Z d � Z d d d d � Z d � Z d � Z d � Z d � Z d � Z e d � Z d � Z d � Z d � Z RS( s� A shared, distributed Lock. Using Redis for locking allows the Lock to be shared across processes and/or machines. It's left to the user to resolve deadlock issues and make sure multiple clients play nicely together. s� local token = redis.call('get', KEYS[1]) if not token or token ~= ARGV[1] then return 0 end redis.call('del', KEYS[1]) return 1 s� local token = redis.call('get', KEYS[1]) if not token or token ~= ARGV[1] then return 0 end local expiration = redis.call('pttl', KEYS[1]) if not expiration then expiration = 0 end if expiration < 0 then return 0 end local newttl = ARGV[2] if ARGV[3] == "0" then newttl = ARGV[2] + expiration end redis.call('pexpire', KEYS[1], newttl) return 1 s� local token = redis.call('get', KEYS[1]) if not token or token ~= ARGV[1] then return 0 end redis.call('pexpire', KEYS[1], ARGV[2]) return 1 g�������?c C s� | | _ | | _ | | _ | | _ | | _ | | _ t | � | _ | j rZ t j � n t � | _ d | j _ | j � d S( s� Create a new Lock instance named ``name`` using the Redis client supplied by ``redis``. ``timeout`` indicates a maximum life for the lock. By default, it will remain locked until release() is called. ``timeout`` can be specified as a float or integer, both representing the number of seconds to wait. ``sleep`` indicates the amount of time to sleep per loop iteration when the lock is in blocking mode and another client is currently holding the lock. ``blocking`` indicates whether calling ``acquire`` should block until the lock has been acquired or to fail immediately, causing ``acquire`` to return False and the lock not being acquired. Defaults to True. Note this value can be overridden by passing a ``blocking`` argument to ``acquire``. ``blocking_timeout`` indicates the maximum amount of time in seconds to spend trying to acquire the lock. A value of ``None`` indicates continue trying forever. ``blocking_timeout`` can be specified as a float or integer, both representing the number of seconds to wait. ``thread_local`` indicates whether the lock token is placed in thread-local storage. By default, the token is placed in thread local storage so that a thread only sees its token, not a token set by another thread. Consider the following timeline: time: 0, thread-1 acquires `my-lock`, with a timeout of 5 seconds. thread-1 sets the token to "abc" time: 1, thread-2 blocks trying to acquire `my-lock` using the Lock instance. time: 5, thread-1 has not yet completed. redis expires the lock key. time: 5, thread-2 acquired `my-lock` now that it's available. thread-2 sets the token to "xyz" time: 6, thread-1 finishes its work and calls release(). if the token is *not* stored in thread local storage, then thread-1 would see the token value as "xyz" and would be able to successfully release the thread-2's lock. In some use cases it's necessary to disable thread local storage. For example, if you have code where one thread acquires a lock and passes that lock instance to a worker thread to release later. If thread local storage isn't disabled in this case, the worker thread won't see the token set by the thread that acquired the lock. Our assumption is that these cases aren't common and as such default to using thread local storage. N( t redist namet timeoutt sleept blockingt blocking_timeoutt boolt thread_localt threadingt localR t Nonet tokent register_scripts( t selfR R R R R R R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt __init__I s 4 ! c C s� | j } | j } | j d k r9 | j | j � | _ n | j d k r` | j | j � | _ n | j d k r� | j | j � | _ n d S( N( t __class__R t lua_releaseR t register_scriptt LUA_RELEASE_SCRIPTt lua_extendt LUA_EXTEND_SCRIPTt lua_reacquiret LUA_REACQUIRE_SCRIPT( R t clst client( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR � s c C s& | j d t � r | St d � � d S( NR s0 Unable to acquire lock within the time specified( t acquiret TrueR ( R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt __enter__� s c C s | j � d S( N( t release( R t exc_typet exc_valuet traceback( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt __exit__� s c C s | j } | d k r- t j � j j � } n! | j j j � } | j | � } | d k rf | j } n | d k r~ | j } n d } | d k r� t j � | } n xl t r| j | � r� | | j _ t S| s� t St j � | } | d k r| | k rt St j | � q� Wd S( ss Use Redis to hold a shared, distributed lock named ``name``. Returns True once the lock is acquired. If ``blocking`` is False, always return immediately. If the lock was acquired, return True, otherwise return False. ``blocking_timeout`` specifies the maximum number of seconds to wait trying to acquire the lock. ``token`` specifies the token value to be used. If provided, token must be a bytes object or a string that can be encoded to a bytes object with the default encoding. If a token isn't specified, a UUID will be generated. N( R R t uuidt uuid1t hext encodeR t connection_poolt get_encoderR R t mod_timet timeR t do_acquireR R t False( R R R R R t encodert stop_trying_att next_try_at( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR � s, c C sQ | j r t | j d � } n d } | j j | j | d t d | �rM t St S( Ni� t nxt px( R t intR R t setR R R. ( R R R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR- � s $c C s | j j | j � d k S( sU Returns True if this key is locked by any process, otherwise False. N( R t getR R ( R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt locked� s c C sq | j j | j � } | rO t | t � rO | j j j � } | j | � } n | j j d k op | | j j k S( sS Returns True if this key is locked by this lock, otherwise False. N( R R6 R t isinstancet bytesR) R* R( R R R ( R t stored_tokenR/ ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt owned� s c C sD | j j } | d k r' t d � � n d | j _ | j | � d S( s" Releases the already acquired locks Cannot release an unlocked lockN( R R R R t do_release( R t expected_token( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR � s c C sC t | j d | j g d | g d | j � � s? t d � � n d S( Nt keyst argsR s, Cannot release a lock that's no longer owned( R R R R R ( R R= ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR< � s c C sO | j j d k r! t d � � n | j d k r? t d � � n | j | | � S( sh Adds more time to an already acquired lock. ``additional_time`` can be specified as an integer or a float, both representing the number of seconds to add. ``replace_ttl`` if False (the default), add `additional_time` to the lock's existing ttl. If True, replace the lock's ttl with `additional_time`. s Cannot extend an unlocked locks$ Cannot extend a lock with no timeoutN( R R R R R t do_extend( R t additional_timet replace_ttl( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt extend� s c C sk t | d � } t | j d | j g d | j j | | r@ d pC d g d | j � � sg t d � � n t S( Ni� R> R? t 1t 0R s+ Cannot extend a lock that's no longer owned( R4 R R R R R R R R ( R RA RB ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR@ s c C sI | j j d k r! t d � � n | j d k r? t d � � n | j � S( sS Resets a TTL of an already acquired lock back to a timeout value. s! Cannot reacquire an unlocked locks' Cannot reacquire a lock with no timeoutN( R R R R R t do_reacquire( R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt reacquire s c C s_ t | j d � } t | j d | j g d | j j | g d | j � � s[ t d � � n t S( Ni� R> R? R s. Cannot reacquire a lock that's no longer owned( R4 R R R R R R R R R ( R R ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyRF s N( t __name__t __module__t __doc__R R R R R R R R R R R R$ R R- R7 R; R R< R. RC R@ RG RF ( ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyR s, > ( ( R R, R+ R% t redis.exceptionsR R t redis.utilsR t objectR ( ( ( s; /opt/alt/python27/lib/python2.7/site-packages/redis/lock.pyt <module> s