File: //usr/local/lib/python3.9/site-packages/celery/events/__pycache__/snapshot.cpython-39.pyc
a
X>h� � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ dd lmZ d
Ze d �ZG d d
� d
�Zddd�ZdS )aS Periodically store events in a database.
Consuming the events as a stream isn't always suitable
so this module implements a system to take snapshots of the
state of a cluster at regular intervals. There's a full
implementation of this writing the snapshots to a database
in :mod:`djcelery.snapshots` in the `django-celery` distribution.
� )� TokenBucket)� platforms)�app_or_default)�Signal)� instantiate)�
get_logger)�rate)�Timer)�Polaroid�evcamz celery.evcamc @ s� e Zd ZdZdZeddhd�Zedd�ZdZdZ dZ
d d d �Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � ZdS ) r
zRecord event snapshots.N�shutter_signal�state)�nameZproviding_args�cleanup_signal)r F� �?� �@c C sJ t |�| _|| _|| _|| _|p*| jp*t� | _t| _|oBtt |��| _
d S �N) r �appr
�freq� cleanup_freq�timerr �loggerr r �maxrate)�selfr
r r r r r � r �@/usr/local/lib/python3.9/site-packages/celery/events/snapshot.py�__init__# s
zPolaroid.__init__c C s, | j �| j| j�| _| j �| j| j�| _d S r )r Zcall_repeatedlyr �capture�_trefr �cleanup�_ctref�r r r r �install- s �zPolaroid.installc C s d S r r )r r
r r r �
on_shutter3 s zPolaroid.on_shutterc C s d S r r r! r r r �
on_cleanup6 s zPolaroid.on_cleanupc C s& t �d� | jj| jd� | �� d S )NzCleanup: Running...�Zsender)r �debugr �sendr
r$ r! r r r r 9 s
zPolaroid.cleanupc C sB | j d u s| j �� r>t�d| j� | jj| jd� | �| j� d S )Nz Shutter: %sr% )r Z can_consumer r& r
r r' r# r! r r r �shutter> s zPolaroid.shutterc C s | j j| j| jd� d S )N)� clear_after)r
Z freeze_whiler( r) r! r r r r D s zPolaroid.capturec C s, | j r| � � | j �� | jr(| j�� d S r )r �cancelr r! r r r r* G s
zPolaroid.cancelc C s | � � | S r )r" r! r r r � __enter__N s zPolaroid.__enter__c G s | � � d S r )r* )r �exc_infor r r �__exit__R s zPolaroid.__exit__)r Nr NN)�__name__�
__module__� __qualname__�__doc__r r r r r) r r r r" r# r$ r r( r r* r+ r- r r r r r
s&
�
r
r Nc
K s� t |�}|rt�|� |j�||� td| � d|� d�� |j�� } t| | ||||d�}
|
� � |�
� } |jj | d| j id�} z<z| j
dd� W n ty� t�Y n0 W |
�� | �� n|
�� | �� 0 dS ) zStart snapshot recorder.z -> evcam: Taking snapshots with z (every z secs.))r r r r �*)�handlersN)�limit)r r Zcreate_pidlock�logZsetup_logging_subsystem�print�events�Stater r" Zconnection_for_readZReceiver�eventr �KeyboardInterrupt�
SystemExitr* �close)
Zcamerar r Zloglevel�logfileZpidfiler r �kwargsr
Zcam�conn�recvr r r r V s*
�
�r )r Nr NNNN)r1 Zkombu.utils.limitsr Zceleryr Z
celery.appr Zcelery.utils.dispatchr Zcelery.utils.importsr Zcelery.utils.logr Zcelery.utils.timer Zcelery.utils.timer2r �__all__r r
r r r r r �<module> s > �