\
@ s d d l Z d d l m Z m Z d d l m Z m Z d d l m Z m Z d d l m
Z
e Z Gd d d e Z
d S) N)ThreadLock)sleeptime)queuecheck_thread_support)loggerc @ s e Z d Z d d Z e d d Z d d Z d d Z d d
Z d d Z d
d d Z
d d Z d d Z d d Z
d
S)BackgroundWorkerc C s; t t j d | _ t | _ d | _ d | _ d S)N )r r Queue_queuer _lock_thread_thread_for_pid)self r
/worker.py__init__ s
zBackgroundWorker.__init__c C s3 | j t j k r d S| j s& d S| j j S)NF)r osgetpidr is_alive)r r r r r s
zBackgroundWorker.is_alivec C s | j s | j d S)N)r start)r r r r _ensure_thread$ s z BackgroundWorker._ensure_threadc
C s t | } | j } | j j zH x= | j re | t } | d k rO d S| j j d | q) Wd SWd | j j Xd S)Nr FtimeoutT)r r
all_tasks_doneacquireunfinished_taskswaitrelease)r r deadliner delayr r r _timed_queue_join) s
z"BackgroundWorker._timed_queue_joinc C se | j U | j sZ t d | j d d | _ | j j d | j j t j | _ Wd QRXd S)Ntargetnamez raven-sentry.BackgroundWorkerT)
r r r _targetr setDaemonr r r r )r r r r r 8 s
zBackgroundWorker.startc C sM t j d | j 0 | j rB | j j t d | _ d | _ Wd QRXd S)Nz"background worker got kill request)r debugr r r
put_nowait _TERMINATORr )r r r r killC s
zBackgroundWorker.killNc
C sT t j d | j * | j r< | d k r< | j | | Wd QRXt j 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 sk t d | } | j | sg | j j } t j d | | d k rV | | | | j | | d S)Ng?z %d event(s) pending on flush)minr" r
qsizer r' )r r r, Zinitial_timeoutpendingr r r r+ T s
z BackgroundWorker._wait_flushc C s | j | j j | d S)N)r r
r( )r r, r r r submit^ s
zBackgroundWorker.submitc C sy xr | j j } zD | t k r" Py | Wn% t k
rT t j d d d Yn XWd | j j Xt d q Wd S)NzFailed processing jobexc_infoTr )r
getr) Exceptionr error task_doner )r r, r r r r% c s
zBackgroundWorker._target)__name__
__module__ __qualname__r propertyr r r" r r* r- r+ r1 r% r r r r r s
r )r threadingr r r r Zsentry_sdk._compatr r Zsentry_sdk.utilsr objectr) r r r r r <module> s