o i@spddlZddlmZddlmZddlmZGdddejZGdddeZGd d d eZ Gd d d Z dS) N)forksafe)service)PeriodicThreadc seZdZdZddedededd ffd d Zed d Zej dd ZddZ fddZ dddZ e ddZddZZS)PeriodicServicez`A service that runs periodically. It automatically resumes its execution after a fork. FTintervalno_wait_at_start autorestartreturnNcs&t||_d|_||_||_dSN)super__init__ _interval_worker_no_wait_at_start _autorestart)selfrrr  __class__M/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/periodic.pyr s  zPeriodicService.__init__cCs|jSr )rrrrrrszPeriodicService.intervalcCs||_|jr ||j_dSdSr )rrr)rvaluerrrrs cOsDt|j|jd|jj|jjf|j|jd|_|j |j_ |j dS)zStart the periodic service.%s:%s)targetname on_shutdownrN) rrperiodicr __module____name__rrrr__autorestart__startrargskwargsrrr_start_service&s zPeriodicService._start_servicecs*|jr|jtt|j|i|dS)zStop the periodic collector.N)rstopr r _stop_servicer"rrrr'3s zPeriodicService._stop_servicecCs|jr |j|dSdSr rjoinrtimeoutrrrr):szPeriodicService.joincCdSr rrrrrrBzPeriodicService.on_shutdowncCr,r rrrrrrFr-zPeriodicService.periodic)rFTr )rr __qualname____doc__floatboolr propertyrsetterr%r'r) staticmethodrr __classcell__rrrrr s      rc@seZdZdZddZdS)AwakeablePeriodicServicezIA service that runs periodically but that can also be awakened on demand.cCs|jr |jdSdSr )rawakerrrrr7NszAwakeablePeriodicService.awakeN)rrr.r/r7rrrrr6Ks r6csdeZdZdZdddZdddZdejd ejddffd d Zdejd ejddffd d Z Z S) ForksafeAwakeablePeriodicServicez7An awakeable periodic service that auto-resets on fork.r NcCsdS)zReset the service on fork. Implement this to clear the service state before restarting the thread in the child process. NrrrrrresetWsz&ForksafeAwakeablePeriodicService.resetcCs |dSr )r9rrrr_restart_s z)ForksafeAwakeablePeriodicService._restartr#r$cs"tj|i|t|jdSr )r r%rregisterr:r"rrrr%bsz/ForksafeAwakeablePeriodicService._start_servicecs"t|jtj|i|dSr )r unregisterr:r r'r"rrrr'fs z.ForksafeAwakeablePeriodicService._stop_servicer N) rrr.r/r9r:typingAnyr%r'r5rrrrr8Ts   &r8c@sveZdZdZdeddfddZdddZdd d Zdd d Zdd dZ ddZ dddZ dde j eddfddZdS)TimeraF A simple one-shot timer that calls timeout after the specified interval. :param interval: The interval in seconds after which to call timeout. The timer can be started with start() and stopped with stop(). The timer can be reset with reset(), which stops the timer (if running) and starts it again. rr NcCs||_|dSr )r _set_worker)rrrrrr vs zTimer.__init__cCs*t|j|jd|jj|jjfdd|_dS)NrF)rrr)rr _periodicrrrrrrrrrAzs  zTimer._set_workercC|jdSr )rr!rrrrr!z Timer.startcCrCr )rr&rrrrr&rDz Timer.stopcCstr )NotImplementedErrorrrrrr+sz Timer.timeoutcCs||dSr )r+r&rrrrrBs zTimer._periodiccCs|||dSr )r&rAr!rrrrr9s z Timer.resetr+cCs|j|dSr r(r*rrrr)sz Timer.joinr=r )rrr.r/r0r rAr!r&r+rBr9r>Optionalr)rrrrr@ks     r@) r>ddtrace.internalrrddtrace.internal.threadsrServicerr6r8r@rrrrs   B