File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/__pycache__/zk_concurrency.cpython-310.pyc
Back
o ;j�, � @ s d Z ddlZddlZz3ddlmZ ddlZddlZddlZddl Zddl m Z m Z ddl mZ G dd� dejjj�ZdZW n eyH d ZY nw d Zd d � Zdd d�Z ddd�Z ddd�Z ddd�Z d dd�ZdS ) a� 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. � N)� gethostname)�CancelledError� NoNodeError)�ForceRetryErrorc @ s& e Zd Z ddd�Zd dd�ZdS ) � _SemaphoreN� Tc C s� |pt � }tjjjj| ||||d� || _| js_z<| j�| j �D ]1}z&| j� | j d | �\}}||� d�krG| j d | | _ d| _ W W d S W q tyQ Y q w W d S ty^ Y d S w d S )N)� identifier� max_leases�/zutf-8T)r �kazoo�recipe�lock� Semaphore�__init__�ephemeral_lease�client� get_children�path�get�decode� create_path� is_acquiredr ) �selfr r r r r �child�data�stat� r �O/opt/saltstack/salt/lib/python3.10/site-packages/salt/modules/zk_concurrency.pyr s. � � �� ��z_Semaphore.__init__c C sz | j rtd��| jrtd��| j�| j| j�}t|�| j k r*| jj | j | j | j d� | j�| j �r7d| _| jS d| _| jS )Nz Retry on session loss at topzSemaphore cancelled)Z ephemeralTF)Z_session_expiredr Z cancelledr r r r Z_watch_lease_change�lenr �creater r r �existsr )r r �childrenr r r � _get_lease: s ��z_Semaphore._get_lease)Nr T�N)�__name__� __module__� __qualname__r r" r r r r r s � r TFZzk_concurrencyc C s t sdS i td< tS )N)Fz*Module zk_concurrency: dependencies failed� semaphore_map)�HAS_DEPS� __context__�__virtualname__r r r r � __virtual___ s r+ c s: | rd| �nd�d� �fdd� }|dd�}|dd �}|dd �}|d d �}|d d �}t |t�r4d �|�}|d urB|d urB|d u rBd }d }|rT|rT|rT|d �||g�fg}|d urot |t�redd� |D �}n td di |��g}t�di ��| px|tjj|||d�� td | p�| j s�td | p�| � � td | p�| S )Nz zookeeper:Z zookeeperc s" � � | �ptd d��| g�|�S )zL look in connection_args first, then default to config file z config.get�:)r �__salt__�join)�key�default��connection_args�prefixr r r n s �z_get_zk_conn.<locals>.get�hostsz127.0.0.1:2181�scheme�username�password� default_acl�,�digestr, c S s g | ] }t d di |���qS )�zookeeper.make_digest_aclr )r- )�.0Zaclr r r � <listcomp>� s �z _get_zk_conn.<locals>.<listcomp>r; Z zkconnection)r4 r8 � auth_datar# r ) � isinstance�listr. r- r) � setdefaultr r Z KazooClient� connected�start) �profiler2 r r4 r5 r6 r7 r8 r> r r1 r � _get_zk_connh s> � ��rE r c C sJ t ||||| | d�} | td vr t| | |||d�td | < td | �� S )aP 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 �rD r4 r5 r6 r7 r8 r' �r r )rE r) r Z lease_holders) r �zk_hostsr �max_concurrency�timeoutr rD r5 r6 r7 r8 �zkr r r � lock_holders� s" (� �rL c C s� t |||| | | d�} | td vr t| | |||d�td | < |r.dtd | _tjtd | _|rBt�d| |� td | j |d� nt�d| � td | � � td | j S )a� 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 rF r' rG Tz!Acquiring lock %s with timeout=%s)rJ z!Acquiring lock %s with no timeout) rE r) r Z assured_path�sysZmaxintr �logging�info�acquirer ) r rH r rI rJ r �forcerD r5 r6 r7 r8 rK r r r r � s2 ,� � r c C st t |||||| d�} | td vr t| | |||d�td | < | td v r2td | �� td | = dS t�d| � dS )a< 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 rF r' rG Tz Unable to find lease for path %sF)rE r) r �releaserN �error) r rH r rI r r5 rD r6 r7 r8 rK r r r �unlock# s, )� � rT c C sb t ||||||d�} tjj�| | �} |r-tjj�| | |�} | �� tjj�| | �} | �� t | �S )a� 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 rF ) rE r r �partyZ ShallowParty�barrierZ DoubleBarrierZenterZleaver@ ) r rH Z min_nodes�blockingrD r5 r6 r7 r8 rK rU rV r r r � party_membersf s "�rX r# ) NNr NFNNNNN) NNr NFFNNNNN) NNr FNNNNN)Nr FNNNNN)�__doc__rN rM �socketr Z kazoo.clientr Zkazoo.recipe.barrierZkazoo.recipe.lockZkazoo.recipe.partyZkazoo.exceptionsr r Z kazoo.retryr r r r r r( � ImportErrorr* r+ rE rL rT rX r r r r �<module> sz ; � 5 �= �O �E�
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings