U
\
@ s^ d dl Z d dlmZmZ d dlmZmZ d dlmZmZ d dl m
Z
e ZG dd deZdS ) N)ThreadLock)sleeptime)queuecheck_thread_support)loggerc @ sb e Zd Zdd Zedd Zdd Zdd Zd d
Zd d Z dddZ
dd Z dd Z dd Z
d
S )BackgroundWorkerc C s* t td| _t | _d | _d | _d S )N)r r Queue_queuer _lock_thread_thread_for_pidself r
/worker.py__init__ s
zBackgroundWorker.__init__c C s& | j t krdS | js dS | j S )NF)r osgetpidr is_aliver r r r r s
zBackgroundWorker.is_alivec C s | j s| d S N)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 Zall_tasks_doneacquirereleaseZunfinished_taskswait)r r Zdeadliner Zdelayr r r _timed_queue_join) s
z"BackgroundWorker._timed_queue_joinc C sL | j < | js>t| jdd| _| jd | j t | _ W 5 Q R X d S )Nz raven-sentry.BackgroundWorker)targetnameT)
r
r r _targetr Z setDaemonr r r r r r r r r 8 s
zBackgroundWorker.startc C s> t d | j$ | jr0| jt d | _d | _W 5 Q R X d S )Nz"background worker got kill request)r debugr
r r
put_nowait _TERMINATORr r r r r killC 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 flushL s
zBackgroundWorker.flushc C sN t d|}| |sJ| j }td| |d k r<||| | || d S )Ng?z %d event(s) pending on flush)minr r Zqsizer r# )r r r( Zinitial_timeoutZpendingr r r r' T s
z BackgroundWorker._wait_flushc C s | | j| d S r )r r r$ r r( r r r submit^ 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 getZ task_doner% Exceptionr errorr r+ r r r r" c s
zBackgroundWorker._target)N)__name__
__module__ __qualname__r propertyr r r r r& r) r' r, r" r r r r r s
r )r Z threadingr r r r Zsentry_sdk._compatr r Zsentry_sdk.utilsr r ZAnyZOptionalCallableobjectr% r r r r r <module> s