File: //home/arjun/projects/env/lib/python3.10/site-packages/kombu/__pycache__/resource.cpython-310.pyc
o
'we7 � @ 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 d S ty Y d S w �N)�force_close_all� Exception)�resource� r �G/home/arjun/projects/env/lib/python3.10/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 � 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� Zdd� Zd(dd�Zd)d d �Zed d � �Zejd!d � �Zej�d"�rfe
ZeZd#Z d$d � Z
d%d� ZdS dS )*�ResourcezPool of resources.FNc C s^ || _ |pd| _d| _|d ur|n| j| _t� | _t� | _| jr)td ur)t| 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��)Nzsubclass responsibility)�NotImplementedError�r r r r r$ 7 s zResource.setupc C s6 | j rt| 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���jrM z
�jj||d�� W n
ty" ��� Y n)w z��� �� W n tyC t � t
�r=�j� � � � �� � � � w �j
�� � nq n����� �� � �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 poolr )�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 r0 h s z!Resource.acquire.<locals>.release)r � RuntimeErrorr% r! �getr r- �prepare�
BaseException�
isinstancer
r+ r0 r# �addr, )r r. r/ r0 r r1 r �acquireB s4 �
�� �
zResource.acquirec C s |S r r �r r r r r r5 v � zResource.preparec C s |� � d S r )�closer: r r r �close_resourcey r zResource.close_resourcec C � d S r r r: r r r �release_resource| r; 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= r: r r r �replace s zResource.replacec C s: | j r| j�|� | j�|� | �|� d S | �|� d S r )r% r# r@ r! r+ r? r= r: r r r r0 � s
zResource.releasec C r> r r r: r r r �collect_resource� r; zResource.collect_resourcec C s� | j rdS d| _ | j}| j} z|�� }W n ty Y nw z| �|� W n ty/ Y nw q z|j�� }W n
tyC Y dS w z| �|� W n tyT Y nw q2)z�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).
NT) r r# r! �pop�KeyErrorrB �AttributeErrorr �
IndexError)r �dirtyr �dres�resr r r r
� s: � �� � ��zResource.force_close_allc C s� | j }| jr"d| k r| j k r"n n|s"|s td�| j |���d}|| _ |r7z| �� W n ty6 Y nw | �� ||k rI| j|dkd� d S d S )Nr z,Can't shrink pool when in use: was={} now={}T)�collect)r r# r3 �formatr
r r$ � _shrink_down)r r% �force�
ignore_errors�reset�
prev_limitr r r �resize� s( $�� ��zResource.resizeTc C s� G dd� d�}| j }t|d|� ��- t|j�| jkr6|j�� }|r&| �|� t|j�| jksW d � d S W d � d S 1 sAw Y d S )Nc @ s e Zd Zdd� Zdd d �Zd
S )z#Resource._shrink_down.<locals>.Noopc S r> r r r( r r r � __enter__� r; z-Resource._shrink_down.<locals>.Noop.__enter__�exc_type�type�exc_valr �exc_tbr �return�Nonec S r> r r )r rS rU rV r r r �__exit__� s z,Resource._shrink_down.<locals>.Noop.__exit__N)rS rT rU r rV r rW rX )r r r rR rY r r r r �Noop� s rZ �mutex)r! �getattrr) r r% �popleftrB )r rJ rZ r r2 r r r rL � s
��"�zResource._shrink_downc C s | j S r )r r( r r r r% � s zResource.limitc C s | � |� d S r )rQ )r r% r r r r% � s �KOMBU_DEBUG_POOLr c O s� dd l }| jd }| _td|� d| jj� �� | j|i |��}||_td|� d| jj� �� t|d�s7g |_|j� |�
� � |S )Nr r �+z ACQUIRE �-� acquired_by) � traceback�_next_resource_id�print� __class__r �
_orig_acquire� _resource_id�hasattrra �append� format_stack)r �args�kwargsrb �id�rr r r r9 � s
c C sR |j }td|� d| jj� �� | �|�}td|� d| jj� �� | jd8 _|S )Nr_ z RELEASE r` r )rg rd re r �
_orig_releaserc )r r rm rn r r r r0 � s
)NNN)FN)FFF)T) r r r r r r* r r& r$ r- r9 r5 r= r? rA r0 rB r
rQ rL �propertyr% �setter�os�environr4 rf ro rc r r r r r ! s8
4
!
�r )r �
__future__r rr � collectionsr r r r �
_LifoQueue�typingr � r � utils.compatr �utils.functionalr
�typesr r r r r r r �<module> s