o BDi,Y@srddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZmZddlmZmZmZmZmZdd lmZerdd lmZdd lmZdd lmZdd lm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddl&m'Z'ddl(m)Z)m*Z*ddlm+Z+m,Z,m-Z-m.Z.m/Z/m0Z0e'de1e.e2dZ3zddl4m5Z5ddl6m7Z8e8Z7e5ddZ9Wne:yej;Z9dZ7Ynwdad d!Z?d"d#Z@d$d%ZAd&d'ZBd(d)ZCd*d+ZDd,d-ZEd.d/ZFGd0d1d1ZGGd2d3d3ZHGd4d5d5eHZIGd6d7d7eHZJd8ZKGd9d:d:ZLGd;d<d<ZMdS)=N)deque)datetimetimezone)VERSION)Envelope)LRUCache)DEFAULT_SAMPLING_FREQUENCY extract_stack)capture_internal_exception is_geventloggernowset_in_app_in_frames) TYPE_CHECKING)Any)Callable)Deque)Dict)List)Optional)Set)Type)Union) TypedDict)ContinuousProfilerModeSDKInfo)ExtractedSampleFrameIdStackIdThreadIdProcessedFrameProcessedStackProcessedSample timestamp thread_idstack_id) get_original) ThreadPooltimesleepcCstdur tddStrtj}ntj}|ddur!|d}n |di}|dp-|}t}|tjkr=t||||an|tjkrJt||||ant d |tdj tjdt t d S) Nz0[Profiling] Continuous Profiler is already setupF profiler_mode _experimentscontinuous_profiling_modez$Unknown continuous profiler mode: {}z9[Profiling] Setting up continuous profiler in {mode} mode)modeT) _schedulerr debugr GeventContinuousSchedulerr.ThreadContinuousSchedulergetr ValueErrorformatatexitregisterteardown_continuous_profiler)optionssdk_info capture_funcdefault_profiler_moder+ experiments frequencyr?`/var/www/Datamplify/venv/lib/python3.10/site-packages/sentry_sdk/profiler/continuous_profiler.pysetup_continuous_profilerJs8       rAcCs$tdurdSts dStdSN)r/is_auto_start_enabled manual_startr?r?r?r@!try_autostart_continuous_profiler}s  rEcCstdurdStSrB)r/ auto_startr?r?r?r@!try_profile_lifecycle_trace_startsrGcCtdurdStdSrB)r/rDr?r?r?r@start_profiler rIcCtjdtddtdS)NzWThe `start_profile_session` function is deprecated. Please use `start_profile` instead. stacklevel)warningswarnDeprecationWarningrIr?r?r?r@start_profile_session  rRcCrHrB)r/ manual_stopr?r?r?r@ stop_profilerrJrUcCrK)NzUThe `stop_profile_session` function is deprecated. Please use `stop_profile` instead.rLrM)rOrPrQrUr?r?r?r@stop_profile_sessionrSrVcCstdadSrB)rUr/r?r?r?r@r8sr8cCstdurdStjSrB)r/ profiler_idr?r?r?r@get_profiler_idsrXcCs|sdStt|kSNF)randomfloat) sample_rater?r?r@+determine_profile_session_sampling_decisionsr]c@s"eZdZUdZeed<ddZdS)ContinuousProfileTactivecCs d|_dSrY)r_selfr?r?r@stops zContinuousProfile.stopN)__name__ __module__ __qualname__r_bool__annotations__rbr?r?r?r@r^s  r^c@steZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ e ddZddZddZdS)ContinuousSchedulerunknowncCszd||_||_||_||_|jd|_|jd}t||_||_ d|_ d|_ d|_ d|_ tdd|_t|_dS)Ng?profile_lifecycleprofile_session_sample_rateF)maxlen)intervalr9r:r;r3 lifecycler]sampled make_samplersamplerbufferpidrunning soft_shutdownr new_profilessetactive_profiles)rar>r9r:r;rkr?r?r@__init__s     zContinuousScheduler.__init__cCs0|jtkr dS|jd}|sdS|dS)NFr,continuous_profiling_auto_start)rtosgetpidr9r3)rar=r?r?r@rCs   z)ContinuousScheduler.is_auto_start_enabledcCs@|jsdS|jdkr dStdt}|j|||S)Ntracez"[Profiling] Auto starting profiler)rpror r0r^rwappendensure_running)raprofiler?r?r@rFs   zContinuousScheduler.auto_startcCs$|jsdS|jdkr dS|dSNmanual)rprorr`r?r?r@rDs   z ContinuousScheduler.manual_startcCs|jdkrdS|dSr)roteardownr`r?r?r@rT s  zContinuousScheduler.manual_stopcCtrBNotImplementedErrorr`r?r?r@r'z"ContinuousScheduler.ensure_runningcCrrBrr`r?r?r@r+rzContinuousScheduler.teardowncCrrBrr`r?r?r@pause/rzContinuousScheduler.pausecCst|j|jt|j|_dSrB) ProfileBufferr9r:PROFILE_BUFFER_SECONDSr;rsr`r?r?r@ reset_buffer3s z ContinuousScheduler.reset_buffercCs|jdurdS|jjSrB)rsrWr`r?r?r@rW9s zContinuousScheduler.profiler_idcsDttddjdkrfdd}|Sfdd}|S)N)max_sizer~csjsjsdStj}t}zfddtD}Wnty0tt YdSwt |D] }j j q5g}jD] }|j sP||qF|D]}j|qSjdurhj||dS) Take a sample of the stack on all the threads in the process. This should be called at a regular interval to collect samples. Tc$g|]\}}t|t|fqSr?strr .0tidframecachecwdr?r@ `KContinuousScheduler.make_sampler.._sample_stack..FN)rwrylenr sys_current_framesitemsAttributeErrorr exc_inforangeaddpopleftr_rremoverswrite)argskwargsrwtssample_inactive_profilesrrrrar?r@ _sample_stackHs0          z7ContinuousScheduler.make_sampler.._sample_stackcsdt}zfddtD}Wnty#ttYdSwjdur0j||dS)rcrr?rrrr?r@rrrFN) r rrrrr rrsr)rrrrrr?r@rs     )r|getcwdrro)rarr?rr@rq@s  [z ContinuousScheduler.make_samplercCszt}|jr,||_t|}||jkrt|j||jr%d|_nt}|js|jdur;|jd|_dSdSrY) r) perf_counterrurrrvrn thread_sleeprsflush)ralastelapsedr?r?r@runs      zContinuousScheduler.runN)rcrdrer.rzrCrFrDrTrrrrpropertyrWrqrr?r?r?r@rhs   erhcs8eZdZdZdZdZfddZddZdd ZZ S) r2zr This scheduler is based on running a daemon thread that will call the sampler at a regular interval. threadz)sentry.profiler.ThreadContinuousSchedulercs&t||||d|_t|_dSrB)superrzr threadingLocklockrar>r9r:r; __class__r?r@rzsz"ThreadContinuousScheduler.__init__c Csd|_t}|jr|j|krdS|jN|jr&|j|kr& WddS||_d|_|tj|j |j dd|_ z|j Wnt yQd|_d|_ Yn wWddSWddS1sewYdS)NFT)nametargetdaemon)rvr|r}rurtrrrThreadrrrstart RuntimeErrorrartr?r?r@rs,  "z(ThreadContinuousScheduler.ensure_runningcC0|jrd|_|jdur|jd|_d|_dSrYrurjoinrsr`r?r?r@r    z"ThreadContinuousScheduler.teardown) rcrdre__doc__r.rrzrr __classcell__r?r?rr@r2s &r2cs4eZdZdZdZfddZddZddZZS) r1as This scheduler is based on the thread scheduler but adapted to work with gevent. When using gevent, it may monkey patch the threading modules (`threading` and `_thread`). This results in the use of greenlets instead of native threads. This is an issue because the sampler CANNOT run in a greenlet because 1. Other greenlets doing sync work will prevent the sampler from running 2. The greenlet runs in the same thread as other greenlets so when taking a sample, other greenlets will have been evicted from the thread. This results in a sample containing only the sampler's code. geventcs>tdur td|jt||||d|_t|_ dS)Nz"Profiler mode: {} is not available) r(r4r5r.rrzrrrrrrr?r@rzs z"GeventContinuousScheduler.__init__c Csd|_t}|jr|j|krdS|jJ|jr&|j|kr& WddS||_d|_|td|_z |j |j Wnt yMd|_d|_Yn wWddSWddS1sawYdS)NFT) rvr|r}rurtrrr(rspawnrrrr?r?r@r"s,   "z(GeventContinuousScheduler.ensure_runningcCrrYrr`r?r?r@rDrz"GeventContinuousScheduler.teardown) rcrdrerr.rzrrrr?r?rr@r1s   "r1<c@s,eZdZddZddZddZddZd S) rcCsP||_||_||_||_tj|_t|_ t |_ t t j|j |_dSrB)r9r: buffer_sizer;uuiduuid4hexrW ProfileChunkchunkr start_monotonic_timerrutcr$start_timestamp)rar9r:rr;r?r?r@rzTs zProfileBuffer.__init__cCs:||r|t|_t|_|j|j||dSrB) should_flushrrrr rrr)ramonotonic_timerr?r?r@rks zProfileBuffer.writecCs||j|jkSrB)rr)rarr?r?r@rtszProfileBuffer.should_flushcCs4|j|j|j|j}t}||||dSrB)rto_jsonrWr9r:radd_profile_chunkr;)rarenveloper?r?r@r{s zProfileBuffer.flushN)rcrdrerzrrrr?r?r?r@rSs  rc@s$eZdZddZddZddZdS)rcCs.tj|_i|_i|_g|_g|_g|_dSrB) rrrchunk_idindexed_framesindexed_stacksframesstackssamplesr`r?r?r@rzs  zProfileChunk.__init__c s|D]]\}\}}}zG|jvrBt|D]\}}|jvr,tjj|<j||qtjj|<jfdd|Dj||j|dWqty_t t YqwdS)Ncsg|]}j|qSr?)r)rframe_idr`r?r@rsz&ProfileChunk.write..r#) r enumeraterrrrrrrr rr) rarrrr& frame_idsrirr?r`r@rs,   zProfileChunk.writecCs|j|j|jddtDd}t|d|d|d|d|j|dtd d ||d d }d D]}||durCt|| ||<q1|S)NcSs"i|] }t|jdt|jiqS)r)ridentr)rrr?r?r@ s  z(ProfileChunk.to_json..)rrrthread_metadatarin_app_excludein_app_include project_rootr)rversionpython2)r client_sdkplatformrrWr)release environmentdist) rrrrrrrrrstrip)rarWr9r:rpayloadkeyr?r?r@rs6  zProfileChunk.to_jsonN)rcrdrerzrrr?r?r?r@rs r)Nr6r|rZrrr)rrO collectionsrrrsentry_sdk.constsrsentry_sdk.envelopersentry_sdk._lru_cachersentry_sdk.profiler.utilsrr sentry_sdk.utilsr r r r rtypingrrrrrrrrrrtyping_extensionsrsentry_sdk._typesrrrrrrr r!r[intr" gevent.monkeyr'gevent.threadpoolr( _ThreadPoolr ImportErrorr*r/rArErGrIrRrUrVr8rXr]r^rhr2r1rrrr?r?r?r@s                    3   lBI0