tniXUddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl m Z ddlmZddlmZddlmZmZddlmZmZmZmZmZdd lmZerddd 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.e2dZ3 ddl4m5Z5ddl6m7Z8e8Z7de9d<e5ddZ:n#e;$r ej<Z:dZ7YnwxYwda=de9d<d d!d"d#d$d%d&e>fd'Z?d&e>fd(Z@dCd)ZAdDd+ZBdCd,ZCdCd-ZDdCd.ZEdCd/ZFdCd0ZGdEd2ZHd3d4d&e>fd5ZIGd6d7ZJGd8d9ZKGd:d;eKZLGd<d=eKZMd>ZNGd?d@ZOGdAdBZPdS)FN)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) 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,r debugteardown_continuous_profilerr GeventContinuousSchedulerr7ThreadContinuousSchedulergetr ValueErrorformatatexitregister)r-r/r0already_initializeddefault_profiler_moder4 experiments frequencys d/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/sentry_sdk/profiler/continuous_profiler.pysetup_continuous_profilerrFJss%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)rJrr r8rprappendensure_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_stop!s% >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_buffer0s(# L$-)?AR   rGrfc,|jdS|jjSrN)rrhrss rErhzContinuousScheduler.profiler_id5s ; 4{&&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..YD&CS=s#C#CDrGF)rrlenrsys_current_framesitemsAttributeErrorr exc_inforangeaddpopleftrqrremoverwrite) rrrtssample_inactive_profilesrrrrts rE _sample_stackz7ContinuousScheduler.make_sampler.._sample_stackBs( 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_stacks UU !*-*=*?*?*E*E*G*GFF&!!!/s|~~>>> 55 ! ;*K%%b&111us2A*A21A2)rgetcwdrrrz)rtrrrs` @@rErz ContinuousScheduler.make_sampler;sikk#&&& >W $ $= U= e= = = = = = = = = = B U e         0rGctj}|jrz||_tj|z }||jkrt |j|z |jrd|_ntj}|jz|j"|jd|_dSdSrI) r* perf_counterrrrr thread_sleeprflush)rtlastelapseds rErunzContinuousScheduler.runs ""l '!%D  '))D0G&&T]W4555! $ $&&D+l '. ; " K     DKKK # "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)c~|jrd|_|j |jd|_d|_dSrIrrjoinrrss rErz"ThreadContinuousScheduler.teardownA < ! DL ; " K     DK 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 Cc~|jrd|_|j |jd|_d|_dSrIrrss rErz"GeventContinuousScheduler.teardown?rrGrv) 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__Ns   &(:<<+!^^ %(EE! L & & 0 0 2 2T5N N rGmonotonic_timerrc||r:|t|_t |_|j|j|z|dSrN) should_flushrrrrrrr)rtrrs rErzProfileBuffer.writejsb   ^ , , . JJLLL%DJ(+D % ->GGGGGrGc(||jz |jkSrN)rr)rtrs rErzProfileBuffer.should_flushrs 99T=MMMrGc|j|j|j|j}t }||||dSrN)rto_jsonrhr-r/radd_profile_chunkr0)rtrenvelopes rErzProfileBuffer.flushwsY ""4#3T\4=QQ::""5))) (#####rGrv) rwrxryrrrmrrzrrrKrGrErrMs !    3       8HEH3DHHHHHN5NTNNNN $$$$$$rGrcBeZdZddZdeddddfdZded d d d dd fd ZdS)rr2Nctjj|_i|_i|_g|_g|_g|_dSrN) rrrchunk_idindexed_framesindexed_stacksframesstackssamplesrss rErzProfileChunk.__init__s; ( 4646.0 .0 02 rGrrrc"|D] \}\}}} |jvrt|D]J\}}|jvr.s!QQQ8,X6QQQrGr$) r enumeraterrrrrrrr rr) rtrrrr' frame_idsrirs ` rErzProfileChunk.writesW28 ; ; .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 rKrGrErr~s3333;;'8;T;;;;8&&)9&EN& &&&&&&rGrrvrr)Qr?rrlrrr*rr\ collectionsrrrsentry_sdk.constsrsentry_sdk.envelopersentry_sdk._lru_cachersentry_sdk.profiler.utilsr r sentry_sdk.utilsr r r rrtypingrrrrrrrrrrtyping_extensionsrsentry_sdk._typesrrrrrr r!r"rmrr# gevent.monkeyr(gevent.threadpoolr) _ThreadPoolr{r ImportErrorr+r,rzrFrLrQrUrWr_rbrdr9rirnrpr}r;r:rrrrKrGrEr<sR  ''''''''%%%%%%((((((******! ++++++AAAAAAAA i!  O******;;;;;;0;J-;;;<00LL:LJJJ /3 +2226 66/6 6666rD    ####"""" 0%0 0000________DAAAAA 3AAAHGGGGG 3GGGT.$.$.$.$.$.$.$.$bLLLLLLLLLLsC%%C65C6