o i @sddlZddlmZddlmZddlmZddlZddlmZddlm Z ddl m Z ddl m Z dd lmZe eZGd d d ejZGd d d eZdS)N)Any)Callable)Optional)periodic)ddup) get_logger)config)Tracerc speZdZdZdejejfdee gdfdee de ddffdd Z dfd d Z dd d Zdd dZZS) Schedulerz!Schedule export of recorded data.N before_flushtracerintervalreturncs8tt|j|d||_|j|_d|_||_tj |_ dS)N)r r) superr __init__r r _configured_interval _last_export_tracerrcode_provenance_enable_code_provenance)selfr r r  __class__O/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/profiling/scheduler.pyrs  zScheduler.__init__cs0tdtt|t|_tddS)zStart the scheduler.zStarting schedulerzScheduler startedN)LOGdebugrr _start_servicetimetime_nsrrrrrr"s  zScheduler._start_servicecCs`td|jdur!z|Wnty tjdddYnwt|j|jt |_ dS)z(Flush events from recorder to exporters.zFlushing eventsNz"Scheduler before_flush hook failedT)exc_info) rrr Exceptionerrorruploadrrrrrr rrrflush)s    zScheduler.flushc CsNt}z|Wtd|jt||_dStd|jt||_w)Nr)r monotonicr%maxrr )r start_timerrrr6s :zScheduler.periodicrN)__name__ __module__ __qualname____doc__ddtracer rupload_intervalrrr floatrrr%r __classcell__rrrrr s"  r csDeZdZdZdZdZdededdffdd Zd fd d ZZ S) ServerlessScheduleraaServerless scheduler that works on, e.g., AWS Lambda. The idea with this scheduler is to not sleep 60s, but to sleep 1s and flush out profiles after 60 sleeping period. As the service can be frozen a few seconds after flushing out a profile, we want to make sure the next flush is not > 60s later, but after at least 60 periods of 1s. g?gN@argskwargsrNcs.|d|jtt|j|i|d|_dS)Nr r) setdefaultFORCED_INTERVALrr2r_profiled_intervals)rr3r4rrrrKs zServerlessScheduler.__init__csl|j|jkr-t|j|j|jkr-ztt|W|j|_ d|_dS|j|_ d|_w|jd7_dS)Nr) r7FLUSH_AFTER_INTERVALSrrrr6rr2rr r rrrrPs  zServerlessScheduler.periodicr)) r*r+r,r-r6r9rrrr1rrrrr2>s  r2)rtypingrrrr.ddtrace.internalr"ddtrace.internal.datadog.profilingrddtrace.internal.loggerr#ddtrace.internal.settings.profilingr ddtrace.tracer r*rPeriodicServicer r2rrrrs        ,