관리-도구
편집 파일: client.cpython-38.pyc
U �Mg6 � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZ erXd dlm Z m Z m Z m Z m Z z@d dlZd dlmZmZ d dlmZ d d lmZ d d lmZ W n ek r� ed ��Y nX G d d � d ejej�ZdS )� N)�OP)� DidNotEnable)� SPAN_ORIGIN)� TYPE_CHECKING)�Any�Callable�Iterator�Iterable�Union)�ClientCallDetails�Call)� _UnaryOutcome)�UnaryStreamCall)�Messagezgrpcio is not installedc @ s, e Zd ZdZdd� Zdd� Zedd� �ZdS ) �ClientInterceptorFc C sv |j }tjtjd| td��P}|�dd� |�d|� | �|�}|||�}|�d|�� j � |W 5 Q R � S Q R X d S )Nzunary unary call to %s��op�name�origin�typez unary unary�method�code) r � sentry_sdk� start_spanr � GRPC_CLIENTr �set_data�/_update_client_call_details_metadata_from_scoper r ��selfZ continuation�client_call_details�requestr �span�response� r# �Q/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/grpc/client.py�intercept_unary_unary s � � z'ClientInterceptor.intercept_unary_unaryc C sd |j }tjtjd| td��>}|�dd� |�d|� | �|�}|||�}|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 r# r# r$ �intercept_unary_stream/ s" � � �z(ClientInterceptor.intercept_unary_streamc C s^ | j rt| j �ng }t�� �� D ]\}}|�||f� q tjj| j | j || j | j | j d�} | S )N)r �timeout�metadata� credentials�wait_for_ready� compression)r( �listr Zget_current_scopeZ iter_trace_propagation_headers�append�grpcZ _interceptorZ_ClientCallDetailsr r' r) r* r+ )r r( �key�valuer# r# r$ r G s � �� zAClientInterceptor._update_client_call_details_metadata_from_scopeN)�__name__� __module__� __qualname__Z_is_interceptedr% r& � staticmethodr r# r# r# r$ r s r )r Zsentry_sdk.constsr Zsentry_sdk.integrationsr Z#sentry_sdk.integrations.grpc.constsr �typingr r r r r r r. r r Zgrpc._interceptorr Zgrpc.aio._interceptorr Zgoogle.protobuf.messager � ImportErrorZUnaryUnaryClientInterceptorZ UnaryStreamClientInterceptorr r# r# r# r$ �<module> s$ �