File: //usr/local/lib/python3.9/site-packages/celery/worker/__pycache__/pidbox.cpython-39.pyc
a
X>h. � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ d
Ze e�Zejejej ZZZG d d � d �ZG d
d� de�ZdS )z Worker Pidbox (remote control).� N)�
ignore_errors)�safe_str)�
AttributeDict)�pass1)�
get_logger� )�control)�Pidbox�gPidboxc @ sT e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )r zWorker mailbox.Nc C s^ || _ |j| _|jjjjt|j�tjjt |j|j||j
j r>t nt
d�d�| _| j jjj| _d S )N)�app�hostname�consumerZtset)�handlers�state)�cr r r Zmailbox�Noder ZPanel�datar �
controllerZ
use_eventloopr �set�nodeZclockZforward�_forward_clock��selfr � r �>/usr/local/lib/python3.9/site-packages/celery/worker/pidbox.py�__init__ s
�� zPidbox.__init__c
C s� | � � z| j�||� W nf tyH } ztd|� W Y d }~nBd }~0 ty� } z"td|dd� | �� W Y d }~n
d }~0 0 d S )NzNo such control command: %szControl command error: %rT)�exc_info)r r Zhandle_message�KeyError�error� Exception�reset)r �body�message�excr r r �
on_message' s zPidbox.on_messagec C s. |j �� | j_| jj| jd�| _|j| j_d S �N)�callback)�
connection�channelr �listenr$ r
Zon_decode_errorr r r r �start3 s z Pidbox.startc C s d S �Nr �r r r r �on_stop8 s zPidbox.on_stopc C s | � � | �|�| _d S r+ )r- �_close_channelr
r r r r �stop; s z Pidbox.stopc C s | � | j� | �| j� d S r+ )r/ r r* r, r r r r ? s z Pidbox.resetc C s" | j r | j jr t|| j jj� d S r+ )r r( r �closer r r r r. C s zPidbox._close_channelc C s4 | � � | jr$td� t|| jj� | �| j� d S )Nz Canceling broadcast consumer...)r- r
�debugr �cancelr/ r r r r r �shutdownG s
zPidbox.shutdown)
�__name__�
__module__� __qualname__�__doc__r
r r$ r* r- r/ r r. r3 r r r r r s r c @ sD e Zd ZdZdZdZdZdd� Zdd� Zdd � Z d
d � Z
d d
� Z dS )r
zWorker pidbox (greenlet).Nr c C s |j �| j|� d S r+ )�poolZspawn_n�loopr r r r r* V s z
gPidbox.startc C s2 | j r.| j�� td� | j �� d | _ | _d S )Nz+Waiting for broadcast thread to shutdown...)�
_node_stopped�_node_shutdownr r1 �waitr, r r r r- Y s
zgPidbox.on_stopc C s | j d7 _ d S )Nr )�_resetsr, r r r r ` s z
gPidbox.resetc C s6 | � |� |�� | j_| jj| jd�| _| j�� d S r% )r. r( r r) r$ r
Zconsume)r r r' r r r � _do_resetc s
zgPidbox._do_resetc C s� | j g}t�� }| _t�� }| _z�|�� ��}td|�� � | �||� |� � s�|j
r�|d | j k r�|d d7 < | �||� z|j dd� W qJ t j
y� Y qJ0 qJW d � n1 s�0 Y W |�� n
|�� 0 d S )Nzpidbox: Connected to %s.r r g �?)�timeout)r= � threading�Eventr; r: Zconnection_for_read�info�as_urir> �is_setr' Z drain_events�socketr? r )r r Zresetsr3 �stoppedr' r r r r9 i s
(z gPidbox.loop) r4 r5 r6 r7 r; r: r= r* r- r r> r9 r r r r r
O s r
)r7 rE r@ Z kombu.commonr Zkombu.utils.encodingr Zcelery.utils.collectionsr Zcelery.utils.functionalr Zcelery.utils.logr � r �__all__r4 �loggerr1 r rB r r
r r r r �<module> s ;