관리-도구
편집 파일: worker.cpython-38.pyc
U �Mgp � @ s� d dl Z d dlZd dlmZmZ d dlmZmZ d dlmZ d dl m Z d dl m Z e r|d dl m Z d dl mZ d d l mZ e� ZG d d � d �ZdS ) � N)�sleep�time)�Queue� FullError)�logger)�DEFAULT_QUEUE_SIZE)� TYPE_CHECKING)�Any)�Optional)�Callablec @ sn e Zd Zefdd�Zedd� �Zdd� Zdd� Zd d � Z d d � Z ddd�Z dd� Z dd� Z dd� Zdd� Zd S )�BackgroundWorkerc C s$ t |�| _t�� | _d | _d | _d S �N)r �_queue� threading�Lock�_lock�_thread�_thread_for_pid)�selfZ queue_size� r �?/opt/hc_python/lib/python3.8/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 r )r �startr r r r �_ensure_thread% s z BackgroundWorker._ensure_threadc C sd t � | }| j}|j�� z8|jrL|t � }|dkr<W � dS |jj|d� q W �dS |j�� X d S )Nr F)�timeoutT)r r �all_tasks_done�acquire�release�unfinished_tasks�wait)r r �deadline�queue�delayr r r �_timed_queue_join* s z"BackgroundWorker._timed_queue_joinc C sj | j �Z | js\tj| jdd�| _d| j_z| j�� t� � | _ W n t k rZ d | _Y nX W 5 Q R X d S )Nzsentry-sdk.BackgroundWorker)�target�nameT) r r r �Thread�_targetr �daemonr r r r � RuntimeErrorr r r r r < s � zBackgroundWorker.startc C sb t �d� | j�H | jrTz| j�t� W n tk rF t �d� Y nX d| _d| _W 5 Q 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 �killM s zBackgroundWorker.killNc C sD t �d� | j� | jr,|dkr,| �||� W 5 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 �flush^ s zBackgroundWorker.flushc C s | j �� S r )r �fullr r r r r5 f s zBackgroundWorker.fullc 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 r3 Zinitial_timeout�pendingr r r r2 j s z BackgroundWorker._wait_flushc C s6 | � � z| j�|� W dS tk r0 Y dS X d S )NTF)r r r/ r �r r3 r r r �submitw s zBackgroundWorker.submitc C sf | j �� }z@|tkrW �4qbz |� W n" tk rF tjddd� Y nX W 5 | j �� X td� q d S )NzFailed processing jobT)�exc_infor )r �get� task_doner0 � Exceptionr r9 r r; r r r r+ � s zBackgroundWorker._target)N)�__name__� __module__� __qualname__r r �propertyr r r' r r1 r4 r5 r2 r< r+ r r r r r s r )r r r r Zsentry_sdk._queuer r Zsentry_sdk.utilsr Zsentry_sdk.constsr �typingr r r r �objectr0 r r r r r �<module> s