File: //usr/local/lib/python3.9/site-packages/kombu/__pycache__/pools.cpython-39.pyc
a
X>h� � @ 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�
Z KOMBU_DISABLE_LIMIT_PROTECTIONc C s | � � d S �N)�clear��group� r �5/usr/local/lib/python3.9/site-packages/kombu/pools.py�_after_fork_cleanup_group s 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 s6 | � � }z | �|�W S ty0 |�� � Y n0 d S 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 r&t| j �D ]}| j�| �� � qd S r )�limit�rangeZ _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 sP t |�r|� }|jd u rL| �� }z|�|� W n tyJ |�� � Y n0 |S r )�callableZ_channelr% Zreviver& r' )r �pr( r r r �prepare; s
zProducerPool.preparec s& |j r|j �� d |_t� �|� d S r )Z__connection__r' Zchannelr r0 r r r r' G s
zProducerPool.release)�__name__�
__module__� __qualname__�__doc__r �close_after_forkr r% r) r* r/ r2 r5 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 s( || _ || _| jr$td ur$t| t� d S r )r+ r: r r )r r+ r: r r r r Q s zPoolGroup.__init__c C s t d��d S )Nz!PoolGroups must define ``create``)�NotImplementedError)r r1 r+ r r r �createW s zPoolGroup.createc C s, | j }|tu rt� }| �||� }| |< |S r )r+ �use_global_limitr r= )r r1 r+ �kr r r � __missing__Z s
zPoolGroup.__missing__)NT)r6 r7 r8 r9 r r= r@ 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 @ s e Zd ZdZdd� ZdS )� Connectionsz Collection of connection pools.c C s |j |d�S �N�r+ )ZPool�r �
connectionr+ r r r r= k s zConnections.createN�r6 r7 r8 r9 r= r r r r rC h s rC rE c @ s e Zd ZdZdd� ZdS )� Producersz Collection of producer pools.c C s t t| |d�S rD )r
r
rF r r r r= u s zProducers.createNrH r r r r rI r s rI c C s t dd� tD �� S )Nc s s" | ]}|r|� � ntg �V qd S r )�values�iter)�.0�gr r r � <genexpr>} � z _all_pools.<locals>.<genexpr>)r rA r r r r �
_all_pools| s rP 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 pd}| |kr:| t d< t� D ]}|�| � q*| S )z Set new connection pool limit.r )rQ rP �resize)r+ �forceZ reset_after�
ignore_errorsZglimit�poolr r r r � s
r c O sB t � D ]$}z |�� W q ty( Y q0 qtD ] }|�� q0dS )z*Reset all pools by closing open resources.N)rP Zforce_close_all� ExceptionrA r )r r rU r r r r r � s
r )FFF)$r9 �
__future__r �os� itertoolsr rG r Z messagingr Zutils.collectionsr Z utils.compatr Zutils.functionalr �__all__rQ rA �objectr> �environ�getZdisable_limit_protectionr r
r r rC r
rI r rP r r r r r r r �<module> s2 4