관리-도구
편집 파일: client.cpython-38.pyc
U �Mg� � @ s� d dl mZmZmZmZ d dlmZmZmZm Z m Z d dl m Z d dl Z d dlmZ d dlmZ G dd� d�ZG d d � d ee�ZG d d � d ee�ZdS ) � )�Callable�Union� AsyncIterable�Any)�UnaryUnaryClientInterceptor� UnaryStreamClientInterceptor�ClientCallDetails�UnaryUnaryCall�UnaryStreamCall)�MessageN)�OP)� SPAN_ORIGINc @ s e Zd Zeeed�dd��ZdS )�ClientInterceptor)�client_call_details�returnc C sV | j rt| j �ng }t�� �� D ]\}}|�||f� q t| j| j|| j | j d�} | S )N)�method�timeout�metadata� credentials�wait_for_ready) r �list� sentry_sdkZget_current_scopeZ iter_trace_propagation_headers�appendr r r r r )r r �key�value� r �U/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/grpc/aio/client.py�/_update_client_call_details_metadata_from_scope s � ��zAClientInterceptor._update_client_call_details_metadata_from_scopeN)�__name__� __module__� __qualname__� staticmethodr r r r r r r s �r c @ s4 e Zd Zeeegef eeeeef d�dd�ZdS )�!SentryUnaryUnaryClientInterceptor�� continuationr �requestr c � s� |j }tjtjd|�� td��`}|�dd� |�d|� | �|�}|||�I d H }|� � I d H }|�d|j � |W 5 Q R � S Q R X d S )Nzunary unary call to %s��op�name�origin�typez unary unaryr �code) r r � start_spanr � GRPC_CLIENT�decoder �set_datar r+ r( )�selfr$ r r% r �span�response� status_coder r r �intercept_unary_unary+ s � �z7SentryUnaryUnaryClientInterceptor.intercept_unary_unaryN) r r r r r r r r r4 r r r r r" * s �r" c @ s8 e Zd Zeeegef eeeee ef d�dd�Z dS )�"SentryUnaryStreamClientInterceptorr# c � sn |j }tjtjd|�� td��D}|�dd� |�d|� | �|�}|||�I d H }|W 5 Q R � S Q R X d S )Nzunary stream call to %sr& r* z unary streamr ) r r r, r r- r. r r/ r )r0 r$ r r% r r1 r2 r r r �intercept_unary_streamI s � �z9SentryUnaryStreamClientInterceptor.intercept_unary_streamN) r r r r r r r r r r r6 r r r r r5 F s �r5 )�typingr r r r Zgrpc.aior r r r r Zgoogle.protobuf.messager r Zsentry_sdk.constsr Z#sentry_sdk.integrations.grpc.constsr r r" r5 r r r r �<module> s �