Current File : //usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d
Z d Z d � Z d d d d e d d d d d d �
Z
d d d e d d d d d d
� Z e d d d d d d � Z d S( u
Control concurrency of steps within state execution using zookeeper
===================================================================
:depends: kazoo
:configuration: See :py:mod:`salt.modules.zookeeper` for setup instructions.
This module allows you to "wrap" a state's execution with concurrency control.
This is useful to protect against all hosts executing highstate simultaneously
if your services don't all HUP restart. The common way of protecting against this
is to run in batch mode, but that doesn't protect from another person running
the same batch command (and thereby having 2x the number of nodes deploying at once).
This module will bock while acquiring a slot, meaning that however the command gets
called it will coordinate with zookeeper to ensure that no more than max_concurrency
steps are executing with a single path.
.. code-block:: yaml
acquire_lock:
zk_concurrency.lock:
- name: /trafficeserver
- zk_hosts: 'zookeeper:2181'
- max_concurrency: 4
- prereq:
- service: trafficserver
trafficserver:
service.running:
- watch:
- file: /etc/trafficserver/records.config
/etc/trafficserver/records.config:
file.managed:
- source: salt://records.config
release_lock:
zk_concurrency.unlock:
- name: /trafficserver
- require:
- service: trafficserver
This example would allow the file state to change, but would limit the
concurrency of the trafficserver service restart to 4.
i ( t absolute_importt print_functiont unicode_literalsu zk_concurrency.locku zk_concurrency.unlocku zk_concurrency.party_membersu zk_concurrencyc C@ s t d � t D� � s t St S( Nc s@ s | ] } | t k Vq d S( N( t __salt__( t .0t func( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pys <genexpr>? s ( t allt REQUIRED_FUNCSt Falset __virtualname__( ( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyt __virtual__> s i c C@ s� i | d 6i d 6t d 6d d 6} i | d 6| d 6| d 6| d 6|
d
6} t d rm d | d <d | d <| S| d k r� t d
} n t d | | d | d | d | d | | �}
|
r� t | d <d | d <n
d | d <| S( uX
Block state execution until you are able to get the lock (or hit the timeout)
u nameu changesu resultu u commentu profileu schemeu usernameu passwordu default_aclu testu Attempt to acquire locku idu zk_concurrency.lockt
identifiert max_concurrencyt timeoutt ephemeral_leaseu
lock acquiredu Unable to acquire lockN( R t __opts__t Nonet
__grains__R t True( t namet zk_hostsR R R
R t profilet schemet usernamet passwordt default_aclt rett conn_kwargst locked( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyt lockE s0
c
C@ s� i | d 6i d 6t d 6d d 6}
i | d 6| d 6| d 6| d 6| d
6} t d rm d |
d <d |
d <|
S| d k r� t d
} n t d | d | d | d | d | | �} | r� t |
d <n d j | � |
d <|
S( u&
Remove lease from semaphore.
u nameu changesu resultu u commentu profileu schemeu usernameu passwordu default_aclu testu Released lock if it is hereu idu zk_concurrency.unlockR R R R u! Unable to find lease for path {0}N( R R R R R R t format(
R R R R R R R R R R R R t unlocked( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyt unlocks s,
c
C@ sG i | d 6i d 6t d 6d d 6} i | d 6| d 6| d 6| d 6| d
6}
t d rm d | d <d | d <| St d
| | | d | |
�} t | t � s� t d j | � � � n t | � } | | k s� | r#t d r� d n t | d <| s
d j | | � | d <qCd j | | � | d <n t | d <d j | | � | d <| S( ui
Ensure that there are `min_nodes` in the party at `name`, optionally blocking if not available.
u nameu changesu resultu u commentu profileu schemeu usernameu passwordu default_aclu testu Attempt to ensure min_partyu zk_concurrency.party_memberst blockinguC Error from zk_concurrency.party_members, return was not a list: {0}u$ Currently {0} nodes, which is >= {1}uR Blocked until {0} nodes were available. Unblocked after {1} nodes became availableu# Currently {0} nodes, which is < {1}N(
R R R R t
isinstancet listt ExceptionR t lenR (
R R t min_nodesR! R R R R R R R t nodest num_nodes( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyt min_party� s,
N( u zk_concurrency.locku zk_concurrency.unlocku zk_concurrency.party_members(
t __doc__t
__future__R R R R R R
R R R R R) ( ( ( s> /usr/lib/python2.7/site-packages/salt/states/zk_concurrency.pyt <module>/ s> % %
Mini Shell