File: //opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/__pycache__/worker.cpython-37.pyc
B
T a+ � @ s� d dl Z d dlZd dlmZmZ d dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZ er�d dlmZ d d lmZ d d
lmZ e� ZG d d � d e�ZdS )
� N)�sleep�time)�check_thread_support)�Queue�Full)�logger)�DEFAULT_QUEUE_SIZE)�MYPY)�Any)�Optional)�Callablec @ sf e Zd Zefdd�Zedd� �Zdd� Zdd� Zd d
� Z d d � Z
ddd�Z dd� Z dd� Z
dd� Zd
S )�BackgroundWorkerc C s* t � t|�| _t�� | _d | _d | _d S )N)r r �_queue� threading�Lock�_lock�_thread�_thread_for_pid)�selfZ
queue_size� r �B/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/worker.py�__init__ s
zBackgroundWorker.__init__c C s& | j t�� krdS | js dS | j�� S )NF)r �os�getpidr �is_alive)r r r r r s
zBackgroundWorker.is_alivec C s | j s| �� d S )N)r �start)r r r r �_ensure_thread' s z BackgroundWorker._ensure_threadc C s` t � | }| j}|j�� z4x.|jrJ|t � }|dkr:dS |jj|d� q W dS |j�� X d S )Nr F)�timeoutT)r r �all_tasks_done�acquire�unfinished_tasks�wait�release)r r Zdeadline�queue�delayr r r �_timed_queue_join, s
z"BackgroundWorker._timed_queue_joinc C sJ | j �: | js<tj| jdd�| _d| j_| j�� t� � | _
W d Q R X d S )Nz raven-sentry.BackgroundWorker)�target�nameT) r r r �Thread�_targetr �daemonr r r r )r r r r r > s
zBackgroundWorker.startc C sb t �d� | j�H | jrTy| j�t� W n tk
rF t �d� Y nX d| _d| _W dQ R X dS )z�
Kill worker thread. Returns immediately. Not useful for
waiting on shutdown for events, use `flush` for that.
z"background worker got kill requestz)background worker queue full, kill failedN) r �debugr r r �
put_nowait� _TERMINATORr r )r r r r �killI s
zBackgroundWorker.killNc C sD t �d� | j� | jr,|dkr,| �||� W d Q R X t �d� d S )Nz#background worker got flush requestg zbackground worker flushed)r r+ r r � _wait_flush)r r �callbackr r r �flushZ s
zBackgroundWorker.flushc C sl t d|�}| �|�sh| j�� d }t�d|� |d k r@|||� | �|| �sh| j�� d }t�d|� d S )Ng�������?� z %d event(s) pending on flushz"flush timed out, dropped %s events)�minr% r �qsizer r+ �error)r r r0 Zinitial_timeout�pendingr r r r/ b s
z BackgroundWorker._wait_flushc C s2 | � � y| j�|� dS tk
r, dS X d S )NTF)r r r, r )r r0 r r r �submito s zBackgroundWorker.submitc C sf x`| j �� }z<|tkrP y
|� W n" tk
rD tjddd� Y nX W d | j �� X td� qW d S )NzFailed processing jobT)�exc_infor )r �getr- � Exceptionr r5 � task_doner )r r0 r r r r) x s
zBackgroundWorker._target)N)�__name__�
__module__� __qualname__r r �propertyr r r% r r. r1 r/ r7 r) r r r r r
s
r
)r r r r Zsentry_sdk._compatr Zsentry_sdk._queuer r Zsentry_sdk.utilsr Zsentry_sdk.constsr Zsentry_sdk._typesr �typingr
r r �objectr- r
r r r r �<module> s