관리-도구
편집 파일: span_processor.cpython-38.pyc
U �Mg�3 � @ s> d dl m Z mZ d dlmZ d dlmZmZ d dlmZ d dlm Z m Z d dl m Z d dlmZmZmZmZ d dlmZmZ d d lmZmZ d d lmZmZ d d l m Z m Z d d l m Z d d l!m"Z"m#Z$ d dl%m&Z& d dl'm(Z) e�rd dlm*Z*m+Z+m,Z, d dl-m.Z/ d dl0m1Z1m2Z2 dZ3dZ4dZ5dd� Z6G dd� de �Z7dS )� )�datetime�timezone)�time)� TYPE_CHECKING�cast)� get_value)� SpanProcessor� ReadableSpan)�SpanAttributes)�format_span_id�format_trace_id�get_current_span�SpanKind)�INVALID_SPAN_ID�INVALID_TRACE_ID)� get_client�start_transaction)� INSTRUMENTER� SPANSTATUS)�SENTRY_BAGGAGE_KEY�SENTRY_TRACE_KEY��add_global_event_processor)� Transaction�Span)�Dsn)� parse_url)�Any�Optional�Union)�context)�Event�HintZotel� z auto.otelc C s� t � }|jd tjkr| S t| d�r4| d dkr4| S t� }|sB| S |�� }|jtks^|j t krb| S |� t |j �d �}|s|| S | � di �}|� di ��|�� � | S )N� instrumenter�typeZ transaction�contexts�trace)r �optionsr �OTEL�hasattrr �get_span_context�trace_idr �span_idr �getr � setdefault�updateZget_trace_context)�event� otel_span_map�client� otel_span�ctx� sentry_spanr& � r7 �b/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/opentelemetry/span_processor.py�!link_trace_context_to_error_event( s" r9 c sz e Zd ZdZi Zi Z� fdd�Zdd� Zdd� Zdd d �Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Z� ZS )�SentrySpanProcessorzZ Converts OTel spans into Sentry spans so they can be sent to the Sentry backend. c s t | d�st� �| �| _| jS )N�instance)r* �super�__new__r; )�cls�� __class__r7 r8 r= P s zSentrySpanProcessor.__new__c s t � fdd��}d S )Nc s t | � j�S )N)r9 r2 )r1 �hint��selfr7 r8 �global_event_processorY s z<SentrySpanProcessor.__init__.<locals>.global_event_processorr )rC rD r7 rB r8 �__init__W s z SentrySpanProcessor.__init__c C sp t t� d �}t| j�� �D ]N}| j| t� kr>| j�|� q || tkr | j�|�D ]}| j�|d� qVq dS )z? Prune spans that have been open for too long. �< N) �intr �list� open_spans�keys�set�pop�SPAN_MAX_TIME_OPEN_MINUTESr2 )rC Zcurrent_time_minutesZspan_start_minutesr- r7 r7 r8 �_prune_old_spans^ s � z$SentrySpanProcessor._prune_old_spansNc C sP t � }|jsd S zt|j�}W n tk r4 Y d S X |jd tjkrJd S |�� jsXd S | � |�rfd S | � ||�}|d }|r�| j � |�nd }d }|j d k r�t�|j d tj�}d } |r�|j|d |j|tjtd�} n(t|j|d ||d |d |tjtd�} | | j |d < |j d k �rDt|j d d �} | j�| t� ��|d � | �� d S ) Nr$ �parent_span_id� e��Ar- )r- �name�start_timestampr$ �originr, �baggage)rQ r- rO r, rT rR r$ rS rF ) r �dsnr � Exceptionr( r r) r+ �is_valid�_is_sentry_span�_get_trace_datar2 r. � start_timer � fromtimestampr �utcZ start_childrQ � SPAN_ORIGINr rG rI r/ rK �addrN ) rC r4 �parent_contextr3 �_� trace_datarO Zsentry_parent_spanrR r6 �span_start_in_minutesr7 r7 r8 �on_startp sf � ��� ��z SentrySpanProcessor.on_startc C s t � }|jd tjkrd S |�� }|js,d S t|j�}| j� |d �}|sLd S |j |_ | � ||� t |t�r�|j |_ |�t| �|�� | �||� n | �||� d }|jd k r�t�|jd tj�}|j|d� |jd k r�t|jd d �}| j � |t � �� |� | � � d S )Nr$ rP )� end_timestamprF )!r r( r r) r+ rW r r- r2 rL rQ �op� _update_span_with_otel_status� isinstancer Z set_context�OPEN_TELEMETRY_CONTEXT�_get_otel_context�"_update_transaction_with_otel_data�_update_span_with_otel_dataZend_timer r[ r r\ �finishrZ rG rI r/ rK �discardrN )rC r4 r3 � span_contextr- r6 rd rb r7 r7 r8 �on_end� sB � � �zSentrySpanProcessor.on_endc C sZ d}|j dk r |j �tj�}td|�}d}t� }|jrBt|j�j}|rV|rV||krVdS dS )zs Break infinite loop: HTTP requests to Sentry are caught by OTel and send again to Sentry. N� Optional[str]TF) � attributesr. r �HTTP_URLr r rU r �netloc)rC r4 Z otel_span_urlZdsn_urlr3 r7 r7 r8 rX � s z#SentrySpanProcessor._is_sentry_spanc C s4 i }|j rt|j �|d< |jj r0t|jj �|d< |S )z� Returns the OTel context for Sentry. See: https://develop.sentry.dev/sdk/performance/opentelemetry/#step-5-add-opentelemetry-context rq �resource)rq �dictrt )rC r4 r5 r7 r7 r8 ri � s z%SentrySpanProcessor._get_otel_contextc C s� i }|� � }t|j�}||d< t|j�}||d< |jrBt|jj�nd}||d< tt|�}td|�}|rn|d nd|d< tt |�} | |d< |S )z^ Extracts tracing information from one OTel span and its parent OTel context. r- r, NrO z!dict[str, Union[str, bool, None]]Zparent_sampledrT ) r+ r r- r r, �parentr r r r ) rC r4 r_ ra rn r- r, rO Zsentry_trace_datarT r7 r7 r8 rY s � � z#SentrySpanProcessor._get_trace_datac C s4 |j jr dS |j jr$|�tj� dS |�tj� dS )z? Set the Sentry span status from the OTel span N)�statusZis_unsetZis_okZ set_statusr �OK�INTERNAL_ERROR)rC r6 r4 r7 r7 r8 rf $ s z1SentrySpanProcessor._update_span_with_otel_statusc C s� |� d|j� |j}|j}|jdk �r�|j�� D ]\}}|� ||� q0|j�tj�}td|�}|j�tj �}|�rjd}|jt j kr�|d7 }n|jt j kr�|d7 }|}|j�tj d�} | r�|d�| �7 }|j�tjd�} | r�|d�| �7 }| �s>| �s>|j�tjd�} td| �} | �r>t| �} d�| j| j| j�} |d�| �7 }|j�tjd�} td | �} | �r�|�| � n.|�r�d }|j�tjd�}td|�}|�r�|}||_||_dS ) z� Convert OTel span data and update the Sentry span with it. This should eventually happen on the server when ingesting the spans. z otel.kindNrp �http�.server�.clientz {}z {}://{}{}� Optional[int]�db)�set_data�kindrQ rq �itemsr. r � HTTP_METHODr Z DB_SYSTEMr �SERVER�CLIENTZ NET_PEER_NAME�formatZ HTTP_TARGETrr �urlparse�schemers �path�HTTP_STATUS_CODE�set_http_statusZ DB_STATEMENTre � description)rC r6 r4 re r� �key�val� http_methodZdb_queryZ peer_name�target�url� parsed_url� status_codeZ statementr7 r7 r8 rk 2 s` � � z/SentrySpanProcessor._update_span_with_otel_datac C s~ |j d krd S |j �tj�}|rz|j �tj�}td|�}|rF|�|� d}|jtj kr`|d7 }n|jtj krt|d7 }||_ d S )Nr} rz r{ r| ) rq r. r r� r� r r� r� r r� r� re )rC r6 r4 r� r� re r7 r7 r8 rj s s z6SentrySpanProcessor._update_transaction_with_otel_data)N)�__name__� __module__� __qualname__�__doc__r2 rI r= rE rN rc ro rX ri rY rf rk rj � __classcell__r7 r7 r? r8 r: E s C. Ar: N)8r r r �typingr r Zopentelemetry.contextr Zopentelemetry.sdk.tracer r ZOTelSpanZopentelemetry.semconv.tracer Zopentelemetry.tracer r r r Zopentelemetry.trace.spanr r Z sentry_sdkr r Zsentry_sdk.constsr r Z,sentry_sdk.integrations.opentelemetry.constsr r Zsentry_sdk.scoper Zsentry_sdk.tracingr r Z SentrySpanZsentry_sdk.utilsr Z urllib3.utilr r� r r r Z opentelemetryr Z context_apiZsentry_sdk._typesr! r" rh rM r] r9 r: r7 r7 r7 r8 �<module> s.