File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/__pycache__/_future.cpython-310.pyc
o
�N�g6X � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZmZ ddlZddlmZmZmZ ddlm Z G d d
� d
e�Z G d d � d �Z G d
d� de ej �Z G dd� d�Z!eddd�Z"G dd� de ej#e �Z$dS )z%Future-returning APIs for coroutines.� N)�Future)�deque)�chain)
�Any� Awaitable�Callable�Dict�List�
NamedTuple�Optional�Tuple�Type�TypeVar�Union�cast�overload)�EVENTS�POLLIN�POLLOUT)�Literalc @ s6 e Zd ZU eed< eed< eed< eed< eed< dS )� _FutureEvent�future�kind�kwargs�msg�timerN)�__name__�
__module__� __qualname__r �__annotations__�strr r � r! r! �?/opt/saltstack/salt/lib/python3.10/site-packages/zmq/_future.pyr s
r c @ sR e Zd ZU dZdZeed< ee ed< defdd�Z defdd �Z
d d
d
d �Z dS )�_Asyncz Mixin for common async logicN�
_current_loop�_Future�returnc C sL | j du r| �� | _ | �| j � | j S | �� }|| j ur$|| _ | �|� |S )z�Get event loop
Notice if event loop has changed,
and register init_io_state on activation of a new event loop
N)r$ �
_default_loop�_init_io_state)�selfZ current_loopr! r! r" � _get_loop6 s
z_Async._get_loopc C s t d��)Nz!Must be implemented in a subclass��NotImplementedError�r) r! r! r" r' G s z_Async._default_loopc C � d S �Nr! �r) �loopr! r! r" r( J s z_Async._init_io_stater/ )r& N) r r r �__doc__r$ r r r
r r* r' r( r! r! r! r" r# 0 s
r# c
s� e Zd ZU dZed ed< eed< eed< ee ed< deded ed
e d d f
d
d�Z
deded d fdd�Z dd e ee
eef f� fdd�
Z� ZS )� _AsyncPollerz:Poller that returns a Future on poll, instead of blocking.� _AsyncSocket�
_socket_class�_READ�_WRITE� raw_socketsr1 �socket�evt�fr& Nc C � t � �)z"Schedule callback for a raw socketr+ )r) r1 r9 r: r; r! r! r" �_watch_raw_socketV � z _AsyncPoller._watch_raw_socket�socketsc G r< )z$Unschedule callback for a raw socketr+ )r) r1 r? r! r! r" �_unwatch_raw_socketsZ r> z!_AsyncPoller._unwatch_raw_sockets���c
s� �� � � |dkr0zt� �d�}W n ty( } z � �|� W Y d}~� S d}~ww � �|� � S ��� ��� � �g ��fdd�}�����fdd�� �jD ]R\}}t |t
j �r}t |�j �sd�j �
|�}|t
j@ rp|jd�d� |t
j@ r||jd�d� qN��|� d}|t
j@ r�|�jO }|t
j@ r�|�jO }���|||� qN�� ��fd d
�}��|� |dur�|dkrчfd d �} ��d
| | ����fdd�}
� �|
� �fdd�} � �| � � S )z Return a Future for a poll eventr Nc � � � � s � �d � d S d S r/ ��done�
set_result)�args��watcherr! r" �wake_rawr � �z#_AsyncPoller.poll.<locals>.wake_rawc s �j � g��R � S r/ )r@ �r; )r1 r8 r) r! r" �<lambda>w s z#_AsyncPoller.poll.<locals>.<lambda>�poll�r c
s� �� � rd S ��� r z��� W d S ty Y d S w ��� r)����� � d S z
tt���d�}W n t yK } z ��|� W Y d }~d S d }~ww ��
|� d S �Nr ) rD � cancelled�cancel� RuntimeError� exception�
set_exception�superr3 rM � ExceptionrE )r; �result�e)� __class__r r) rH r! r" �
on_poll_ready� s&
����z(_AsyncPoller.poll.<locals>.on_poll_readyc rB r/ rC r! rG r! r" �trigger_timeout� rJ z*_AsyncPoller.poll.<locals>.trigger_timeout���MbP?c s$ t �d�r ��� d S � ��� d S )NrQ )�hasattrrQ Zremove_timeoutrK )r1 �timeout_handler! r" �cancel_timeout� s
z)_AsyncPoller.poll.<locals>.cancel_timeoutc s � � � s
� �� d S d S r/ )rD rQ rK rG r! r" �cancel_watcher� s �z)_AsyncPoller.poll.<locals>.cancel_watcher)r% rU rM rV rT rE r* �add_done_callbackr? �
isinstance�_zmq�Socketr5 � from_socketr �_add_recv_eventr �_add_send_event�appendr6 r7 r= �
call_later) r) �timeoutrW rX rI r9 �maskr: rZ r[ r_ r` �rY )r r1 r8 r) r^ rH r" rM ^ sX ��
�
�
z_AsyncPoller.poll)rA )r r r r2 r
r �intr r r r= r@ r r rM �
__classcell__r! r! rl r" r3 N s
,r3 c @ s e Zd Zedd� �ZdS )�_NoTimerc C r. r/ r! r! r! r! r" rQ � s z_NoTimer.cancelN)r r r � staticmethodrQ r! r! r! r" ro � s ro �Tr4 )Zboundc s e Zd ZU dZdZdZded< eZdZ dPde
d ddf� fdd �
Z e dQd
e
e d dd edefd
d��ZdQde
e ddf� fdd�
Zejjje_� fdd�Zejjje_e dRdd�dededeee fdd��Z e dRdd�dede d dedeee fd d��Z e dRdd�dede d dedeeej fd d��Z e dSdedededee ee eej f fd d��Z dSdedededee ee eej f fd d�Z dSdedededee eej f fd d!�Z dSd"edededee
ej! fd#d$�Z" dSd%edededed&edee
ej! f d'd(�Z#d)d*� Z$dej%fdee fd+d,�Z&dee' f� fd-d.� Z(dTd0e'ded1e'ded f� fd2d3�
Z)d4d5� Z*d6d7� Z+e,d8d9� �Z-dUd:d;�Z.dVd<d=�Z/d>d?� Z0d@dA� Z1dWdBdC�Z2dQdDdE�Z3dFdG� Z4dHdI� Z5dJdK� Z6dQdLdM�Z7dNdO� Z8� Z9S )Xr4 Nr z _zmq.Socket� _shadow_sockrA � _from_socketr& c s� t |tj�r d |}}|d urt� j|jd� || _nt� j||fi |�� tj�| j�| _|d ur?tj | j
j � d�t dd� t
� | _t
� | _d| _| jj| _d S )N)�shadowz^(io_loop) argument is deprecated in pyzmq 22.2. The currently active loop will always be used.� )�
stacklevelr )rb rc rd rU �__init__Z
underlyingrr rt �warnings�warnrY r �DeprecationWarningr �
_recv_futures�
_send_futures�_stateZFD�_fd)r) �contextZ socket_type�io_looprs r rl r! r" rw � s"
�z_AsyncSocket.__init__�clsr9 r� c C s | ||d�S )z.Create an async socket from an existing Socket)rs r� r! )r� r9 r� r! r! r" re � s z_AsyncSocket.from_socket�lingerc sz | j s4| jd ur4tt| jpg | jpg ��}|D ]}|j�� s/z|j�� W q t y. Y qw q| �
� t � j |d� d S )N)r� )
�closedr~ �listr r{ r| r rD rQ rR �_clear_io_staterU �close)r) r� �
event_list�eventrl r! r" r� � s �
��z_AsyncSocket.closec s" t � �|�}|tkr| �|� |S r/ )rU �getr �_schedule_remaining_events)r) �keyrW rl r! r" r� s
z_AsyncSocket.getF)�track�flagsr� c C r. r/ r! )r) r� r� r! r! r" �recv_multipart
� z_AsyncSocket.recv_multipart�copyTc C r. r/ r! �r) r� r� r� r! r! r" r� r� c C r. r/ r! r� r! r! r" r� r� c C r. r/ r! r� r! r! r" r� r� c C � | � dt|||d��S )zvReceive a complete multipart zmq message.
Returns a Future whose result will be a multipart message.
r� �r� r� r� �rf �dictr� r! r! r" r� " s �c C r� )z�Receive a single zmq frame.
Returns a Future, whose result will be the received frame.
Recommend using recv_multipart instead.
�recvr� r� r� r! r! r" r� - s z_AsyncSocket.recv� msg_partsc K s( ||d< ||d< ||d< | j d||d�S )zqSend a complete multipart zmq message.
Returns a Future that resolves when sending is complete.
r� r� r� �send_multipart�r r )rg )r) r� r� r� r� r r! r! r" r� 8 s z_AsyncSocket.send_multipart�datar c K s< ||d< ||d< ||d< |� t|||d�� | jd||d�S )z�Send a single zmq frame.
Returns a Future that resolves when sending is complete.
Recommend using send_multipart instead.
r� r� r� r� �sendr� )�updater� rg )r) r� r� r� r� r r! r! r" r� D s
z_AsyncSocket.sendc s> | � � � � ��fdd�}��|� � �fdd�}� �|� � S )zDeserialize with Futuresc
sz � � � rdS ��� r� ���� � dS ��� }z�|�}W n ty5 } z � �|� W Y d}~dS d}~ww � �|� dS )z+Chain result through serialization to recvdN)rD rS rT rW rV rE )�_�bufZloadedrX �r; �load�recvdr! r" �_chain\ s ��z)_AsyncSocket._deserialize.<locals>._chainc s$ �� � rdS � �� r��� dS dS )z"Chain cancellation from f to recvdN)rD rP rQ )r� )r; r� r! r" �
_chain_cancelm s
�z0_AsyncSocket._deserialize.<locals>._chain_cancel)r% ra )r) r� r� r� r� r! r� r" � _deserializeX s
z_AsyncSocket._deserializec sn �j r t�tj����� }|��|� tt|�|��}�� � � � �fdd�}|�
� r0||� � S |� |� � S )zSpoll the socket for events
returns a Future for the poll results.
c st � � � rd S | �� r z� �� W d S ty Y d S w | �� r)� �| �� � d S t| �� �}� �|� �d�� d S rO )
rD rP rQ rR rS rT r� rW rE r� )r; Zevts�r r) r! r" �
unwrap_result� s
�� z(_AsyncSocket.poll.<locals>.unwrap_result) r� rc ZZMQError�ENOTSUP�
_poller_class�registerr r rM r% rD ra )r) rj r� �pr; r� r! r� r" rM x s
�z_AsyncSocket.pollc s t � j|i |��S r/ )rU � recv_string)r) rF r rl r! r" r� � � z_AsyncSocket.recv_string�utf-8�s�encodingc s t � j|||d�S )N)r� r� )rU � send_string)r) r� r� r� rl r! r" r� � r� z_AsyncSocket.send_stringc s � fdd�}| � ||�S )z'Add a timeout for a send or recv Futurec s � � � rd S � �t�� � d S r/ )rD rT rc �Againr! rN r! r" �future_timeout� s z1_AsyncSocket._add_timeout.<locals>.future_timeout)� _call_later)r) r rj r� r! rN r" � _add_timeout� s z_AsyncSocket._add_timeoutc C s | � � �||�S )z�Schedule a function to be called later
Override for different IOLoop implementations
Tornado and asyncio happen to both have ioloop.call_later
with the same signature.
)r* ri )r) �delay�callbackr! r! r" r� � s z_AsyncSocket._call_laterc C s6 t |�D ] \}}|j| u r nqdS |�|| � dS )z�Make sure that futures are removed from the event list when they resolve
Avoids delaying cleanup until the next send/recv event,
which may never come.
N)� enumerater �remove)r r� Zf_idxr� r! r! r" �_remove_finished_future� s
�z$_AsyncSocket._remove_finished_futurec
s |p� � � }|�d�rC|�dd�tj@ rCt� j|�}z |d
i |��}W n ty; } z |�|� W Y d}~|S d}~ww |� |� |S t
}t td�rZ� jj } | dkrZ� �
|| d �}� j�t|||d|d�� |�� fdd �� � j�t�t@ r|� �� � jr�� �t� |S ) z4Add a recv event, returning the corresponding Futurer� r� r NZRCVTIMEOr\ )r r c � � � | � j�S r/ )r� r{ rK r- r! r" rL � � z._AsyncSocket._add_recv_event.<locals>.<lambda>r! )r% �
startswithr� rc �DONTWAIT�getattrrr rV rT rE ro r] Zrcvtimeor� r{ rh r ra r r � _handle_recv�
_add_io_state)
r) r r r r; r� �rrX r �
timeout_msr! r- r" rf � s4 ��
�
z _AsyncSocket._add_recv_eventc
s\ |p� � � }|dv rw� jsw|�dd�}|�� }|tjB |d< t� j|�}d} z
||fi |��}
W n6 tjyP } z|tj@ rD|� | � nd} W Y d} ~ n d} ~ w t
yf } z |� | � W Y d} ~ n
d} ~ ww |� |
� | rw� j ru� �
� |S t} ttd�r�� j�tj�}
|
dkr�� �||
d �} � j�t||||| d �� |�� fd
d �� � �t� |S ) z4Add a send event, returning the corresponding Future)r� r� r� r TFN�SNDTIMEOr\ )r r r c r� r/ )r� r| rK r- r! r" rL " r� z._AsyncSocket._add_send_event.<locals>.<lambda>)r% r| r� r� rc r� r� rr r� rT rV rE r{ r� ro r] r� r� rh r ra r� r )r) r r r r r; r� Z
nowait_kwargsr� Z finish_earlyr� rX r r� r! r- r" rg � sH
���
�
�
z _AsyncSocket._add_send_eventc
C s | j �t�t@ s
dS d}| jr$| j�� \}}}}}|�� r d}nn| js| js,| �t� |du r2dS |�� |dkrA|� d� dS |dkrJ| j j
}n|dkrS| j j }nt d| ��|d t
jO < z |di |��}W n ty� } z |�|� W Y d}~dS d}~ww |� |� dS )zHandle recv eventsNrM r� r� z Unhandled recv event type: %rr� r! )rr r� r r r{ �popleftrD �_drop_io_staterQ rE r� r� �
ValueErrorrc r� rV rT ) r) r; r r r� r r� rW rX r! r! r" r� ( s< �
��z_AsyncSocket._handle_recvc
C s | j �t�t@ s
d S d }| jr$| j�� \}}}}}|�� r d }nn| js| js,| �t� |d u r2d S |�� |dkrA|� d � d S |dkrJ| j j
}n|dkrS| j j }nt d| ��|d t
jO < z
||fi |��}W n ty� } z |�|� W Y d }~d S d }~ww |� |� d S )NrM r� r� z Unhandled send event type: %rr� )rr r� r r r| r� rD r� rQ rE r� r� r� rc r� rV rT ) r) r; r r r r r� rW rX r! r! r" � _handle_sendQ s< �
��z_AsyncSocket._handle_sendc C s< | j �t�}|tj@ r| �� |tj@ r| �� | �� dS )z(Dispatch IO events to _handle_recv, etc.N) rr r� r rc r r� r r� r� )r) �fd�eventsZ
zmq_eventsr! r! r" �_handle_eventsz s
z_AsyncSocket._handle_eventsc C sB | j dkrdS |du r| j�t�}|| j @ r | �d| j� dS dS )zkSchedule a call to handle_events next loop iteration
If there are still events to handle.
r N)r} rr r� r r� r� )r) r� r! r! r" r� � s
�z'_AsyncSocket._schedule_remaining_eventsc C s* | j |kr
| j |B }| _ | �| j � dS )zAdd io_state to poller.N�r} �_update_handler�r) �stater! r! r" r� � s
z_AsyncSocket._add_io_statec C s( | j |@ r | j | @ | _ | �| j � dS )z&Stop poller from watching an io_state.Nr� r� r! r! r" r� � s
z_AsyncSocket._drop_io_statec C s |r| � � | �� dS )zOUpdate IOLoop handler with state.
zmq FD is always read-only.
N)r* r� r� r! r! r" r� � s z _AsyncSocket._update_handlerc C s6 |du r| � � }|�| j| j| j� | �d| j� dS )z#initialize the ioloop event handlerNr )r* Z add_handlerrr r� r6 r� r0 r! r! r" r( � s z_AsyncSocket._init_io_statec C s2 | j }| j jr
| j}| jdur| j�|� dS dS )zNunregister the ioloop event handler
called once during close
N)rr r� r~ r$ Zremove_handler)r) r� r! r! r" r� � s
�z _AsyncSocket._clear_io_state)NrA NNr/ )r )r TF)r r� )NN)NNN)r r ):r r r r{ r| r} r r3 r� r~ r rw � classmethodr
rq r re rm r� rc rd r2 r� r �boolr r �bytesr� r ZFramer r� ZMessageTrackerr� r� r� r rM r r� r� r� r� rp r� rf rg r� r� r� r� r� r� r� r( r� rn r! r! rl r" r4 � s
��� " ����
������
������ ����� �����
� ����
� ����
�������
� '$
$8)
)
)%r2 rx Zasyncior � collectionsr � itertoolsr �typingr r r r r r
r r r
r r r r Zzmqrc r r r Z zmq._typingr r r# ZPollerr3 ro rq rd r4 r! r! r! r" �<module> s < m