File: //usr/local/lib/python3.9/site-packages/kombu/__pycache__/resource.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddlmZ ddl
m Z dd l m
Z
dd
lmZ dd lmZ e r|dd lmZ d
d� ZG dd� de �ZG dd� d�ZdS )z%Generic resource pool implementation.� )� annotationsN)�deque)�Empty)� LifoQueue)�
TYPE_CHECKING� )�
exceptions)�register_after_fork)�lazy)�
TracebackTypec C s$ z | � � W n ty Y n0 d S �N)�force_close_all� Exception)�resource� r �8/usr/local/lib/python3.9/site-packages/kombu/resource.py� _after_fork_cleanup_resource s r c @ s e Zd ZdZdd� ZdS )r z#Last in first out version of Queue.c C s t � | _d S r )r �queue)�self�maxsizer r r �_init s zLifoQueue._initN)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ s� e Zd ZdZejZdZd&dd�Zdd� Zdd � Z d'd
d �Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd(dd�Zd)dd �Zd*d d �Zed d � �Zejd!d � �Zej�d"�r�e
ZeZd#Z d$d � Z
d%d� ZdS )+�ResourcezPool of resources.FNc C s^ || _ |p d| _d| _|d ur"|n| j| _t� | _t� | _| jrRtd urRt| t � | �
� d S )Nr F) �_limit�preload�_closed�close_after_forkr � _resource�set�_dirtyr r �setup)r �limitr r r r r �__init__( s
��
zResource.__init__c C s t d��d S )Nzsubclass responsibility)�NotImplementedError�r r r r r# 7 s zResource.setupc C s6 | j r"t| j�| j kr"| �| j ��| j�| �� � d S r )r$ �lenr" �
LimitExceededr �
put_nowait�newr' r r r �_add_when_empty: s zResource._add_when_emptyc s� �j rtd���jr�z�jj||d�� W n tyB ��� Y q0 z��� �� W n6 ty� t � t
�rx�j� � � n
�� � � � Y n0 �j
�� � q�qn����� �� � �fdd�}|� _ � S )a� Acquire resource.
Arguments:
---------
block (bool): If the limit is exceeded,
then block until there is an available item.
timeout (float): Timeout to wait
if ``block`` is true. Default is :const:`None` (forever).
Raises
------
LimitExceeded: if block is false and the limit has been exceeded.
zAcquire on closed pool)�block�timeoutc s �� � � dS )a' Release resource so it can be used by another thread.
Warnings:
--------
The caller is responsible for discarding the object,
and to never use the resource again. A new resource must
be acquired if so needed.
N)�releaser ��Rr r r r/ h s z!Resource.acquire.<locals>.release)r � RuntimeErrorr$ r �getr r, �prepare�
BaseException�
isinstancer
r* r/ r" �addr+ )r r- r. r/ r r0 r �acquireB s(
zResource.acquirec C s |S r r �r r r r r r4 v s zResource.preparec C s |� � d S r )�closer9 r r r �close_resourcey s zResource.close_resourcec C s d S r r r9 r r r �release_resource| s zResource.release_resourcec C s | j r| j�|� | �|� dS )zqReplace existing resource with a new instance.
This can be used in case of defective resources.
N)r$ r" �discardr; r9 r r r �replace s zResource.replacec C s8 | j r*| j�|� | j�|� | �|� n
| �|� d S r )r$ r" r= r r* r< r; r9 r r r r/ � s
zResource.releasec C s d S r r r9 r r r �collect_resource� s zResource.collect_resourceTc C s� | j r
dS || _ | j}| j}z |�� }W n ty> Y qdY n0 z| �|� W q ty` Y q 0 q z|j�� }W n ty� Y q�Y n0 z| �|� W qd ty� Y qd0 qddS )aa Close and remove all resources in the pool (also those in use).
Used to close resources from parent processes after fork
(e.g. sockets/connections).
Arguments:
---------
close_pool (bool): If True (default) then the pool is marked
as closed. In case of False the pool can be reused.
N) r r" r �pop�KeyErrorr? �AttributeErrorr �
IndexError)r �
close_poolZdirtyr Zdres�resr r r r
� s*
zResource.force_close_allc C s� | j }| jrDd| k r"| j k rDn n |sD|s@td�| j |���d}|| _ |rrz| jdd� W n typ Y n0 | �� ||k r�| j|dkd� d S )Nr z,Can't shrink pool when in use: was={} now={}TF)rD )�collect)r r" r2 �formatr
r r# � _shrink_down)r r$ �force�
ignore_errors�resetZ
prev_limitr r r �resize� s$ $�� zResource.resizec C s� G dd� d�}| j }t|d|� ��N t|j�rbt|j�t| j� | jkrb|j�� }|r$| �|� q$W d � n1 sv0 Y d S )Nc @ s( e Zd Zdd� Zddddd�dd �Zd
S ) z#Resource._shrink_down.<locals>.Noopc S s d S r r r' r r r � __enter__� s z-Resource._shrink_down.<locals>.Noop.__enter__�typer r �None)�exc_type�exc_val�exc_tb�returnc S s d S r r )r rP rQ rR r r r �__exit__� s z,Resource._shrink_down.<locals>.Noop.__exit__N)r r r rM rT r r r r �Noop� s rU �mutex)r �getattrr( r r" r$ �popleftr? )r rF rU r r1 r r r rH � s
�
zResource._shrink_downc C s | j S r )r r' r r r r$ � s zResource.limitc C s | � |� d S r )rL )r r$ r r r r$ � s ZKOMBU_DEBUG_POOLr c O s� dd l }| jd }| _td|� d| jj� �� | j|i |��}||_td|� d| jj� �� t|d�sng |_|j� |�
� � |S )Nr r �+z ACQUIRE �-� acquired_by) � traceback�_next_resource_id�print� __class__r �
_orig_acquire� _resource_id�hasattrr[ �append� format_stack)r �args�kwargsr\ �id�rr r r r8 � s
c C sR |j }td|� d| jj� �� | �|�}td|� d| jj� �� | jd8 _|S )NrY z RELEASE rZ r )ra r^ r_ r �
_orig_releaser] )r r rg rh r r r r/ � s
)NNN)FN)T)FFF)T) r r r r r r) r r% r# r, r8 r4 r; r< r> r/ r? r
rL rH �propertyr$ �setter�os�environr3 r` ri r] r r r r r ! s4
4
&
r )r �
__future__r rl � collectionsr r r r Z
_LifoQueue�typingr � r Z utils.compatr Zutils.functionalr
�typesr r r r r r r �<module> s