o io@sddlmZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddlm&Z&ddlm'Z'ddlm(Z(ddl)m*Z*dd l)m+Z+dd!l,m-Z-dd"l.m/Z/dd#l.m0Z0dd$l.m1Z1dd%l2m3Z4dd&l2m5Z6dd'l2m7Z8dd(l2m9Z9dd)l2m:Z:dd*l2m;Z;dd+lm?Z?dd-l>m@Z@dd.lAmBZBdd/lCmDZDGd0d1d1e@ZEe=eFZGd2eHd3eHfd4d5ZId2eHd3eJfd6d7ZKGd8d9d9e?ZLdS):)StringION) TracebackType)Any)Callable)Optional)Text)Union)cast)MAX_SPAN_META_VALUE_LEN)SpanLink) SpanLinkKind) _SpanPointer)_SpanPointerDirection)Context)_AttributeValueType)_SAMPLING_AGENT_DECISION)_SAMPLING_LIMIT_DECISION)_SAMPLING_RULE_DECISION)_SPAN_MEASURED_KEY) ERROR_MSG) ERROR_STACK) ERROR_TYPE)MANUAL_DROP_KEY)MANUAL_KEEP_KEY) SERVICE_KEY)SERVICE_VERSION_KEY) USER_KEEP) USER_REJECT) VERSION_KEY)http)net)core) NumericType) ensure_text) is_integer)MAX_INT_64BITS)MAX_UINT_64BITS)MIN_INT_64BITS)SAMPLING_DECISION_TRACE_TAG_KEY)SPAN_API_DATADOG)SamplingMechanism) get_logger)SpanData) SpanEventData)config)Timecs`eZdZgdZ  ddedeeeefdeeffdd Z dd Z d d Z d d Z Z S) SpanEvent)name attributestime_unix_nanoNr1r2r3csDt|||||_|durt}||_|r||_dSi|_dSN)super__init__r1r/time_nsr3r2)selfr1r2r3 __class__G/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/_trace/span.pyr68s zSpanEvent.__init__cCs"|j|jd}|jr|j|d<|S)N)r1r3r2r1r3r2)r8dr;r;r<__dict__Es zSpanEvent.__dict__cCsd|jd|jd|jdS)z~ Stringify and return value. Attribute value can be either str, bool, int, float, or a list of these. zSpanEvent(name='z', time=z , attributes=)r=r8r;r;r<__repr__KszSpanEvent.__repr__ccs4d|jfVd|jfV|jrd|jfVdSdS)Nr1r3r2r=rAr;r;r<__iter__Rs   zSpanEvent.__iter__NN)__name__ __module__ __qualname__ __slots__strrdictrintr6r?rBrC __classcell__r;r;r9r<r05s r0 large_intreturncCst|@S)z2Get the 64 lowest order bits from a 128bit integer)_MAX_UINT_64BITSrMr;r;r< _get_64_lowest_order_bits_as_int\rQcCs|dddS)z3Get the 64 highest order bits from a 128bit integer032xNr;rPr;r;r<!_get_64_highest_order_bits_as_hexasrUc@seZdZgdZdddddddddedf dedeedeedeedeedeed eed eed eed ee e dgdfd edee e ddfddZ dddZ deeddfddZdededdfddZdeeefddfddZdedeefddZdd eeddfd!d"Zd#eddfd$d%Zd&eefd'd(Zd)edeefd*d+Zdded,eeddfd-d.Zded,eeefddfd/d0Zdedeeeeffd1d2Zded,eddfd3d4Z dedeefd5d6Z!deeeffd7d8Z"d9eeefddfd:d;Z#ded,eddfdeeefddfd?d@Z%dedeefdAdBZ& ddedCeeee'fdDeeddfdEdFZ(dGe dgdffdHdIZ)deeeffdJdKZ*ddLeefdMdNZ+ ddOee,dPe,dQee-dLeedef dRdSZ. ddOee,dPe,dQee-dLeeddf dTdUZ/ ddVe,dCeeee'fddfdWdXZ0ded,e1de2fdYdZZ3ded,e4e2eeefde2fd[d\Z5e6dd]d^Z7e7j8dd_d^Z7e7j9dd`d^Z7e6ddadbZ:e:j8ddddbZ:e:j9ddedbZ:dd edCeeeefddfdfdgZ;   ddededheedieedCeeeefddf djdkZ< ddledme=dnedoeeeefddf dpdqZ>dre4e e?fddfdsdtZ@ddudvZAddwdxZBdOeee,dPee,dQee-ddfdydzZCdefd{d|ZDdefd}d~ZEe6de2fddZFdS)Span)_meta _meta_structcontext_metrics_store_parent_context_local_root_value_service_entry_span_value_parent_ignored_exceptions_on_finish_callbacks_links_events __weakref__Nr1serviceresource span_typetrace_idspan_id parent_idstartrY on_finishspan_apilinksrNc Csi|_i|_i|_| durgn| |_| |_|j} |j}| r#| | |nt| |dd|_ g|_ | r:| D]}| |q2g|_ d|_ d|_d|_d|_d|_dS)a Create a new span. Call `finish` once the traced operation is over. **Note:** A ``Span`` should only be accessed or modified in the process that it was created in. Using a ``Span`` from within a child process could result in a deadlock or unexpected behavior. :param str name: the name of the traced operation. :param str service: the service name :param str resource: the resource name :param str span_type: the span type :param int trace_id: the id of this trace's root span. :param int parent_id: the id of this span's direct parent span. :param int span_id: the id of this span. :param int start: the start time of request as a unix epoch in seconds :param object context: the Context of the span. :param on_finish: list of functions called when the span finishes. NF)rhri is_remote)rWrZrXrar\rhricopyrrYrb_set_link_or_append_pointerrcr_r`r]r^r[)r8r1rerfrgrhrirjrkrYrlrmrn _trace_id_span_idnew_linkr;r;r<r6zs*$   z Span.__init__cCst|j-|jjD] }|j||jj|q|jjD] }|j||jj|qWddS1s3wYdSr4)rYrW setdefaultrZ)r8tagmetricr;r;r<_update_tags_from_contexts  "zSpan._update_tags_from_contextexccCs&|jdur |g|_dS|j|dSr4)r`append)r8ryr;r;r<_ignore_exceptions  zSpan._ignore_exceptionkeyvalcCs|jsi|_||j|<dSr4)r[)r8r|r}r;r;r< _set_ctx_itemszSpan._set_ctx_itemitemscCs|jsi|_|j|dSr4)r[update)r8rr;r;r<_set_ctx_itemsszSpan._set_ctx_itemscCs|jsdS|j|Sr4)r[getr8r|r;r;r< _get_ctx_items zSpan._get_ctx_item finish_timecCs0|dur |tdS|t|ddS)zMark the end time of the span and submit it to the tracer. If the span has already been finished don't do anything. :param finish_time: The end time of the span, in seconds. Defaults to ``now``. NgeA) _finish_nsr/r7rK)r8rr;r;r<finishsz Span.finishfinish_time_nscCs6|jdurdS||jp ||_|jD]}||qdSr4) duration_nsstart_nsra)r8rcbr;r;r<rs   zSpan._finish_nsdecisioncCsH||j_|tj|jr tttfD]}||jj vr|jj |=qdSdSr4) rYsampling_priority_set_sampling_decision_makerr*MANUAL _local_rootrrrrZ)r8rr|r;r;r<_override_sampling_decisions   z Span._override_sampling_decisionsampling_mechanismcCsd|}||jjt<|S)Nz-%d)rYrWr()r8rvaluer;r;r<rs z!Span._set_sampling_decision_makerrc Cs`|tjkr t|}t|}tjf}||vr+|s+zt|}d}Wn ttfy*Ynw|dur?|r?t |dkr?| ||dSt |t rL| ||dS|t krW|tdS|tkrb|tdS|tkrj||_n|tkru|t|n|tkr|durd}| ||dSzt||j|<||jvr|j|=WdSWdStytjd|ddYdSw)aSet a tag key/value pair on the span. Keys must be strings, values must be ``str``-able. :param key: Key to use for the tag :type key: str :param value: Value to assign for the tag :type value: ``str``-able value TNlz!error setting tag %s, ignoring itexc_info)r STATUS_CODErIr$r TARGET_PORTrK ValueError TypeErrorabs set_metric isinstancefloatrrrrrrrerset_tagrrrWrZ Exceptionlogwarning)r8r|r val_is_an_int INT_TYPESr;r;r<rsP          z Span.set_tagcCs||j|<dS)z~ Set a tag key/value pair on the span meta_struct Currently it will only be exported with V4 encoding N)rXr8r|rr;r;r<_set_struct_tag>szSpan._set_struct_tagcC|j|dS)z4Return the given struct or None if it doesn't exist.N)rXrrr;r;r<_get_struct_tagEzSpan._get_struct_tagc CsZz t|dd|j|<WdSty,}ztjr|tjd|ddWYd}~dSd}~ww)zSet a value for a tag. Values are coerced to unicode in Python 2 and str in Python 3, with decoding errors in conversion being replaced with U+FFFD. replace)errorszFailed to set text tag '%s'TrN)r#rWrr._raiserr)r8r|rer;r;r< _set_tag_strIszSpan._set_tag_strcCr)z1Return the given tag or None if it doesn't exist.N)rWrrr;r;r<get_tagUrz Span.get_tagcC |jS)zReturn all tags.)rWrprAr;r;r<get_tagsY z Span.get_tagstagscCs.|rt|D] \}}|||qdSdS)zlSet a dictionary of tags on the given span. Keys and values must be strings (or stringable) N)iterrr)r8rkvr;r;r<set_tags]s z Span.set_tagsc Cs|tkr ztt|}Wnttfytd||YdSwt|ttfsAzt|}Wnttfy@t d||YdSwt |sKt |rTt d||dS||j vr]|j |=||j|<dS)z7This method sets a numeric tag value for the given key.z.failed to convert %r tag to an integer from %rNz ignoring not number metric %s:%szignoring not real metric %s:%s)rrKboolrrrrrrdebugmathisnanisinfrWrZrr;r;r<res(  zSpan.set_metricmetricscCs*|r|D] \}}|||qdSdS)zSet a dictionary of metrics on the given span. Keys must be must be strings (or stringable). Values must be numeric. N)rr)r8rrrr;r;r< set_metricss zSpan.set_metricscCs |j|S)z4Return the given metric or None if it doesn't exist.)rZrrr;r;r< get_metrics zSpan.get_metricr2 timestampcCs|jt|||dSr4)rcrzr0)r8r1r2rr;r;r< _add_eventszSpan._add_eventcallbackcCs|jd|dS)zEAdd an errortracking related callback to the on_finish_callback arrayrN)rainsert)r8rr;r;r<!_add_on_finish_exception_callbacksz&Span._add_on_finish_exception_callbackcCr)zReturn all metrics.)rZrprAr;r;r< get_metricsrzSpan.get_metricslimitcCsxt\}}}|r|r|r|rt| }|j||||ddS|dur&tj}dtj|dddd}||j t <dS)zIf the current stack has an exception, tag the span with the relevant error info. If not, tag it with the current python stack. rNr) sysrr set_exc_infor._span_traceback_max_sizejoin traceback format_stackrWr)r8rexc_typeexc_valexc_tbtbr;r;r< set_tracebacks  zSpan.set_tracebackrrrcCs|durtj}t| }t}tj|||||d|}t|tkrJt|dkrJ|d}t}tj|||||d|}t|tkrJt|dks)|S)a Return a formatted traceback as a string. If the traceback is too long, it will be truncated to the limit parameter, but from the end of the traceback (keeping the most recent frames). If the traceback surpasses the MAX_SPAN_META_VALUE_LEN limit, it will try to reduce the traceback size by half until it fits within this limit (limit for tag values). :param exc_type: the exception type :param exc_val: the exception value :param exc_tb: the exception traceback :param limit: the maximum number of frames to keep :return: the formatted traceback as a string N)filerr) r.rrrrprint_exceptiongetvaluelenr )r8rrrrbuffrr;r;r<_get_tracebacks zSpan._get_tracebackcsr|r|sdSttrtt|jdkrdS|jr(tfdd|jDr(dSd|_|j|||d}djj f}||j t <z t ||j t <Wntya|r_t|dr_|jj |j t <Ynw||j t<td |ftd |||fdS) z:Tag the span with an error tuple as from `sys.exc_info()`.Nrcsg|]}t|qSr;) issubclass).0rrr;r< sz%Span.set_exc_info..rr%s.%sr:zweb.request.final_tagszspan.exception)r SystemExitr coder`anyerrorrrFrErWrrIrrhasattrr:rr!dispatch)r8rrrrr exc_type_strr;rr<rs(    zSpan.set_exc_info exceptioncspt|||j}d|jj|jjft||d}|r,fdd|D}||j d|t ddS)a Records an exception as a span event. Multiple exceptions can be recorded on a span. :param exception: The exception to record. :param attributes: Optional dictionary of additional attributes to add to the exception event. These attributes will override the default exception attributes if they contain the same keys. Valid attribute values include (homogeneous array of) strings, booleans, integers, floats. r)zexception.typezexception.messagezexception.stacktracecs"i|] \}}||r||qSr;)_validate_attributerrrrAr;r< s"z)Span.record_exception..r)r1r2rN) rtype __traceback__r:rFrErIrrrr/r7)r8rr2rattrsr;rAr<record_exceptions  zSpan.record_exceptioncCst|ttttfr|||St|tstd||dSt |dkr%dSt|dttttfs9td||dSt |d}|D]}t||rN|||sXtd||dSqAdS)NzHrecord_exception: Attribute %s must be a string, number, or boolean: %s.FrTzHrecord_exception: List values %s must be string, number, or boolean: %s.z=record_exception: Attribute %s array must be homogeneous: %s.) rrIrrKr_validate_scalarlistrrrr)r8r|r first_typer}r;r;r<rs"    zSpan._validate_attributecCspt|ttfr dSt|tr!|tks|tkrtd||dSdSt|tr6t |s4td||dSdSdS)NTzWrecord_exception: Attribute %s must be within the range of a signed 64-bit integer: %s.Fz;record_exception: Attribute %s must be a finite number: %s.) rrrIrK_MIN_INT_64BITS_MAX_INT_64BITSrrrrisfiniterr;r;r<r4s"   zSpan._validate_scalarcC |jp|Sr4r]rAr;r;r<rJrzSpan._local_rootcCs||ur ||_dSd|_dSr4r)r8rr;r;r<rNcC|`dSr4rrAr;r;r<rRrRcCrr4r^rAr;r;r<_service_entry_spanVrzSpan._service_entry_spanspancCs||ur d|_dS||_dSr4rr8rr;r;r<rZrcCrr4rrAr;r;r<r^rRcCsp|jr|jsd|jd|j}tjrt|t||jr4|jr6|j|j|j|jt |j |ddSdSdS)z/Defines a causal relationship between two spansz#Invalid span or trace id. trace_id:z span_id:rhri tracestateflagsr2N) rhrir.rrrrset_link _tracestaterK _traceflags)r8rYr2msgr;r;r< link_spanbs    zSpan.link_spanrrc Cs*|durt}|t|||||ddS)Nr)rJrqr )r8rhrirrr2r;r;r<rtsz Span.set_link pointer_kindpointer_direction pointer_hashextra_attributescCs|t||||ddS)N)rrrr )rqr )r8rrrr r;r;r<_add_span_pointers zSpan._add_span_pointerlinkcCs|jtjjkr|j|dSz#dd|jD|j}t d|j|jt |j|||j|<WdSt yB|j|YdSw)NcSsg|]}|jqSr;)ri)rr r;r;r<rsz4Span._set_link_or_append_pointer..z@Span %d already linked to span %d. Overwriting existing link: %s) kindr SPAN_POINTERrrbrzindexrirrrIr)r8r #existing_link_idx_with_same_span_idr;r;r<rqs   z Span._set_link_or_append_pointercCs*|}|dur||j}|dusdSdS)zFinish this span along with all (accessible) ancestors of this span. This method is useful if a sudden program shutdown is required and finishing the trace is desired. N)rr_rr;r;r<_finish_with_ancestorss zSpan._finish_with_ancestorscCs|Sr4r;rAr;r;r< __enter__szSpan.__enter__cCs@z|r |||||WdStytdYdSw)Nzerror closing trace)rrrrr)r8rrrr;r;r<__exit__s z Span.__exit__cCsdd|jD}dgd|jd|jd|jd|jd|jd |jd |j d |j d |j oJ|j oJ|j |j d |j d|j d|j d|jd|jd|jd|jd|jjd|S)z2Return a detailed string representation of a span.cSs6i|]\}}|t|tr|ndt|jdqS)z wrong type [])rrJkeysrrErr;r;r<rs$z!Span.__repr__..rz Span(name='z ', span_id=z , parent_id=z , trace_id=z , service='z ', resource='z ', type='z ', start=z, end=z , duration=z, error=z, tags=z , metrics=z, links=z , events=z , context=z, service_entry_span_name=z), metastruct=)rXrrr1rirjrhrerfrgrrrrWrZrbrcrYr)r8metar;r;r<rBsN     z Span.__repr__cCsd|j|j|j|jfS)z1Return a concise string representation of a span.z.)rirhrjr1rAr;r;r<__str__s z Span.__str__cCs,|j|up|jduo|jj|jko|jduS)zReturn whether the span is a "top level" span. Top level meaning the root of the trace or a child span whose service is different from its parent. N)rr_rerAr;r;r< _is_top_levels  zSpan._is_top_level)rNNr4rD)rNrV)rrVrNN)rrVrNN)NNN)GrErFrGrHr)rIrrKrrrr r6rxrrr{rr~rJrrrrrr"rrrrrrrrrrrrrrrrrr BaseExceptionrrrrobjectrrrrpropertyrsetterdeleterrrrrr r rqrrrrBrrr;r;r;r<rVfsN      A   B   1 - "      $        rV)Miorrrrtypesrtypingrrrrrr ddtrace._trace._limitsr ddtrace._trace._span_linkr r ddtrace._trace._span_pointerr rddtrace._trace.contextrddtrace._trace.typesrddtrace.constantsrrrrrrrrrrrrrr ddtrace.extrr ddtrace.internalr!ddtrace.internal.compatr"r#r$ddtrace.internal.constantsr%rr&rOr'rr(r)r*ddtrace.internal.loggerr+ddtrace.internal.native._nativer,r-!ddtrace.internal.settings._configr.ddtrace.internal.utils.timer/r0rErrKrQrIrUrVr;r;r;r<sl                                              $