a

�,�h��@szdd
lm
Z

ddlmZmZmZ
ddlmZ
ddlm	Z	m
Z

dZed�
Zd
e
e
eed�dd	�Z
de
ee
d
�dd�ZdS)�)
�Optional)�
create_key�	get_value�	set_value)
�Context)�INVALID_SPAN�Spanzcurrent-spanN)�span�context�returncCstt
||
d
�}|S)z�Set the span in the given context.

    Args:
        span: The Span to set.
        context: a Context object. if one is not passed, the
            default current context is used instead.
    �
r
)r�	_SPAN_KEY)r	r
�ctx�r�a/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/opentelemetry/trace/propagation/__init__.py�set_span_in_contexts
r)r
rc
Cs&tt
|d
�}
|
dust|
t�s"tS|
S)z�Retrieve the current span.

    Args:
        context: A Context object. If one is not passed, the
            default current context is used instead.

    Returns:
        The Span set in the context if it exists. INVALID_SPAN otherwise.
    rN)rr
�
isinstancerr)r
r	rrr�get_current_span$s



r)
N)
N)�typingrZmysql.opentelemetry.contextrrrZ#mysql.opentelemetry.context.contextrZmysql.opentelemetry.trace.spanrrZSPAN_KEYr
rrrrrr�<module>s