xj7WUddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZddlmZmZdd lmZmZmZmZmZe rDdd l mZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'dd l(m)Z)m*Z*dd lm+Z+m,Z,m-Z-m.Z.m/Z/m0Z0e'de1e0e2dZ3 ddl4m5Z5ddl6m7Z8e8Z7de9d<e5ddZ:n#e;$r ej<Z:dZ7YnwxYwda=de9d<ddddddde>fdZ?de>fd Z@d;d!ZAdfd-ZIGd.d/ZJGd0d1ZKGd2d3eKZLGd4d5eKZMd6ZNGd7d8ZOGd9d:ZPdS)>N)deque)datetimetimezone) TYPE_CHECKING)LRUCache)VERSION)Envelope)DEFAULT_SAMPLING_FREQUENCY extract_stack)capture_internal_exception is_geventloggernowset_in_app_in_frames) AnyCallableDequeDictListOptionalSetTypeUnion) TypedDict)ContinuousProfilerModeSDKInfo)ExtractedSampleFrameIdProcessedFrameProcessedStackStackIdThreadIdProcessedSample timestamp thread_idstack_id) get_original) ThreadPoolzOptional[Type[_ThreadPool]]r)timesleepzOptional[ContinuousScheduler] _scheduleroptionsDict[str, Any]sdk_infor capture_funcCallable[[Envelope], None]returnctdu}|r"tjdtt r t j}n tj}|d |d}n-|di}|dp|}t}|tjkrt||||anE|t jkrt ||||an"td |tjd tj|stj tdS) Nz0[Profiling] Continuous Profiler is already setup profiler_mode _experimentscontinuous_profiling_modez$Unknown continuous profiler mode: {}z9[Profiling] Setting up continuous profiler in {mode} mode)modeT)r,rdebugteardown_continuous_profilerr GeventContinuousSchedulerr7ThreadContinuousSchedulergetr ValueErrorformatatexitregister)r-r/r0already_initializeddefault_profiler_moder4 experiments frequencys XC:\Projects\Historical\venv\Lib\site-packages\sentry_sdk/profiler/continuous_profiler.pysetup_continuous_profilerrFCss%D0' GHHH$&&&{{? !: > 9 >{{?##/0 kk."55  OO7 8 8 Q>&*l&6&67T&U&U#B '   ((** 15 $( "8=S8I8I8I9<rGc|jtjkrdS|jd}|sdS|dS)NFr5continuous_profiling_auto_start)rosgetpidr-r<)rtrCs rErOz)ContinuousScheduler.is_auto_start_enabledsQ 8ry{{ " "5l&&~66  5@AAArGrRc|jsdS|jdkrdStjdt }|j|||S)Ntracez"[Profiling] Auto starting profiler)rJrrr8rprappendensure_running)rtprofiles rErTzContinuousScheduler.auto_startsm| 4 >W $ $4 9:::#%%   ))) rGcZ|jsdS|jdkrdS|dSNmanual)rJrrrss rErPz ContinuousScheduler.manual_starts;|  F >X % % F rGcH|jdkrdS|dSr)rteardownrss rErazContinuousScheduler.manual_stops% >X % % F rGctrNNotImplementedErrorrss rErz"ContinuousScheduler.ensure_running !!rGctrNrrss rErzContinuousScheduler.teardown#rrGctrNrrss rEpausezContinuousScheduler.pause&rrGc\t|j|jt|j|_dSrN) ProfileBufferr-r/PROFILE_BUFFER_SECONDSr0rrss rE reset_bufferz ContinuousScheduler.reset_buffer)s(# L$-)?AR   rGrfc:|jr|jdS|jjSrN)rrrhrss rErhzContinuousScheduler.profiler_id.s#| t{24{&&rGCallable[..., bool]ctjtdjdkrdddddtffd }ndddddtffd }|S) N)max_sizerargsrkwargsr2c js jsdSt j}t}  fdt jD}n1#t$r$tt j YdSwxYwt|D]3} j j 4g} jD]}|j s|||D]} j| j j||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. TcVg|]%\}}t|t|f&SrKstrr .0tidframecachecwds rE zKContinuousScheduler.make_sampler.._sample_stack..RD&CS=s#C#CDrGF)rrlenrsys_current_framesitemsAttributeErrorr exc_inforangeaddpopleftrqrremoverwrite) rrrtssample_inactive_profilesrrrrts rE _sample_stackz7ContinuousScheduler.make_sampler.._sample_stack;s( 1E 4 #4#455 UU !*-*=*?*?*E*E*G*GFF&!!!/s|~~>>> 55 ! |,,JJA(,,T->-F-F-H-HIIII$&!#3::G">: *00999099G(//8888;*K%%b&111us2A((*BBc2t} fdtjD}n1#t$r$t tjYdSwxYwjj||dS)rcVg|]%\}}t|t|f&SrKrrs rErzKContinuousScheduler.make_sampler.._sample_stack..rrGF) rrrrrr rrr)rrrrrrrts rErz7ContinuousScheduler.make_sampler.._sample_stack|s UU !*-*=*?*?*E*E*G*GFF&!!!/s|~~>>> 55 ! ;*K%%b&111us2A*A21A2)rgetcwdrrrz)rtrrrs` @@rErz ContinuousScheduler.make_sampler4sikk#&&& >W $ $= U= e= = = = = = = = = = B U e         0rGcltj}|jrz||_tj|z }||jkrt |j|z |jrd|_ntj}|jz|j}||dSdSrI) r* perf_counterrrrr thread_sleeprflush)rtlastelapsedrs rErunzContinuousScheduler.runs ""l '!%D  '))D0G&&T]W4555! $ $&&D+l '.   LLNNNNN  rGr2rRrvr2rf)r2r)rwrxryr7r{intrrzrOrTrPrarrrrpropertyrhrrrKrGrEr}r}sd%.D "...??"? ? 3 ?  ????8 Bt B B B B  """"""""""""    '''X' ````DrGr}c \eZdZUdZdZded<dZdeddd d d d d df fd ZddZ ddZ xZ S)r;zr This scheduler is based on running a daemon thread that will call the sampler at a regular interval. threadrr7z)sentry.profiler.ThreadContinuousSchedulerrDr-r.r/rr0r1r2Nct||||d|_tj|_dSrN)superrr threadingLocklockrtrDr-r/r0 __class__s rErz"ThreadContinuousScheduler.__init__s= GX|DDD48 N$$ rGcd|_tj}|jr |j|krdS|j5|jr|j|kr ddddS||_d|_|tj|j |j d|_ |j n#t$rd|_d|_ YnwxYwddddS#1swxYwYdS)NFT)nametargetdaemon)rrrrrrrrThreadrrrstart RuntimeErrorrtrs rErz(ThreadContinuousScheduler.ensure_runningsl"ikk < DHOO F Y # #| C  # # # # # # # #DHDL       $* $(SWXXXDK # !!#### # # # % "  #+ # # # # # # # # # # # # # # # # # #s<C%AC% B:9C%:CC%CC%%C),C)ct|jrd|_|j"|jd|_dSdSrIrrjoinrss rErz"ThreadContinuousScheduler.teardownB < ! DL ; " K     DKKK # "rGrv) rwrxry__doc__r7r{rrrrr __classcell__rs@rEr;r;s &.D "--- 6D % %" % % 3 %  % % % % % %"#"#"#"#HrGr;c XeZdZUdZdZded<dedddd d d d d f fd ZddZddZ xZ S)r:as 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. geventrr7rDr-r.r/rr0r1r2Nct'td|jt ||||d|_tj|_ dS)Nz"Profiler mode: {} is not available) r)r=r>r7rrrrrrrs rErz"GeventContinuousScheduler.__init__sb  AHHSSTT T GX|DDD/3 N$$ rGcd|_tj}|jr |j|krdS|j5|jr|j|kr ddddS||_d|_|td|_ |j |j n#t$rd|_d|_YnwxYwddddS#1swxYwYdS)NFT) rrrrrrrr)rspawnrrrs rErz(GeventContinuousScheduler.ensure_runnings]"ikk < DHOO F Y # #| C  # # # # # # # #DHDL      $Q--DK # !!$(++++ # # # % "  ## # # # # # # # # # # # # # # # # # #s;C6CB.-C.C CC  CC Cct|jrd|_|j"|jd|_dSdSrIrrss rErz"GeventContinuousScheduler.teardown6rrGrv) rwrxryrr7r{rrrrrrs@rEr:r:s  &.D "--- % %" % % 3 %  % % % % % %####@rGr:<c VeZdZdddddedddd f d Zd ed d dd fdZd edefdZddZ d S)rr-r.r/r buffer_sizer0r1r2Nc>||_||_||_||_t jj|_t|_ t|_ tj tj|j z |_dSrN)r-r/rr0uuiduuid4hexrh ProfileChunkchunkrstart_monotonic_timerrutcr%start_timestamp)rtr-r/rr0s rErzProfileBuffer.__init__Cs   &(:<<+!^^ %(EE! L & & 0 0 2 2T5N N rGmonotonic_timerrc||r:|t|_t |_|j|j|z|dSrN) should_flushrrrrrrr)rtrrs rErzProfileBuffer.write_sb   ^ , , . JJLLL%DJ(+D % ->GGGGGrGc(||jz |jkSrN)rr)rtrs rErzProfileBuffer.should_flushgs 99T=MMMrGc|j|j|j|j}t }||||dSrN)rto_jsonrhr-r/r add_profile_chunkr0)rtrenvelopes rErzProfileBuffer.flushlsY ""4#3T\4=QQ::""5))) (#####rGrv) rwrxryrrrmrrzrrrKrGrErrBs !    3       8HEH3DHHHHHN5NTNNNN $$$$$$rGrcBeZdZddZdeddddfdZded d d d dd fd ZdS)rr2Nctjj|_i|_i|_g|_g|_g|_dSrN) rrrchunk_idindexed_framesindexed_stacksframesstackssamplesrss rErzProfileChunk.__init__ts; ( 4646.0 .0 02 rGrrrc"|D] \}\}}} |jvrt|D]J\}}|jvr.s!QQQ8,X6QQQrGr$) r enumeraterrrrrrrr rr) rtrrrr' frame_idsrirs ` rErzProfileChunk.write}sW28 ; ; .C.(Iv ;4#666'0';';:: 8#4+>>>>::::: ;- ; ;sCC*D  D rhr-r.r/rcb|j|j|jdtjDd}t |d|d|d|d|j|dtdd ||d d }d D]4}||*t|| ||<5|S) Nc`i|]+}t|jdt|ji,S)r)ridentr)rrs rE z(ProfileChunk.to_json..sH   FL!!C ,,$   rG)rrrthread_metadatarin_app_excludein_app_include project_rootr)rversionpython2)r client_sdkplatformrrhr$)release environmentdist) rrrrrrrrrstrip)rtrhr-r/rpayloadkeys rEr zProfileChunk.to_jsonskk|  (133         H  $ % $ % N #      ("!&   4 9 9Cs|'"73<006688 rGrv)rwrxryrrmrrr rKrGrErrss3333;;'8;T;;;;8&&)9&EN& &&&&&&rGrrvrr)Qr?rrlrrr*rr\ collectionsrrrtypingrsentry_sdk._lru_cachersentry_sdk.constsrsentry_sdk.enveloper sentry_sdk.profiler.utilsr r sentry_sdk.utilsr r rrrrrrrrrrrrtyping_extensionsrsentry_sdk._typesrrrrrr r!r"rmrr# gevent.monkeyr(gevent.threadpoolr) _ThreadPoolr{r ImportErrorr+r,rzrFrLrQrUrWr_rbrdr9rirnrpr}r;r:rrrrKrGrEr<s  '''''''' ******%%%%%%((((((SSSSSSSSSSSSSSSSSSSSSS++++++AAAAAAAA i!  O******;;;;;;0;J-;;;<00LL:LJJJ /3 +2226 66/6 6666rD    ####"""" 0%0 0000________D????? 3???DEEEEE 3EEEP.$.$.$.$.$.$.$.$bLLLLLLLLLLs%CCC