File: //usr/local/lib/python3.9/site-packages/kombu/__pycache__/pidbox.cpython-39.pyc
a
X>h�9 � @ s" d Z ddlmZ ddlZddlZddlmZmZ ddlm Z ddl
m
Z
ddl m Z ddl
m
Z
d d
lmZmZmZmZ d d lmZ d d lmZmZ d d
lmZ d dlmZ d dl m Z d dl m Z m Z d dl!m"Z" d dl#m$Z$ dZ%dZ&dZ'ee(�Z)e)j*e)j+ Z*Z+G dd� d�Z,G dd� d�Z-dS )zGeneric process mailbox.� )� annotationsN)� defaultdict�deque)�contextmanager)�copy)�count)�time� )�Consumer�Exchange�Producer�Queue)� LamportClock)�
maybe_declare�oid_from)�InconsistencyError)�
get_logger)�match)�maybe_evaluate�reprcall)�cached_property)�uuid�
z�A node named {node.hostname} is already using this process mailbox!
Maybe you forgot to shutdown the other node or did not do so properly?
Or if you meant to start multiple nodes on the same host please make sure
you give each node a unique node name!
)�Node�Mailboxc @ s� e Zd ZdZdZdZdZdZdZddd�Z ddd�Z
dd � Z d
d � Z d d d
�Z
d dd�Zd dd�Zdd� Zdd� Zd dd�ZeZdd� ZdS ) r z
Mailbox node.Nc C s: || _ || _|| _|| _| jjj| _|d u r0i }|| _d S �N)�channel�mailbox�hostname�state�clock�adjust� adjust_clock�handlers)�selfr r r r# r � r% �6/usr/local/lib/python3.9/site-packages/kombu/pidbox.py�__init__8 s z
Node.__init__Tc sP � j �� j�}� fdd�}||_t|p*� j|gf||d u rB� j jn|d�|��S )Nc s |rt �tj� d�� d S )N)�node)�warnings�warn�W_PIDBOX_IN_USE�format)�name�messagesZ consumers�r$ r% r& �verify_exclusiveF s z'Node.Consumer.<locals>.verify_exclusive)�no_ack�accept)r � get_queuer Z on_declaredr
r r2 )r$ r r1 r2 �options�queuer0 r% r/ r& r
C s ���z
Node.Consumerc C s || j |j< |S r )r# �__name__)r$ Zfunr% r% r& �handlerQ s z Node.handlerc C s t d|dd� d S )NzCannot decode message: %rr ��exc_info)�error)r$ �message�excr% r% r& �on_decode_errorU s zNode.on_decode_errorc C s& | j ||p| jg| jd�}|�� |S )N)r � callbacksr= )r
�handle_messager= Zconsume)r$ r �callback�consumerr% r% r& �listenX s
�z Node.listenc
K s� |pi }t dt|d|d�||� |r*| jp.| j}z|||�}W nP tyR � Y n> ty� } z&td|dd� dt|�i}W Y d }~n
d }~0 0 |r�| j| j |i|d |d |d
� |S ) Nz1pidbox received method %s [reply_to:%s ticket:%s]r% )�kwargszpidbox command error: %rr r8 r: �exchange� routing_key)rD rE �ticket)
�debugr � handle_call� handle_cast�
SystemExit� Exceptionr: �repr�replyr ) r$ �method� arguments�reply_torF rC �handlerM r< r% r% r& �dispatch_ s& � " �z
Node.dispatchc C s$ |si n|}| j | | jfi |��S r )r# r �r$ rN rO r% r% r& rQ t s z Node.handlec C s | � ||�S r �rQ rS r% r% r& rH x s zNode.handle_callc C s | � ||�S r rT rS r% r% r& rI { s zNode.handle_castc C s� |� d�}|� d�}|� d�}|r8| �|j� d�p4d� | j}d}|rT||v rrd}n |rn|rnt|||�rrd}nd}|r�| jf i |��S d S )N� destination�pattern�matcherr r FT)�getr" �headersr r rR )r$ �bodyr; rU rV rW r Z run_dispatchr% r% r& r? ~ s
zNode.handle_messagec K s" | j j||||| j| j jd� d S )N)r �
serializer)r �_publish_replyr r[ )r$ �datarD rE rF rC r% r% r& rM � s �z
Node.reply)NNNN)NTN)NN)NNN)N)N)r6 �
__module__� __qualname__�__doc__r r r# r r r' r
r7 r= rB rR rQ rH rI r? Zdispatch_from_messagerM r% r% r% r& r & s, �
�
r c
@ s� e Zd ZdZeZdZdZdZdZ dZ
dZ dZ dgZ
dZd0dd �Zd
d � Zd1d d
�Zd2dd�Zd3dd�Zd4dd�Zd5dd�Zdd� Zedd� �Zdd � Zed6d d ��Zd7d d �Zd8d!d"�Z d9d$d%�Z d:d&d'�Z d(d)� Z d*d+� Z e!d,d-� �Z"ed.d/� �Z#dS );r zProcess Mailbox.z %s.pidboxzreply.%s.pidboxN�direct�json� $@c C s� || _ || _|| _|d u r t� n|| _| �| j | j�| _| �| j �| _t t
�| _ |d u r^| j n|| _ |d u rr| j
n|| _
|| _| | _|
| _| | _|| _d S r )� namespace�
connection�typer r �
_get_exchangerD �_get_reply_exchange�reply_exchanger r � unclaimedr2 r[ � queue_ttl�
queue_expires�reply_queue_ttl�reply_queue_expires�_producer_pool) r$ rd rf re r r2 r[ �
producer_poolrk rl rm rn r% r% r& r' � s
zMailbox.__init__c C s t | �}||_|S r )r re )r$ re �boundr% r% r& �__call__� s zMailbox.__call__c C s |p
t �� }| j||||| d�S )N)r )�socket� gethostname�node_cls)r$ r r r r# r% r% r& r � s z Mailbox.Nodec C s$ |si n|}| j |||d|||d�S )NT)rM �timeoutr@ r ��
_broadcast)r$ rU �commandrC rv r@ r r% r% r& �call� s
�z Mailbox.callc C s |si n|}| j |||dd�S �NF)rM rw )r$ rU ry rC r% r% r& �cast� s z Mailbox.castc C s |si n|}| j ||dd�S r{ rw )r$ ry rC r% r% r& �abcast� s zMailbox.abcastr c C s$ |si n|}| j ||d||||d�S )NT)rM rv �limitr@ r rw )r$ ry rC rv r~ r@ r r% r% r& �
multi_call� s
�zMailbox.multi_callc C s0 | j }t|� d| jj� �| j|dd| j| jd�S )N�.FT)rD rE �durable� auto_delete�expires� message_ttl)�oidr
ri r- rn rm )r$ r� r% r% r& �get_reply_queue� s �zMailbox.get_reply_queuec C s | � � S r )r� r/ r% r% r& � reply_queue� s zMailbox.reply_queuec C s( t |� d| j� d�| jdd| j| jd�S )Nr� z.pidboxFT)rD r� r� r� r� )r
rd rD rl rk )r$ r r% r% r& r3 � s �zMailbox.get_queuec c sV |r |V nF| j rD| j �� �}|V W d � qR1 s80 Y nt|dd�V d S )NF)Z auto_declare)rp �acquirer )r$ �producerr r% r% r& �producer_or_acquire s &zMailbox.producer_or_acquirec K s� |p
| j j}t|dddd�}| �||��T}z0|j|f|||g|| j�� d�dd�|�� W n tyl Y n0 W d � n1 s�0 Y d S )Nra � transientF)Z
exchange_type�
delivery_moder� )rF r T)rD rE �declarerY �retry)re �default_channelr r� �publishr �forwardr ) r$ rM rD rE rF r r� �opts�chanr% r% r& r\
s( ��
���
zMailbox._publish_replyc C s� |||| |
d�} |p| j j} | j}
|rNt| �| �� | j|| jj| jd�d� |pV| j }| �
|| ��D}|j | |
j|
g| j �
� |r�t� | ndd�|dd� W d � n1 s�0 Y d S )N)rN rO rU rV rW )rD rE )rF rP r )r r� T)rD r� rY r[ r� )re r� rD r r� �updateri r- r� r[ r� r� r r� r )r$ rf rO rU � reply_ticketr rv r[ r� rV rW r; r� rD r% r% r& �_publish s0 � ��
��zMailbox._publishFc C s� |d ur(t |ttf�s(td�t|����|
d urdt |
t�sd| d urdt | t�sdtd�t|
�t| ����|pji }|rvt� pxd } |p�| jj }
|d u r�|r�|r�t
|�p�d }| p�| j } | j |||| |
|| |
| d� | r�| j
| ||||
d�S d S )Nz'destination must be a list/tuple not {}z.pattern and matcher must be strings not {}, {})rU r� r rv r[ rV rW )r~ rv r@ r )�
isinstance�list�tuple�
ValueErrorr, rf �strr re r� �lenr[ r� �_collect)r$ ry rO rU rM rv r~ r@ r r[ rV rW r� r� r% r% r& rx 8 sL ����� ��
��zMailbox._broadcastc
s$ |d u r| j }|p| jj}| j}t||g|dd�} g �| j�| jj� z ����W S t yb Y n0 � ����fdd�}
| �
|
� z�| �h |r�t |�p�t � D ]2} z| jj
|d� W q� tjy� Y q�Y q�0 q��W d � W |�|j� S 1 � s�0 Y W |�|j� n|�|j� 0 d S )NT)r2 r1 c sn |j j}� |d�pd� |d�}|r2t� |kr2d S |d��}|�kr\�rP�| � ��| � n�| �| � d S )Nr r r� rF )rY rX r �append)rZ r; �headerr� Zthis_id�r" r@ � responsesrF rj r% r& �
on_messagep s
z$Mailbox._collect.<locals>.on_message)rv )r2 re r� r� r
rj r r! �pop�KeyErrorZregister_callback�ranger Z drain_eventsrs rv Z after_reply_message_receivedr- ) r$ rF r~ rv r@ r r2 r� r5 rA r� �ir% r� r& r� _ s2
�zMailbox._collectc C s t | j| |ddd�S )NFr� �rf r� r� )r � exchange_fmt)r$ rd rf r% r% r& rg � s
�zMailbox._get_exchangec C s t | j| dddd�S )Nra Fr� r� )r �reply_exchange_fmt)r$ rd r% r% r& rh � s
�zMailbox._get_reply_exchangec C s t | �S r )r r/ r% r% r& r� � s z Mailbox.oidc C s
t | j�S r )r ro r/ r% r% r& rp � s zMailbox.producer_pool)
ra NNNNNNNNrc )NNNN)NNNN)N)N)Nr NNN)NN)NN)NNNNNNNN)
NNFr NNNNNN)Nr NNN)$r6 r^ r_ r` r ru r� r� rd re rf rD ri r2 r[ r' rr rz r| r} r r� r r� r3 r r� r\ r� rx r� rg rh �propertyr� rp r% r% r% r& r � sh �
�
�
�
�
�
( �
,
r ).r` �
__future__r rs r) � collectionsr r �
contextlibr r � itertoolsr r � r
r r r
Zclocksr �commonr r �
exceptionsr �logr rW r Zutils.functionalr r Z
utils.objectsr Z
utils.uuidr ZREPLY_QUEUE_EXPIRESr+ �__all__r6 �loggerrG r: r r r% r% r% r& �<module> s0 r