관리-도구
편집 파일: server.cpython-38.pyc
U �Mg� � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl m Z d dl m Z e rxd dlmZmZ d d l mZmZ z,d dlZd d lmZmZ d d lmZmZ W n ek r� ed ��Y nX G d d� dejj�ZdS )� N)�OP)� DidNotEnable)� SPAN_ORIGIN)� Transaction�TRANSACTION_SOURCE_CUSTOM)�event_from_exception)� TYPE_CHECKING)� Awaitable�Callable)�Any�Optional)�HandlerCallDetails�RpcMethodHandler)� AbortError�ServicerContextzgrpcio is not installedc s. e Zd Zd� fdd� Zdd� Zdd� Z� ZS ) �ServerInterceptorNc s |p| j | _t� �� d S �N)� _find_name�_find_method_name�super�__init__)�selfZ find_name�� __class__� �U/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/grpc/aio/server.pyr s zServerInterceptor.__init__c � s� |�_ ||�I d H � � d kr d S � jsB� jsBtj}� �fdd�}n^� jsb� jrbtj}� fdd�}n>� jr�� js�tj}� fdd�}n � jr�� jr�tj}� fdd�}||� j� j d�S )Nc � s� �� |�}|s � | |�I d H S tjt|�� �tj|ttd�}t j |d��� z � � | |�I d H W W 5 Q R � S t k r� � Y nH t k r� } z*t|ddd�d�\}}t j||d� � W 5 d }~X Y nX W 5 Q R X d S )N)�op�name�source�origin)� transaction�grpcF)�typeZhandled)Z mechanism)�hint)r r Zcontinue_from_headers�dictZinvocation_metadatar Z GRPC_SERVERr r � sentry_sdkZstart_transactionZ unary_unaryr � Exceptionr Z capture_event)�request�contextr r �exc�eventr# ��handlerr r r �wrapped( s, � � z4ServerInterceptor.intercept_service.<locals>.wrappedc s& � � | |�2 z3 d H W }|V q 6 d S r )Z unary_stream�r' r( �r�r, r r r- G s c � s � � | |�}|I d H S r )Z stream_unary)r' r( �responser0 r r r- O s c s& � � | |�2 z3 d H W }|V q 6 d S r )Z stream_streamr. r0 r r r- W s )�request_deserializer�response_serializer) �_handler_call_detailsZrequest_streamingZresponse_streamingr! Z unary_unary_rpc_method_handlerZ unary_stream_rpc_method_handlerZ stream_unary_rpc_method_handlerZ stream_stream_rpc_method_handlerr2 r3 )r Z continuationZhandler_call_detailsZhandler_factoryr- r r+ r �intercept_service s* �z#ServerInterceptor.intercept_servicec C s | j jS r )r4 �method)r r( r r r r b s z ServerInterceptor._find_name)N)�__name__� __module__� __qualname__r r5 r � __classcell__r r r r r s Dr ) r% Zsentry_sdk.constsr Zsentry_sdk.integrationsr Z#sentry_sdk.integrations.grpc.constsr Zsentry_sdk.tracingr r Zsentry_sdk.utilsr �typingr �collections.abcr r r r r! r r Zgrpc.aior r � ImportErrorZaior r r r r �<module> s