o i@sddlmZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZdd lmZd efd d Zd eeeffddZdddZdddZded eegeffddZdS))Any)CallableN)config) SPAN_KIND) trace_utils)SpanKind) SpanTypes)Config)tracerreturncCs ttjS)N)strdramatiq __version__rr[/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/dramatiq/patch.py get_versions rcCsddiS)Nr z>=1.10.0rrrrr_supported_versionssrcCs.ttddrdSdt_tddttjdS)zM Instrument dramatiq so any new Actor is automatically instrumented. __datadog_patchFNTr zActor.send_with_options)getattrr rrwrap"_traced_send_with_options_functionrrrrrpatchs rcCs(ttddsdSdt_ttjddS)zE Disconnect remove tracing capabilities from dramatiq Actors rFNsend_with_options)rr rrunwrapActorrrrrunpatch"s rintegration_configc sHdttgtfdtjdttdtttfdttgtff fdd }|S)a NOTE: This accounts for both the send() and send_with_options() methods, since send() just wraps around send_with_options() with empty options. In terms of expected behavior, this traces the send_with_options() calls, but does not reflect the actual execution time of the background task itself. The duration of this span is the duration of the send_with_options() call itself. funcinstanceargskwargsr c sjtjdtjtjddd}|ttj d|j d|j i||i|WdS1s.wYdS)Nz dramatiq.Actor.send_with_options)pin int_config) span_typeservicez actor.namez actor.options) r tracerWORKERr ext_serviceset_tagsrrPRODUCER actor_nameoptions)rrrr spanrrr_traced_send_with_options8s  $zE_traced_send_with_options_function.._traced_send_with_options)rrr rtupledict)rr.rr-rr-s   r)r N)typingrrr ddtracerddtrace.constantsrddtrace.contribr ddtrace.extrr!ddtrace.internal.settings._configr ddtrace.tracer r rr0rrrrrrrrs