Current File : //lib/python2.7/site-packages/salt/modules/zk_concurrency.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z y� d d l Z d d l m
Z
d d l Z d d l Z d d l
Z d d l m Z d d l m Z d d l m Z d e j j j f d � � YZ e Z Wn e k
r� e Z n Xd
Z d � Z d d � Z d d d
d e d d d d d d �
Z d d d
d e e d d d d d d � Z d d d
e d d d d d d � Z d d
e d d d d d d � Z! d S( u�
Concurrency controls in zookeeper
=========================================================================
:depends: kazoo
:configuration: See :py:mod:`salt.modules.zookeeper` for setup instructions.
This module allows you to acquire and release a slot. This is primarily useful
for ensureing that no more than N hosts take a specific action at once. This can
also be used to coordinate between masters.
i ( t absolute_importt print_functiont unicode_literalsN( t ForceRetryError( t CancelledError( t NoNodeError( t gethostnamet
_Semaphorec B@ s& e Z d d e d � Z d d � Z RS( i c C@ s� | p t � } t j j j j | | | d | d | �| | _ | j s� y� x� | j j | j � D]w } y] | j j
| j d | � \ } } | | j d � k r� | j d | | _ t
| _ Pn Wqb t k
r� qb Xqb WWq� t k
r� q� Xn d S( Nt
identifiert
max_leasesu /u utf-8( R t kazoot recipet lockt Semaphoret __init__t ephemeral_leaset clientt get_childrent patht gett decodet create_patht Truet is_acquiredR ( t selfR R R R R t childt datat stat( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyR " s( #
c C@ s� | j r t d � � n | j r0 t d � � n | j j | j | j � } t | � | j k r� | j j
| j | j d | j
�n | j j | j � r� t | _ n t | _ | j S( Nu Retry on session loss at topu Semaphore cancelledt ephemeral( t _session_expiredR t cancelledR R R R t _watch_lease_changet lenR t createR R R t existsR R t False( R R t children( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt
_get_lease@ s % N( t __name__t
__module__t NoneR R R% ( ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyR ! s u zk_concurrencyc C@ s t s t d f Si t d <t S( Nu* Module zk_concurrency: dependencies failedu
semaphore_map( t HAS_DEPSR# t __context__t __virtualname__( ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt __virtual__d s
c
@ s� | r d | � n d � d � � f d � } | d d � } | d d � } | d d � } | d d � } | d d � } t | t � r� d
j | � } n | d k r� | d k r� | d k r� d } n d } | r | r | r | d j | | g � f g } n | d k rWt | t � rAg | D] } t d
| � ^ q"} qWt d
| � g } n t j d i � j | pr| t j j d | d | d | � � t d | p�| j
s�t d | p�| j � n t d | p�| S( Nu
zookeeper:u zookeeperc @ s/ � j | � p. t d d j � | g � | � S( uL
look in connection_args first, then default to config file
u
config.getu :( R t __salt__t join( t keyt default( t connection_argst prefix( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyR s s u hostsu 127.0.0.1:2181u schemeu usernameu passwordu default_aclu ,u digestu :u zookeeper.make_digest_aclu zkconnectiont hostst default_aclt auth_data( R( t
isinstancet listR. R- R* t
setdefaultR
R t KazooClientt connectedt start(
t profileR1 R R3 t schemet usernamet passwordR4 R5 t acl( ( R1 R2 s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt _get_zk_connm s6
$ ! &
i c
C@ sx t d | d | d | d | d | d |
� } | t d k rf t | | | d | d | �t d | <n t d | j � S(
uO
Return an un-ordered list of lock holders
path
The path in zookeeper where the lock is
zk_hosts
zookeeper connect string
identifier
Name to identify this minion, if unspecified defaults to hostname
max_concurrency
Maximum number of lock holders
timeout
timeout to wait for the lock. A None timeout will block forever
ephemeral_lease
Whether the locks in zookeper should be ephemeral
Example:
.. code-block: bash
salt minion zk_concurrency.lock_holders /lock/path host1:1234,host2:1234
R< R3 R= R> R? R4 u
semaphore_mapR R ( RA R* R t
lease_holders( R t zk_hostsR t max_concurrencyt timeoutR R< R= R> R? R4 t zk( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt lock_holders� s &c
C@ s� t d | d | d | d | d |
d | � } | t d k rf t | | | d | d | �t d | <n | r� t t d | _ t j t d | _ n | r� t j d
| | � t d | j
d | � n" t j d | � t d | j
� t d | j S(
u�
Get lock (with optional timeout)
path
The path in zookeeper where the lock is
zk_hosts
zookeeper connect string
identifier
Name to identify this minion, if unspecified defaults to the hostname
max_concurrency
Maximum number of lock holders
timeout
timeout to wait for the lock. A None timeout will block forever
ephemeral_lease
Whether the locks in zookeper should be ephemeral
force
Forcibly acquire the lock regardless of available slots
Example:
.. code-block: bash
salt minion zk_concurrency.lock /lock/path host1:1234,host2:1234
R< R3 R= R> R? R4 u
semaphore_mapR R u! Acquiring lock %s with timeout=%sRE u! Acquiring lock %s with no timeout( RA R* R R t assured_patht syst maxintR t loggingt infot acquireR (
R RC R RD RE R t forceR< R= R> R? R4 RF ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyR � s +c
C@ s� t d | d | d | d | d | d | � }
| t d k rf t |
| | d | d | �t d | <n | t d k r� t d | j � t d | =t St j d
| � t Sd S( u;
Remove lease from semaphore
path
The path in zookeeper where the lock is
zk_hosts
zookeeper connect string
identifier
Name to identify this minion, if unspecified defaults to hostname
max_concurrency
Maximum number of lock holders
timeout
timeout to wait for the lock. A None timeout will block forever
ephemeral_lease
Whether the locks in zookeper should be ephemeral
Example:
.. code-block: bash
salt minion zk_concurrency.unlock /lock/path host1:1234,host2:1234
R< R3 R= R> R? R4 u
semaphore_mapR R u Unable to find lease for path %sN( RA R* R t releaseR RK t errorR# ( R RC R RD R R= R< R> R? R4 RF ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt unlock s ( c
C@ s� t d | d | d | d | d | d | � } t j j j | | � }
| r� t j j j | | | � } | j � t j j j | | � }
| j � n t |
� S( u�
Get the List of identifiers in a particular party, optionally waiting for the
specified minimum number of nodes (min_nodes) to appear
path
The path in zookeeper where the lock is
zk_hosts
zookeeper connect string
min_nodes
The minimum number of nodes expected to be present in the party
blocking
The boolean indicating if we need to block until min_nodes are available
Example:
.. code-block: bash
salt minion zk_concurrency.party_members /lock/path host1:1234,host2:1234
salt minion zk_concurrency.party_members /lock/path host1:1234,host2:1234 min_nodes=3 blocking=True
R< R3 R= R> R? R4 (
RA R
R t partyt ShallowPartyt barriert
DoubleBarriert entert leaveR7 ( R RC t min_nodest blockingR< R= R> R? R4 RF RR RT ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt
party_membersC s !
(" t __doc__t
__future__R R R RK RI t kazoo.clientR
t kazoo.retryR t kazoo.recipe.lockt kazoo.recipe.barriert kazoo.recipe.partyt kazoo.exceptionsR R t socketR R R R
R R R) t ImportErrorR# R+ R, R( RA RG RQ RZ ( ( ( s? /usr/lib/python2.7/site-packages/salt/modules/zk_concurrency.pyt <module> sr ;
. & 8 0
Mini Shell