o i.@s4UddlmZddlZddlmZddlZejr ddlZddlZddl m Z ddl m Z ddlmZddlmZddlmZdd lmZd aeed < d d ejd dejdfddZdejd fddZdddefddZdeddfddZd!ddZe ddeddfddZ!e ddeddfddZ!dS)")partialN) ModuleType) _threading)stack)ModuleWatchdog)config)get_argument_value)wrapFASYNCIO_IMPORTEDloopasyncio.AbstractEventLoopreturnzasyncio.Task[typing.Any]cCdSNr rrN/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/profiling/_asyncio.py current_tasksrcCrrrrrrrget_running_loopsrtaskcCs dt|S)NzTask-%d)idrrrr_task_get_name!s rasynciocCsBddlm}tjdkr|jj}|j}n|jj}d}t ||dS)Nr)tasksi ) rrsys hexversion_scheduled_tasksdata _eager_tasks _all_tasksr init_asyncio)r asyncio_tasksscheduled_tasks eager_tasksrrr_call_init_asyncio%s  r%cCsftrdtjvr dSddl}d}z|}Wn tyYdSwtt t t j |t|dS)Nrr)r rmodulesrr RuntimeErrorrtrack_asyncio_looptypingcastintddtrace_threadingcurrent_threadidentr%)r running_looprrr$link_existing_loop_to_current_thread2s   r0c s0datdrjtd<n tjdrjjtd<dtjdffdd }|td<dd td <tjj o9tj t j d }t j d krKt|d d}nt|dd}|dur{tt|jdtjdtjfdttjdfdtttjfdtjffdd }rttt j djjjdtjddttjdfdtttjfddfdd}ttt j djjdtjdttdtdffdttjdfdtttjfdtjfdd}ttt j djjdtjdtjdtjdffdttjdfdtttjfdtjffdd }ttt j djjdtjddttjdfdtttjfdtjffdd }t j dkrlt j d}|durlt|dd}|durlt|d rltt|j dtjd!dttjdfdtttjfdtjfd"d}ttt j djj dtjd!dttjdfdtttjfdd#fd$d}t!dSdS)%NTrr aio.AbstractEventLoopcs zWStyYdSwr)rr'rrrr_get_running_loopTs   z_.._get_running_looprcSs|Sr)get_namerrrr[sz_..rzasyncio.eventsi_BaseDefaultEventLoopPolicyBaseDefaultEventLoopPolicyf.argskwargscs:t||dd}rttttj|||i|S)Nr ) rrr(r)r*r+r,r-r.)r8r9r:r ) init_stackrr_ks _.._r).Nc Ssz.||i|Wt||dd}|dusJt||dd}td|}|D]}t||q%St||dd}|dussz_.._..r;) r)r*OptionalgetrAIterablerrIrrB)r8r9r:rCrFrJrKr2rrr=s).rEcs|ttjd|d}tdt||dd}j||d}td}|dur.t|||f|dd}||i|S) Nr1r rErargrrr;) r)r*rOrPrrLrArrB)r8r9r:r awaitablerKrCr2rrr=s  i zasyncio.taskgroups TaskGroup create_task).rHcSs:||i|}td}|dur|durt|||SNr)rArrB)r8r9r:resultrCrrrr=s   rHcSs2||i|}td}|durt|||SrV)rArweak_link_tasks)r8r9r:rrCrrrr=s   )"r hasattrrrATaskr)rOrrenabled is_availablerr&rgetattrrr set_event_loopCallableAnytupledictstrr_GatheringFuture__init___waitrI as_completed GeneratorshieldrPrUr%)rr3 events_module policy_classr=taskgroups_moduletaskgroup_classrrr<rr=Is         6                  r=uvloopc stjjsdSddltjjotjt|dd}|dur=tt|dt j ddt t j dfdt tt j fd d ffd d }t|d d}|durut|drwtt|jdt j dt j fdt t j dfdt tt j fd t j ffdd }dSdSdS)auHook uvloop to track event loops. uvloop doesn't inherit from BaseDefaultEventLoopPolicy, and on Python 3.11+ uvloop.run() uses asyncio.Runner which bypasses set_event_loop entirely. We hook new_event_loop to catch all uvloop loop creations. We also hook EventLoopPolicy.set_event_loop for the deprecated uvloop.install() + asyncio.run() pattern. Nrnew_event_loopr8).r r9.r:r r csH||i|}r"tttj}t|dt||t |S)NT) r)r*r+r,r-r.rset_uvloop_moder(r%)r8r9r:r thread_idrnrrr=s  r>EventLoopPolicyr^csltttj}rt|dt||dd}r/|dur/t tttj|t ||i|S)NTr;r ) r)r*r+r,r-r.rrqrr(r%)r8r9r:rrr rnrrr=s  )rrrorr[r\r]rr r)r_rar`rbrcrYr^)ronew_event_loop_funcr=rkrrnrr=s:        r)r N)" functoolsrrtypesrr) TYPE_CHECKINGraioddtrace.internal._unpatchedrr,"ddtrace.internal.datadog.profilingrddtrace.internal.moduler#ddtrace.internal.settings.profilingrddtrace.internal.utilsrddtrace.internal.wrappingr r bool__annotations__rOrrrcrr%r0after_module_importedr=rrrrs:          #