a

�,�h��@s�dZd
dl
mZmZmZ
ddlmZmZ
ervddlm	Z	
e	rZd
dl
mZ
d
dlm
Z
mZ
nd
dlmZ
d
dlm
Z
mZ
er�dd	lmZ
eed
�dd�Zeed
�dd�ZdS)z$Trace context propagation utilities.�)�
TYPE_CHECKING�Any�Callable�
)�OTEL_ENABLED�TRACEPARENT_HEADER_NAME)
�OTEL_SYSTEM_AVAILABLE)
�trace)�format_span_id�format_trace_id�)
�MySQLConnectionAbstract)�span�returnc
Cs6|��}
d
}t
|
j�
}t|
j�
}d
}d�||||g�
S)ahBuild a traceparent header according to the provided span.

    The context information from the provided span is used to build the traceparent
    header that will be propagated to the MySQL server. For particulars regarding
    the header creation, refer to [1].

    This method assumes version 0 of the W3C specification.

    Args:
        span (opentelemetry.trace.span.Span): current span in trace.

    Returns:
        traceparent_header (str): HTTP header field that identifies requests in a
        tracing system.

    References:
        [1]: https://www.w3.org/TR/trace-context/#traceparent-header
    Z00�
-)Zget_span_contextr�trace_idr
�span_id�join)r�ctx�versionrrZtrace_flags�r�d/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/opentelemetry/context_propagation.py�build_traceparent_header7s




r)�methodrc
sd
tttd��f
dd�}
|
S)a�
Perform trace context propagation.

    The trace context is propagated via query attributes. The `traceparent` header
    from W3C specification [1] is used, in this sense, the attribute name is
    `traceparent` (is RESERVED, avoid using it), and its value is built as per
    instructed in [1].

    If opentelemetry API/SDK is unavailable or there is no recording span,
    trace context propagation is skipped.

    References:
        [1]: https://www.w3.org/TR/trace-context/#traceparent-header
    r
)�cnx�args�kwargsrc
	s�tr
|j
s �|g
|
�
Ri|�
�
St��}d
}|��rLt|�
}|jt|fd�

z.�|g
|
�
Ri|�
�
}W|d
u
r�|jtd�

n|d
u
r�|jtd�

0|S)zContext propagation decorator.N)
�value)
�name)	rZotel_context_propagationr	Zget_current_spanZis_recordingrZquery_attrs_appendrZquery_attrs_remove)rrrZcurrent_spanZ	tp_header�result�
rrr�wrappercs







�

z)with_context_propagation.<locals>.wrapper)
r)rr!rr r�with_context_propagationTsr"N)�__doc__�typingrrr�	constantsrrZinstrumentationrZ
opentelemetryr	Zopentelemetry.trace.spanr
rZmysql.opentelemetryZmysql.opentelemetry.trace.spanZ	abstractsr
�strrr"rrrr�<module>s