File: //home/arjun/projects/env/lib/python3.10/site-packages/kombu/__pycache__/pools.cpython-310.pyc
o
'we� � @ s d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl
m Z dd l m
Z
dd
lmZ d Zd gZg Ze� Zej�d
�Zdd� ZG dd� de�ZG dd� de �Zdd� ZG dd� de�Z ee ed��Z G dd� de�Z ee ed��Z dd � Z d d � Z!d$d d!�Z"d"d#� Z#dS )%zPublic resource pools.� )� annotationsN)�chain� )�Resource)�Producer)� EqualityDict)�register_after_fork)�lazy)� ProducerPool� PoolGroup�register_group� connections� producers� get_limit� set_limit�reset�
� KOMBU_DISABLE_LIMIT_PROTECTIONc C s | � � d S �N)�clear��group� r �D/home/arjun/projects/env/lib/python3.10/site-packages/kombu/pools.py�_after_fork_cleanup_group � r c sd e Zd ZdZeZdZ� fdd�Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z � fdd�Z
� ZS )r
z*Pool of :class:`kombu.Producer` instances.Tc s0 || _ |�dd �p | j| _t� j|i |�� d S )Nr )r
�popr �super�__init__)�selfr
�args�kwargs�� __class__r r r s zProducerPool.__init__c C s | j jdd�S )NT)�block)r
�acquire�r r r r �_acquire_connection% s z ProducerPool._acquire_connectionc C s. | � � }z| �|�W S ty |�� � w r )r' r �
BaseException�release)r �connr r r �create_producer( s �z ProducerPool.create_producerc C s
t | j�S r )r r+ r& r r r �new0 s
zProducerPool.newc C s. | j rt| j �D ] }| j�| �� � qd S d S r )�limit�range� _resource�
put_nowaitr, )r �_r r r �setup3 s
�zProducerPool.setupc C s d S r r �r �resourcer r r �close_resource8 s zProducerPool.close_resourcec C sN t |�r|� }|jd u r%| �� }z|�|� W |S ty$ |�� � w |S r )�callable�_channelr' �reviver( r) )r �pr* r r r �prepare; s
��zProducerPool.preparec s& |j r|j �� d |_t� �|� d S r )�__connection__r) �channelr r3 r" r r r) G s
zProducerPool.release)�__name__�
__module__� __qualname__�__doc__r �close_after_forkr r' r+ r, r2 r5 r: r) �
__classcell__r r r" r r
s r
c @ s* e Zd ZdZd
dd�Zdd� Zdd � ZdS ) r z Collection of resource pools.NTc C s0 || _ || _| jrtd urt| t� d S d S d S r )r- rA r r )r r- rA r r r r Q s
�zPoolGroup.__init__c C s t d��)Nz!PoolGroups must define ``create``)�NotImplementedError)r r4 r- r r r �createW s zPoolGroup.createc C s, | j }|tu r
t� }| �||� }| |< |S r )r- �use_global_limitr rD )r r4 r- �kr r r � __missing__Z s
zPoolGroup.__missing__)NT)r= r> r? r@ r rD rG r r r r r N s
r c C s t �| � | S )z*Register group (can be used as decorator).)�_groups�appendr r r r r b s
r c @ � e Zd ZdZdd� ZdS )� Connectionsz Collection of connection pools.c C s |j |d�S �N�r- )�Pool�r �
connectionr- r r r rD k r zConnections.createN�r= r> r? r@ rD r r r r rK h � rK rM c @ rJ )� Producersz Collection of producer pools.c C s t t| |d�S rL )r
r
rO r r r rD u s zProducers.createNrQ r r r r rS r rR rS c C s t dd� tD �� S )Nc s s$ � | ]
}|r
|� � ntg �V qd S r )�values�iter)�.0�gr r r � <genexpr>} s �" z _all_pools.<locals>.<genexpr>)r rH r r r r �
_all_pools| s rY c C s t d S )z"Get current connection pool limit.r )�_limitr r r r r � s r Fc C s> | pd} t d p d}| |kr | t d< t� D ]}|�| � q| S )z Set new connection pool limit.r )rZ rY �resize)r- �force� reset_after�
ignore_errors�glimit�poolr r r r � s
r c O s@ t � D ]}z|�� W q ty Y qw tD ]}|�� qdS )z*Reset all pools by closing open resources.N)rY �force_close_all� ExceptionrH r )r r! r` r r r r r � s
�
�r )FFF)$r@ �
__future__r �os� itertoolsr rP r � messagingr �utils.collectionsr � utils.compatr �utils.functionalr �__all__rZ rH �objectrE �environ�get�disable_limit_protectionr r
r r rK r
rS r rY r r r r r r r �<module> s4 4