ruạṛ
a K��f�� � @ s, d Z ddlmZ ddlZddlZddlmZ ddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZmZmZmZmZmZmZmZmZmZmZmZ ddlZddlmZ ddlm Z ddlm Z dd lm Z dd l m Z dd l!m"Z" dd l!m#Z# dd l!m$Z$ ddl!m%Z% ddl!m&Z& ddl!m'Z' ddl!m(Z( ddl!m)Z) ddl!m*Z* e �+e,�Z-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5d Z6d Z7d Z8d Z9dZ:d Z;d!Z<d"d#d$�d%d&�Z=d'd(d)�d*d+�Z>d,d(d-�d.d/�Z?d,d(d(d0�d1d2�Z@d,d#d-�d3d4�ZAG d5d6� d6e�Bd6d7�ejC�ZDG d8d9� d9ejE�ZFG d:d;� d;eF�ZGG d<d=� d=eF�ZHG d>d,� d,eI�ZJd,d?d-�d@dA�ZKd,dBdCdD�dEdF�ZLd,dBdGdD�dHdI�ZMd,dJdJdK�dLdM�ZNd,dJdNdK�dOdP�ZOd,dQd(d#d?dR�dSdT�ZPd,dGd-�dUdV�ZQd,dQdWdGdX�dYdZ�ZRd,dGd-�d[d\�ZSd,dBdGdD�d]d^�ZTG d_d`� d`ejU�ZVG dadb� dbeI�ZWd"d,dWdcdd�dedf�ZXd�d"d,dgdhdWdidjdk�dldm�ZYd"d,dndodp�dqdr�ZZd"d,dhdsdtdu�dvdw�Z[d,dxd-�dydz�Z\d,d?d-�d{d|�Z]d"d,d#d}d~�dd��Z^d"d,dtd?d~�d�d��Z_d"d,dgdcdsdsd?d��d�d��Z`d"d,dgdcdWdsd?d��d�d��Zad,d}d-�d�d��Zbd"d,d�dnd?d��d�d��Zcdgd�d�d��d�d��Zdd"d,d�d�d�d��d�d��Zed"d,d�d�d�d��d�d��Zfd"d,d�d�d�d��d�d��Zgd"d,d�d�d�d��d�d��Zhd"d,d9d�d�d��d�d��Zid"d,d(d#d��d�d��Zjd"d,d�d�d�d��d�d��Zkd"d9d�d�d}d�d��d�d��ZlejmG d�d�� d�ejn��ZoG d�d�� d�eI�Zpd�d�d?d��d�d��Zqd�d�d?d��d�d��Zrd�d#d�d��d�d��Zsd�d#d�d�d��d�d��Ztd�d?d-�d�d��Zud�dBd?dd�dĄZvd�d}d-�d�dƄZwd�d?d-�d�dȄZxd�d"d}dɜd�d˄Zyd�d?d-�d�d̈́Zzd�d?d-�d�dτZ{d�d�d�dҜd�dԄZ|d�d?d-�d�dքZ}d�d?dלd�dلZ~d�d�d}d�dܜd�dބZG d�d� d�ej��Z�d�d�d�d�d�d�d}d�d�d�d�Z�dS )�z+Service-side implementation of gRPC Python.� )� annotationsN)�futures) �Any�Callable�Dict�Iterable�Iterator�List�Mapping�Optional�Sequence�Set�Tuple�Union)�_common)� _compression)� _interceptor)�_observability)�cygrpc)�ArityAgnosticMethodHandler)�ChannelArgumentType)�DeserializingFunction)� MetadataType)�NullaryCallbackType)� ResponseType)�SerializingFunction)�ServerCallbackTag)�ServerTagCallbackType�shutdown� request_call�receive_close_on_server�send_initial_metadata�receive_message� send_messagez$send_initial_metadata * send_message�send_status_from_serverz/send_initial_metadata * send_status_from_server�open�closed� cancelledg �?g e��A�cygrpc.BaseEvent�bytes)� request_event�returnc C s | j d �� S �Nr )�batch_operations�message)r* � r/ �p/sparta/input/_build_configuration/image_build+validate/lib/bmcenv/lib64/python3.9/site-packages/grpc/_server.py�_serialized_requestU s r1 �grpc.StatusCodezcygrpc.StatusCode��coder+ c C s t j�| �}|d u r tjjS |S �N)r �!STATUS_CODE_TO_CYGRPC_STATUS_CODE�getr � StatusCode�unknown)r4 � cygrpc_coder/ r/ r0 �_application_codeY s r; � _RPCState)�stater+ c C s | j d u rtjjS t| j �S d S r5 )r4 r r8 �okr; �r= r/ r/ r0 �_completion_code^ s r@ )r= r4 r+ c C s | j d u r|S t| j �S d S r5 )r4 r; )r= r4 r/ r/ r0 �_abortion_codee s rA c C s | j d u rdS | j S )N� )�detailsr? r/ r/ r0 �_detailsn s rD c @ s e Zd ZdS )�_HandlerCallDetailsN)�__name__� __module__� __qualname__r/ r/ r/ r0 rE r s rE )�method�invocation_metadatac @ s6 e Zd Zejdd�dd��Zejddd�dd ��Zd S ) �_Method� Optional[str]�r+ c C s t � �d S r5 ��NotImplementedError��selfr/ r/ r0 �name� s z _Method.namerE � Optional[grpc.RpcMethodHandler]��handler_call_detailsr+ c C s t � �d S r5 rN �rQ rU r/ r/ r0 �handler� s z_Method.handlerN)rF rG rH �abc�abstractmethodrR rW r/ r/ r/ r0 rK s rK c @ s: e Zd Zddd�dd�Zdd�dd �Zd dd �d d �ZdS )�_RegisteredMethod�strrS )rR �registered_handlerc C s || _ || _d S r5 )�_name�_registered_handler)rQ rR r\ r/ r/ r0 �__init__� s z_RegisteredMethod.__init__rL rM c C s | j S r5 )r] rP r/ r/ r0 rR � s z_RegisteredMethod.namerE rT c C s | j S r5 )r^ rV r/ r/ r0 rW � s z_RegisteredMethod.handlerN�rF rG rH r_ rR rW r/ r/ r/ r0 rZ � s rZ c @ s8 e Zd Zdd�dd�Zdd�dd�Zd d d �d d �ZdS )�_GenericMethod� List[grpc.GenericRpcHandler])�generic_handlersc C s || _ d S r5 )�_generic_handlers)rQ rc r/ r/ r0 r_ � s z_GenericMethod.__init__rL rM c C s d S r5 r/ rP r/ r/ r0 rR � s z_GenericMethod.namerE rS rT c C s* | j D ] }|�|�}|d ur| S qd S r5 )rd �service)rQ rU �generic_handler�method_handlerr/ r/ r0 rW � s z_GenericMethod.handlerNr` r/ r/ r/ r0 ra � s ra c @ s� e Zd ZU ded< ded< ee Zded< ded< d ed < d ed < d ed < ded< ded< ded< d ed< ded< ded< d ed< dd� Zd S ) r< zcontextvars.Context�contextzthreading.Condition� conditionr �requestr[ �client�bool�initial_metadata_allowed�Optional[grpc.Compression]�compression_algorithm�disable_next_compression�Optional[MetadataType]�trailing_metadatazOptional[grpc.StatusCode]r4 �Optional[bytes]rC �statusedzList[Exception]� rpc_errorsz#Optional[List[NullaryCallbackType]]� callbacks�abortedc C sh t �� | _t�� | _t� | _d | _t | _ d| _ d | _ d| _ d | _d | _d | _d| _g | _g | _d| _d S �NTF)� contextvars�Contextrh � threading� Conditionri �set�duerj �_OPENrk rm ro rp rr r4 rC rt ru rv rw rP r/ r/ r0 r_ � s z_RPCState.__init__N)rF rG rH �__annotations__r r[ r~ r_ r/ r/ r/ r0 r< � s �Nonec C s t �� }| j�|� |�d S r5 )�grpc�RpcErrorru �append)r= � rpc_errorr/ r/ r0 �_raise_rpc_error� s r� r[ r )r= �tokenr+ c C s6 | j �|� t| �s.| j s.| j}d | _| |fS dS d S )N�Nr/ )r~ �remove�_is_rpc_state_activerv )r= r� rv r/ r/ r0 �_possibly_finish_call� s r� r c s � �fdd�}|S )Nc s4 � j � t� ��W d � S 1 s&0 Y d S r5 )ri r� )�$unused_send_status_from_server_event�r= r� r/ r0 r$ � s z9_send_status_from_server.<locals>.send_status_from_serverr/ )r= r� r$ r/ r� r0 �_send_status_from_server� s r� rq )r= �metadatar+ c C s� | j �l | jrRt�| j�f}|d u r6|W d � S |t|� W d � S n|W d � S W d � n1 sx0 Y d S r5 )ri ro r �!compression_algorithm_to_metadata�tuple)r= r� �compression_metadatar/ r/ r0 �_get_initial_metadata� s �� r� zcygrpc.Operationc C s t �t| |�t�}|S r5 )r � SendInitialMetadataOperationr� � _EMPTY_FLAGS)r= r� � operationr/ r/ r0 � _get_initial_metadata_operation s �r� � cygrpc.Call)r= �callr4 rC r+ c C s� | j tur�t| |�}| jd u r"|n| j}| jrPt| d �t�| j||t �f}t }nt�| j||t �f}t }|� |t | |�� d| _| j�|� d S �NT)rk � _CANCELLEDrA rC rm r� r � SendStatusFromServerOperationrr r� �8_SEND_INITIAL_METADATA_AND_SEND_STATUS_FROM_SERVER_TOKEN� _SEND_STATUS_FROM_SERVER_TOKEN�start_server_batchr� rt r~ �add)r= r� r4 rC �effective_code�effective_details� operationsr� r/ r/ r0 �_abort s4 �� �� �r� c s � fdd�}|S )Nc sd � j �J | jd �� r t� _n� jtu r.t� _� j �� t� t �W d � S 1 sV0 Y d S r, ) ri r- r' r� rk r �_CLOSED� notify_allr� � _RECEIVE_CLOSE_ON_SERVER_TOKEN)� receive_close_on_server_eventr? r/ r0 r . s z9_receive_close_on_server.<locals>.receive_close_on_serverr/ )r= r r/ r? r0 �_receive_close_on_server- s r� � Optional[DeserializingFunction])r= r� �request_deserializerr+ c s � ��fdd�}|S )Nc s� t | �}|d u r\�j�4 �jtu r(t�_�j�� t�t�W d � S 1 sP0 Y nht� |��}�j�F |d u r�t �� t j j d� n|�_�j�� t�t�W d � S 1 s�0 Y d S )Ns Exception deserializing request!)r1 ri rk r r� r� r� �_RECEIVE_MESSAGE_TOKENr � deserializer� r r8 �internalrj )�receive_message_event�serialized_requestrj �r� r� r= r/ r0 r" ? s* *�� z)_receive_message.<locals>.receive_messager/ )r= r� r� r" r/ r� r0 �_receive_message: s r� c s � fdd�}|S )Nc s4 � j � t� t�W d � S 1 s&0 Y d S r5 )ri r� � _SEND_INITIAL_METADATA_TOKEN)�"unused_send_initial_metadata_eventr? r/ r0 r! \ s z5_send_initial_metadata.<locals>.send_initial_metadatar/ )r= r! r/ r? r0 �_send_initial_metadata[ s r� c s � �fdd�}|S )Nc s> � j �$ � j �� t� ��W d � S 1 s00 Y d S r5 )ri r� r� )�unused_send_message_eventr� r/ r0 r# d s z#_send_message.<locals>.send_messager/ )r= r� r# r/ r� r0 � _send_messagec s r� c @ sp e Zd ZU ded< ded< ded< dddd�dd �Zd d �d d �Zdd �dd�Zdd �dd�Zdd d�dd�Zdd �dd�Z dd �dd �Z d d �d d �Z d d �d!d"�Z d#d �d$d%�Z d&d �d'd(�Zd)dd*�d+d,�Zd-dd.�d/d0�Zd-dd1�d2d3�Zdd �d4d5�Zd6d dd7�d8d9�Zd:dd;�d<d=�Zd6dd>�d?d@�Zd6d �dAdB�Zd ddC�dDdE�ZdFd �dGdH�Zdd �dIdJ�ZdKS )L�_Contextr( � _rpc_eventr< �_stater� r� )� rpc_eventr= r� c C s || _ || _|| _d S r5 )r� r� �_request_deserializer)rQ r� r= r� r/ r/ r0 r_ q s z_Context.__init__rl rM c C s6 | j j� t| j �W d � S 1 s(0 Y d S r5 )r� ri r� rP r/ r/ r0 � is_active{ s z_Context.is_active�floatc C s t | jjjt�� d�S r, )�maxr� � call_details�deadline�timerP r/ r/ r0 �time_remaining s z_Context.time_remainingr� c C s | j j�� d S r5 )r� r� �cancelrP r/ r/ r0 r� � s z_Context.cancelr )�callbackr+ c C sf | j j�J | j jd u r&W d � dS | j j�|� W d � dS W d � n1 sX0 Y d S )NFT)r� ri rv r� )rQ r� r/ r/ r0 � add_callback� s z_Context.add_callbackc C s4 | j j� d| j _W d � n1 s&0 Y d S r� )r� ri rp rP r/ r/ r0 � disable_next_message_compression� s z)_Context.disable_next_message_compressionrq c C s | j jS r5 )r� rJ rP r/ r/ r0 rJ � s z _Context.invocation_metadatar[ c C s t �| jj�� �S r5 )r �decoder� r� �peerrP r/ r/ r0 r� � s z _Context.peerzOptional[Sequence[bytes]]c C s t �| jj�S r5 )r �peer_identitiesr� r� rP r/ r/ r0 r� � s z_Context.peer_identitiesrL c C s$ t �| jj�}|d u r|S t�|�S r5 )r �peer_identity_keyr� r� r r� )rQ �id_keyr/ r/ r0 r� � s z_Context.peer_identity_keyz Mapping[str, Sequence[bytes]]c C s0 t �| jj�}|d u ri n|}dd� |�� D �S )Nc S s i | ]\}}t �|�|�qS r/ )r r� )�.0�key�valuer/ r/ r0 � <dictcomp>� s �z)_Context.auth_context.<locals>.<dictcomp>)r � auth_contextr� r� �items)rQ r� �auth_context_dictr/ r/ r0 r� � s �z_Context.auth_contextzgrpc.Compression)� compressionr+ c C s4 | j j� || j _W d � n1 s&0 Y d S r5 )r� ri ro )rQ r� r/ r/ r0 �set_compression� s z_Context.set_compressionr )�initial_metadatar+ c C s� | j j�t | j jtu r"t| j � nL| j jrft| j |�}| jj� |ft | j �� d| j _| j j � t � ntd��W d � n1 s�0 Y d S )NFz#Initial metadata no longer allowed!)r� ri rk r� r� rm r� r� r� r� r� r~ r� r� � ValueError)rQ r� r� r/ r/ r0 r! � s � �z _Context.send_initial_metadata)rr r+ c C s4 | j j� || j _W d � n1 s&0 Y d S r5 )r� ri rr )rQ rr r/ r/ r0 �set_trailing_metadata� s z _Context.set_trailing_metadatac C s | j jS r5 )r� rr rP r/ r/ r0 rr � s z_Context.trailing_metadatar2 )r4 rC r+ c C sr |t jjkr"t�d� t jj}d}| jj�4 || j_t � |�| j_ d| j_ t � �W d � n1 sd0 Y d S )Nz4abort() called with StatusCode.OK; returning UNKNOWN� T)r� r8 �OK�_LOGGER�error�UNKNOWNr� ri r4 r �encoderC rw � Exception)rQ r4 rC r/ r/ r0 �abort� s � z_Context.abortz grpc.Status)�statusr+ c C s |j | j_ | �|j|j� d S r5 )rr r� r� r4 rC )rQ r� r/ r/ r0 �abort_with_status� s z_Context.abort_with_statusr3 c C s4 | j j� || j _W d � n1 s&0 Y d S r5 )r� ri r4 )rQ r4 r/ r/ r0 �set_code� s z_Context.set_codec C s | j jS r5 )r� r4 rP r/ r/ r0 r4 � s z _Context.code)rC r+ c C s: | j j� t�|�| j _W d � n1 s,0 Y d S r5 )r� ri r r� rC )rQ rC r/ r/ r0 � set_details� s z_Context.set_detailsr) c C s | j jS r5 )r� rC rP r/ r/ r0 rC � s z_Context.detailsc C s d S r5 r/ rP r/ r/ r0 �_finalize_state� s z_Context._finalize_stateN)rF rG rH r� r_ r� r� r� r� r� rJ r� r� r� r� r� r! r� rr r� r� r� r4 r� rC r� r/ r/ r/ r0 r� l s2 r� c @ s� e Zd ZU ded< ded< ded< dddd�dd �Zd d �d d �Zdd �dd�Zdd �dd�Zd d �dd�Zdd �dd�Z dd �dd�Z dS )�_RequestIteratorr< r� r� �_callr� r� )r= r� r� c C s || _ || _|| _d S r5 )r� r� r� )rQ r= r� r� r/ r/ r0 r_ � s z_RequestIterator.__init__r� rM c C s` | j jtu rt| j � nDt| j �s*t� �n2| j�t� t �ft | j | j| j �� | j j �t� d S r5 )r� rk r� r� r� � StopIterationr� r� r �ReceiveMessageOperationr� r� r� r~ r� r� rP r/ r/ r0 � _raise_or_start_receive_message� s ��z0_RequestIterator._raise_or_start_receive_messager c C sV | j jtu rt| j � n4| j jd u r8t| j jvr8t� �n| j j}d | j _|S t� �d S r5 ) r� rk r� r� rj r� r~ r� �AssertionError�rQ rj r/ r/ r0 �_look_for_request s � �z"_RequestIterator._look_for_requestc C sd | j j�H | �� | j j�� | �� }|d ur|W d � S qW d � n1 sV0 Y d S r5 )r� ri r� �waitr� r� r/ r/ r0 �_next s z_RequestIterator._nextc C s | S r5 r/ rP r/ r/ r0 �__iter__ s z_RequestIterator.__iter__c C s | � � S r5 �r� rP r/ r/ r0 �__next__ s z_RequestIterator.__next__c C s | � � S r5 r� rP r/ r/ r0 �next! s z_RequestIterator.nextN) rF rG rH r� r_ r� r� r� r� r� r� r/ r/ r/ r0 r� � s r� zCallable[[], Any])r� r= r� r+ c s � ��fdd�}|S )Nc s �j �� t��s W d � d S �j�t�t�ft��j� �� �j� t � �j � � �j d u r��j tu r�d��jj�} t��jtjjt�| �� W d � d S �j tu r�W d � d S qL�j }d �_ |W d � S qLW d � n1 s�0 Y d S )Nz*"{}" requires exactly one request message.)ri r� r� r� r r� r� r� r~ r� r� r� rj rk r� �formatr� rI r� r8 � unimplementedr r� r� )rC rj �r� r� r= r/ r0 � unary_request* s8 �� �� z%_unary_request.<locals>.unary_requestr/ )r� r= r� r� r/ r� r0 �_unary_request% s !r� r r z(Optional[Callable[[ResponseType], None]]z8Tuple[Union[ResponseType, Iterator[ResponseType]], bool])r� r= �behavior�argumentr� �send_response_callbackr+ c C sn ddl m} || ||���>}z<d }|d ur8||||�}n |||�}|dfW W d � S t�yH } z�|j�� |jr�t|| jtjj d� nx| |j v�rzd� | �} W n8 ty� d} t � t�t| �| | j�� t�� Y n0 t � | � t|| jtjj t�| �� W d � n1 �s 0 Y W Y d } ~ W d � dS d } ~ 0 0 W d � n1 �s`0 Y d S )Nr )�_create_servicer_contextT� RPC Abortedz!Exception calling application: {}z1Calling application raised unprintable Exception!�NF)r� r� r� ri rw r� r� r r8 r9 ru r� r� � exception� traceback�format_exception�type� __traceback__� print_excr r� ) r� r= r� r� r� r� r� rh �response_or_iteratorr� rC r/ r/ r0 �_call_behaviorN s\ �� � � ��� �$r zIterator[ResponseType]zTuple[ResponseType, bool])r� r= �response_iteratorr+ c C s� zt |�dfW S ty" Y dS ty� } z�|j�d |jrTt|| jtjj d� n8||j vr�d� |�}t � |� t|| jtjj t�|�� W d � n1 s�0 Y W Y d }~dS d }~0 0 d S )NTr� r� z!Exception iterating responses: {}r� )r� r� r� ri rw r� r� r r8 r9 ru r� r� r� r r� )r� r= r r� rC r/ r/ r0 �%_take_response_from_response_iterator� s. � �"r z Optional[SerializingFunction]rs )r� r= �response�response_serializerr+ c C sZ t �||�}|d u rR|j�$ t|| jtjjd� W d � n1 sD0 Y d S |S d S )Ns Failed to serialize response!)r � serializeri r� r� r r8 r� )r� r= r r �serialized_responser/ r/ r0 �_serialize_response� s �"r z Union[int, cygrpc.WriteFlag]c C s | j rtjjS tS d S r5 )rp r � WriteFlag� no_compressr� r? r/ r/ r0 �%_get_send_message_op_flags_from_state� s r c C s0 | j � d| _W d � n1 s"0 Y d S r� )ri rp r? r/ r/ r0 �_reset_per_message_state� s r rl )r� r= r r+ c C s� |j �� t|�s W d � dS |jrLt|d �t�|t|��f}d|_t}nt�|t|��f}t}| j � |t ||�� |j � |� t|� |j �� ||j vr�t|�W d � S q�W d � n1 s�0 Y d S r� )ri r� rm r� r �SendMessageOperationr �-_SEND_INITIAL_METADATA_AND_SEND_MESSAGE_TOKEN�_SEND_MESSAGE_TOKENr� r� r� r~ r� r r� )r� r= r r� r� r/ r/ r0 �_send_response� s4 ���� � r c C s� |j �� |jtur�t|�}t|�}t�|j||t�g}|j rL|� t |d �� |d urj|� t� |t |��� | j�|t|t�� d|_t|� |j�t� W d � n1 s�0 Y d S r� )ri rk r� r@ rD r r� rr r� rm r� r� r r r� r� r� r� rt r r~ r� )r� r= r r4 rC r� r/ r/ r0 �_status� s0 �����r )r� r= r� �argument_thunkr� r r+ c C s� t �| � ztzL|� }|d urVt| ||||�\}}|rVt| |||�} | d urVt| || � W n tyr t�� Y n0 W t �� n t �� 0 d S r5 ) r �'install_context_from_request_call_eventr r r r� r r �uninstall_context) r� r= r� r r� r r� r �proceedr r/ r/ r0 �_unary_response_in_pool s �� r c s� t ��� ddd�� ��fdd� }z�z^|� }|d ur�t|d�rZ|jrZt��||||d� n&t��|||�\}} | r�t��||� W n ty� t�� Y n0 W t � � n t � � 0 d S )Nr r� )r r+ c s< | d u rt ��d � n"t��| � �}|d ur8t��|� d S r5 )r r r )r r �r r� r= r/ r0 � send_response2 s �z/_stream_response_in_pool.<locals>.send_response�experimental_non_blocking)r� ) r r �hasattrr r �3_send_message_callback_to_blocking_iterator_adapterr� r r r ) r� r= r� r r� r r r� r r r/ r r0 �_stream_response_in_pool( s6 ��� �� r! c C s | j tuo| j S r5 )rk r� rt r? r/ r/ r0 r� Y s r� z Callable[[ResponseType], None])r� r= r� r r+ c C s0 t | ||�\}}|r,||� t|�s*q,q q,q d S r5 )r r� )r� r= r� r r r r/ r/ r0 r ] s �r �futures.ThreadPoolExecutor)r� �default_thread_poolr+ c C s& t | d�r t| jtj�r | jS |S d S )N�experimental_thread_pool)r � isinstancer$ r �ThreadPoolExecutor)r� r# r/ r/ r0 � _select_thread_pool_for_behavioro s �r' zgrpc.RpcMethodHandlerzfutures.Future)r� r= rg r# r+ c C s< t | ||j�}t|j|�}|�|jjt| ||j||j|j�S r5 ) r� r� r' � unary_unary�submitrh �runr r �r� r= rg r# r� � thread_poolr/ r/ r0 �_handle_unary_unary{ s ���r- c C s< t | ||j�}t|j|�}|�|jjt| ||j||j|j�S r5 ) r� r� r' � unary_streamr) rh r* r! r r+ r/ r/ r0 �_handle_unary_stream� s ���r/ c sF t || j|j�� t|j|�}|�|jjt| ||j� fdd�|j|j �S )Nc s � S r5 r/ r/ ��request_iteratorr/ r0 �<lambda>� rB z&_handle_stream_unary.<locals>.<lambda>) r� r� r� r' � stream_unaryr) rh r* r r �r� r= rg r# r, r/ r0 r0 �_handle_stream_unary� s �� �r5 c sF t || j|j�� t|j|�}|�|jjt| ||j� fdd�|j|j �S )Nc s � S r5 r/ r/ r0 r/ r0 r2 � rB z'_handle_stream_stream.<locals>.<lambda>) r� r� r� r' � stream_streamr) rh r* r! r r4 r/ r0 r0 �_handle_stream_stream� s �� �r7 �'Optional[_interceptor._ServicePipeline]rS )r� r= �method_with_handler�interceptor_pipeliner+ c sf ddd�� fdd� }� � � }|s.t�| jj�}t|| j�}|d urT|j�|j ||�S |j�||�S d S )NrE rS rT c s � � | �S r5 )rW )rU �r9 r/ r0 �query_handlers� s z,_find_method_handler.<locals>.query_handlers) rR r r� r� rI rE rJ rh r* �execute)r� r= r9 r: r< � method_namerU r/ r; r0 �_find_method_handler� s ��r? )r� � rpc_stater� rC c s< t � d �t�t�t�d ||t�f}| j�|� fdd�� d S )Nc s � dfS r� r/ )� ignored_event�r@ r/ r0 r2 s �z _reject_rpc.<locals>.<lambda>)r� r � ReceiveCloseOnServerOperationr� r� r� r� )r� r@ r� rC r� r/ rB r0 � _reject_rpc� s �� �rD )r� r= rg r, r+ c C s� |j �� | j�t�t�ft|�� |j�t � |j rt|j rVt | |||�W d � S t | |||�W d � S n>|j r�t| |||�W d � S t| |||�W d � S W d � n1 s�0 Y d S r5 )ri r� r� r rC r� r� r~ r� r� �request_streaming�response_streamingr7 r5 r/ r- )r� r= rg r, r/ r/ r0 �_handle_with_method_handler s* � ����rG z4Tuple[Optional[_RPCState], Optional[futures.Future]])r� r9 r: r, �concurrency_exceededr+ c C s� | j s dS | jjs|�� r�t� }zt| |||�}W nR ty� } z:d�|�}t� |� t | |t j j d� |dfW Y d}~S d}~0 0 |du r�t | |t j jd� |dfS |r�t | |t j jd� |dfS |t| |||�fS ndS dS )a� Handles RPC based on provided handlers. When receiving a call event from Core, registered method will have it's name as tag, we pass the tag as registered_method_name to this method, then we can find the handler in registered_method_handlers based on the method name. For call event with unregistered method, the method name will be included in rpc_event.call_details.method and we need to query the generics handlers to find the actual handler. )NNz Exception servicing handler: {}s Error in service handler!Ns Method not found!s Concurrent RPC limit exceeded!)�successr� rI rR r< r? r� r� r� r� rD r r8 r9 r� �resource_exhaustedrG ) r� r9 r: r, rH r@ rg r� rC r/ r/ r0 � _handle_call- sV � � ����rK c @ s e Zd ZdZdZdZdS )� _ServerStage�stopped�started�graceN)rF rG rH �STOPPED�STARTED�GRACEr/ r/ r/ r0 rL o s rL c @ s� e Zd ZU ded< ded< ded< ded< d ed < d ed < d ed< ded< ded< ded< ded< ded< ded< ded < d ed < ddd d d dd �d!d"�Zd#S )$� _ServerStatezthreading.RLock�lockzcygrpc.CompletionQueue�completion_queuez cygrpc.Server�serverrb rc � Dict[str, grpc.RpcMethodHandler]�registered_method_handlersr8 r: r" r, rL �stage�threading.Event�termination_eventzList[threading.Event]�shutdown_events� Optional[int]�maximum_concurrent_rpcs�int�active_rpc_countzSet[_RPCState]� rpc_stateszSet[str]r~ rl �server_deallocated� Sequence[grpc.GenericRpcHandler])rU rV rc r: r, r^ c C st t �� | _|| _|| _t|�| _|| _|| _t j | _ t � � | _ | j g| _|| _d| _i | _t� | _t� | _d| _d S )Nr F)r{ �RLockrT rU rV �listrc r: r, rL rP rY �Eventr[ r\ r^ r` rX r} ra r~ rb )rQ rU rV rc r: r, r^ r/ r/ r0 r_ � s z_ServerState.__init__N)rF rG rH r� r_ r/ r/ r/ r0 rS v s rS � Iterable[grpc.GenericRpcHandler])r= rc r+ c C s6 | j � | j�|� W d � n1 s(0 Y d S r5 )rT rc �extend)r= rc r/ r/ r0 �_add_generic_handlers� s ri rW )r= �method_handlersr+ c C s6 | j � | j�|� W d � n1 s(0 Y d S r5 )rT rX �update)r= rj r/ r/ r0 � _add_registered_method_handlers� s rl r_ )r= �addressr+ c C s6 | j � | j�|�W d � S 1 s(0 Y d S r5 )rT rV �add_http2_port)r= rm r/ r/ r0 �_add_insecure_port� s ro �grpc.ServerCredentials)r= rm �server_credentialsr+ c C s: | j � | j�||j�W d � S 1 s,0 Y d S r5 )rT rV rn � _credentials)r= rm rq r/ r/ r0 �_add_secure_port� s �rs c C s$ | j �| j| jt� | j�t� d S r5 )rV r rU �_REQUEST_CALL_TAGr~ r� r? r/ r/ r0 � _request_call� s �ru )r= rI r+ c C s* |}| j �| j| j||� | j�|� d S r5 )rV �request_registered_callrU r~ r� )r= rI �registered_call_tagr/ r/ r0 �_request_registered_call� s �rx c C s> | j s6| js6| j�� | jD ] }|�� q tj| _dS dS d S rx ) ra r~ rV �destroyr\ r} rL rP rY )r= �shutdown_eventr/ r/ r0 � _stop_serving� s r{ c C s8 | j � | jd8 _W d � n1 s*0 Y d S )N� )rT r` r? r/ r/ r0 �_on_call_completed� s r} )r= �eventr+ c s d}|j tu rP� j�( � j�t� t� �r.d}W d � n1 sB0 Y �n�|j tu sl|j � j�� v �r�d }|j � j�� v r�|j }t |� j� |d ��}n t � j �}� j�� � j�|j � � j d uoЈ j� j k}t||� j� j|�\}}|d ur�� j�|� |d u�r(� jd7 _|�� fdd�� � jtju �r\|� j�� v �rRt� |� nt� � nt� ��rjd}W d � n1 �s�0 Y n�|� |�\}}|D ]0} z | � W n t�y� t� d� Y n0 �q�|d u�r � j�* � j�|� t� ��r�d}W d � n1 �s0 Y |S )NTFr| c s t � �S r5 )r} )� unused_futurer? r/ r0 r2 rB z-_process_event_and_continue.<locals>.<lambda>zException calling callback!) �tag� _SHUTDOWN_TAGrT r~ r� r{ rt rX �keysrZ r7 ra rc r^ r` rK r: r, ra r� �add_done_callbackrY rL rQ rx ru r� r� r� ) r= r~ �should_continue�registered_method_namer9 rH r@ � rpc_futurerv r� r/ r? r0 �_process_event_and_continue� s| &����� �� ��� & $r� c C sL t � � t }| j�|�}| jr&t| � |jtjj krBt | |�sBd S d }q d S r5 ) r� �"_DEALLOCATED_SERVER_CHECK_PERIOD_SrU �pollrb �_begin_shutdown_once�completion_typer �CompletionType� queue_timeoutr� )r= �timeoutr~ r/ r/ r0 �_serve/ s r� c C sZ | j �@ | jtju r8| j�| jt� tj| _| j � t� W d � n1 sL0 Y d S r5 ) rT rY rL rQ rV r rU r� rR r~ r� r? r/ r/ r0 r� > s r� �Optional[float]rZ )r= rO r+ c s� �j �� �jtju r6t�� ���� �W d � S t�� t�� ��j� �� � d u rf�j � � n6� ��fdd�}tj |d�}|� � �W d � S W d � n1 s�0 Y ��� �S )Nc s@ �j � d� �j� �j�� W d � n1 s20 Y d S �N)r� )r� rT rV �cancel_all_callsr/ �rO rz r= r/ r0 � cancel_all_calls_after_graceT s z+_stop.<locals>.cancel_all_calls_after_grace)�target)rT rY rL rP r{ rf r} r� r\ r� rV r� �Thread�startr� )r= rO r� �threadr/ r� r0 �_stopF s 0r� c C s� | j �v | jtjur td��| j�� tj| _| j� � D ]}t | |� q8t | � t j t| fd�}d|_|�� W d � n1 s�0 Y d S )Nz$Cannot start already-started server!)r� �argsT)rT rY rL rP r� rV r� rQ rX r� rx ru r{ r� r� �daemon)r= rI r� r/ r/ r0 �_start` s r� ��generic_rpc_handlersr+ c C s0 | D ]&}t |dd �}|d u rtd�|���qd S )Nre zT"{}" must conform to grpc.GenericRpcHandler type but does not have "service" method!)�getattr�AttributeErrorr� )r� �generic_rpc_handler�service_attributer/ r/ r0 � _validate_generic_rpc_handlersp s ��r� � Sequence[ChannelArgumentType]rn )� base_optionsr� �xdsr+ c C s$ t �|�}t�|�}t| �| | S r5 )r �create_channel_optionr �(create_server_call_tracer_factory_optionr� )r� r� r� �compression_option�'maybe_server_call_tracer_factory_optionr/ r/ r0 �_augment_options| s ����r� c @ s� e Zd ZU ded< ddddddd d �d d �Zd dd�dd�Zdddd�dd�Zddd�dd�Zdddd �d d �Zdd �d d!�Z d-d#d d$�d%d&�Z d#d'd(�d)d*�Z d+d,� Z d"S ).�_ServerrS r� r" rc � Sequence[grpc.ServerInterceptor]r� r] rn rl )r, rc � interceptors�optionsr^ r� r� c C sJ t �� }t �t|||�|�} | �|� t|| |t�|�||�| _| | _ d S r5 ) r �CompletionQueue�Serverr� �register_completion_queuerS r �service_pipeliner� � _cy_server) rQ r, rc r� r� r^ r� r� rU rV r/ r/ r0 r_ � s �z_Server.__init__rg r� r� c C s t |� t| j|� d S r5 )r� ri r� )rQ r� r/ r/ r0 �add_generic_rpc_handlers� s z _Server.add_generic_rpc_handlersr[ rW )� service_namerj r+ c s� | j j�. | j jtju r(W d � d S W d � n1 s<0 Y � fdd�|�� D �}|�� D ]}| j�|� qdt | j |� d S )Nc s i | ]\}}t �� |�|�qS r/ )r �fully_qualified_method)r� rI rg �r� r/ r0 r� � s �z:_Server.add_registered_method_handlers.<locals>.<dictcomp>) r� rT rY rL rQ r� r� r� �register_methodrl )rQ r� rj �method_to_handlersr� r/ r� r0 � add_registered_method_handlers� s . � z&_Server.add_registered_method_handlersr_ )rm r+ c C s t �|t| jt �|���S r5 )r � validate_port_binding_resultro r� r� )rQ rm r/ r/ r0 �add_insecure_port� s �z_Server.add_insecure_portrp )rm rq r+ c C s t �|t| jt �|�|��S r5 )r r� rs r� r� )rQ rm rq r/ r/ r0 �add_secure_port� s ��z_Server.add_secure_portrM c C s t | j� d S r5 )r� r� rP r/ r/ r0 r� � s z _Server.startNr� )r� r+ c C s t j| jjj| jjj|d�S r� )r r� r� r[ �is_set)rQ r� r/ r/ r0 �wait_for_termination� s �z _Server.wait_for_terminationrZ )rO r+ c C s t | j|�S r5 )r� r� )rQ rO r/ r/ r0 �stop� s z _Server.stopc C s t | d�rd| j_d S )Nr� T)r r� rb rP r/ r/ r0 �__del__� s z_Server.__del__)N) rF rG rH r� r_ r� r� r� r� r� r� r� r� r/ r/ r/ r0 r� � s r� rc r� r] )r, r� r� r� r^ r� r� r+ c C s t |� t| ||||||�S r5 )r� r� )r, r� r� r� r^ r� r� r/ r/ r0 � create_server� s �r� )N)��__doc__� __future__r rX � collections� concurrentr ry �enum�loggingr{ r� r �typingr r r r r r r r r r r r r� r r r r � grpc._cythonr � grpc._typingr r r r r r r r r � getLoggerrF r� r� rt r� r� r� r r r� r� r r� r� r� r� � _INF_TIMEOUTr1 r; r@ rA rD � namedtuple�HandlerCallDetailsrE �ABCrK rZ ra �objectr< r� r� r� r� r� r� r� r� r� r� �ServicerContextr� r� r� r r r r r r r r r! r� r r' r- r/ r5 r7 r? rD rG rK �unique�EnumrL rS ri rl ro rs ru rx r{ r} r� r� r� r� r� r� r� r� r� r� r/ r/ r/ r0 �<module> s� 8 �� �� # " ! z?/ � : $ 1 B0 G Z
cải xoăn