File: //usr/local/lib/python3.9/site-packages/kombu/asynchronous/__pycache__/semaphore.cpython-39.pyc
a
X>h�
� @ s� d Z ddlmZ ddlZddlmZ ddlmZ er|ddlm Z ddlm
Z
m Z ej dk rhdd l
mZ n dd lmZ ed
�Zd ZG d d
� d
�ZG dd� d�ZdS )z&Semaphores and concurrency primitives.� )� annotationsN)�deque)�
TYPE_CHECKING)�
TracebackType)�Callable�Deque)� �
)� ParamSpec�P)� DummyLock�LaxBoundedSemaphorec @ s� e Zd ZdZddd�dd�Zddd d
d �d d
�Zdd�dd�Zd ddd�dd�Zd ddd�dd�Zdd�dd�Z dd�dd�Z
d S ) r
a� Asynchronous Bounded Semaphore.
Lax means that the value will stay within the specified
range even if released more times than it was acquired.
Example:
-------
>>> x = LaxBoundedSemaphore(2)
>>> x.acquire(print, 'HELLO 1')
HELLO 1
>>> x.acquire(print, 'HELLO 2')
HELLO 2
>>> x.acquire(print, 'HELLO 3')
>>> x._waiters # private, do not access directly
[print, ('HELLO 3',)]
>>> x.release()
HELLO 3
�int�None)�value�returnc C s, | | _ | _t� | _| jj| _| jj| _d S �N)�
initial_valuer r �_waiting�append� _add_waiter�popleft� _pop_waiter)�selfr � r �F/usr/local/lib/python3.9/site-packages/kombu/asynchronous/semaphore.py�__init__- s
z LaxBoundedSemaphore.__init__zCallable[P, None]zP.argszP.kwargs�bool)�callback� partial_args�partial_kwargsr c O sH | j }|dkr"| �|||f� dS t|d d�| _ ||i |�� dS dS )a^ Acquire semaphore.
This will immediately apply ``callback`` if
the resource is available, otherwise the callback is suspended
until the semaphore is released.
Arguments:
---------
callback (Callable): The callback to apply.
*partial_args (Any): partial arguments to callback.
r F� TN)r r �max)r r r r r r r r �acquire3 s zLaxBoundedSemaphore.acquire�r c C sL z| � � \}}}W n& ty8 t| jd | j�| _Y n0 ||i |�� dS )z�Release semaphore.
Note:
----
If there are any waiters this will apply the first waiter
that is waiting for the resource (FIFO order).
r! N)r �
IndexError�minr r )r �waiter�args�kwargsr r r �releaseM s
zLaxBoundedSemaphore.releaser! )�nr c C s6 | j |7 _ | j|7 _t|�D ] }| �� q$dS )z6Change the size of the semaphore to accept more users.N)r r �ranger* )r r+ �_r r r �grow\ s zLaxBoundedSemaphore.growc C s( t | j| d�| _t | j| d�| _dS )z6Change the size of the semaphore to accept less users.r N)r" r r )r r+ r r r �shrinkc s zLaxBoundedSemaphore.shrinkc C s | j �� | j| _dS )z@Reset the semaphore, which also wipes out any waiting callbacks.N)r �clearr r �r r r r r0 h s
zLaxBoundedSemaphore.clear�strc C s d� | jjt| �| jt| j��S )Nz!<{} at {:#x} value:{} waiting:{}>)�format� __class__�__name__�idr �lenr r1 r r r �__repr__m s �z LaxBoundedSemaphore.__repr__N)r! )r! ) r5 �
__module__� __qualname__�__doc__r r# r* r. r/ r0 r8 r r r r r
s r
c @ s2 e Zd ZdZd d�dd�Zddddd �d
d �Zd S )
r zPretending to be a lock.r$ c C s | S r r r1 r r r � __enter__v s zDummyLock.__enter__ztype[BaseException] | NonezBaseException | NonezTracebackType | Noner )�exc_type�exc_val�exc_tbr c C s d S r r )r r= r> r? r r r �__exit__y s zDummyLock.__exit__N)r5 r9 r: r; r< r@ r r r r r s s r )r; �
__future__r �sys� collectionsr �typingr �typesr r r � version_infoZtyping_extensionsr
r �__all__r
r r r r r �<module> s
^