File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/__pycache__/_future.cpython-310.pyc
o
;j_ � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlZdd lmZmZmZ G d
d � d e�ZG d d
� d
�ZG dd� deej�ZG dd� d�Z eddd�Z G dd� deej e �Z dS )z%Future-returning APIs for coroutines.� )� annotationsN)�Future)�deque)�partial)�chain)�Any� Awaitable�Callable�
NamedTuple�TypeVar�cast)�EVENTS�POLLIN�POLLOUTc @ s> e Zd ZU ded< ded< ded< ded< d ed
< d ed < d S )
� _FutureEventr �future�str�kind�tuple�args�dict�kwargsr �msg�timerN)�__name__�
__module__� __qualname__�__annotations__� r r �?/opt/saltstack/salt/lib/python3.10/site-packages/zmq/_future.pyr s
r c @ sF e Zd ZU dZdZded< ded< ddd �Zdd
d �Zddd
d�ZdS )�_Asyncz Mixin for common async logicNr �
_current_loopz type[Future]�_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_loop1 s
z_Async._get_loopc C s t d��)Nz!Must be implemented in a subclass��NotImplementedError)r& r r r r$ B s z_Async._default_loop�Nonec C � d S �Nr �r& �loopr r r r% E s z_Async._init_io_state)r# r r, )r# r* ) r r r �__doc__r! r r' r$ r% r r r r r + s
r c sZ e Zd ZU dZded< ded< ded< ded< ddd�Zddd�Zd d � fdd�
Z� ZS ) � _AsyncPollerz:Poller that returns a Future on poll, instead of blocking.ztype[_AsyncSocket]�
_socket_class�int�_READ�_WRITEz list[Any]� raw_socketsr. r �socket�evt�fr r# r* c C � t � �)z"Schedule callback for a raw socketr( )r& r. r6 r7 r8 r r r �_watch_raw_socketQ � z _AsyncPoller._watch_raw_socket�socketsc G r9 )z$Unschedule callback for a raw socketr( )r& r. r< r r r �_unwatch_raw_socketsU r; z!_AsyncPoller._unwatch_raw_sockets���� Awaitable[list[tuple[Any, int]]]c
s� �� � � |dkr0zt� �d�}W n ty( } z � �|� W Y d}~� S d}~ww � �|� � S ��� ��� � �g ��fdd�}�����fdd�� g ��fdd�}�jD ]W\}}t |t
j �r�t |�j �sq�j �
|�}��|� |t
j@ r}|jd �d
� |t
j@ r�|jd �d
� qV��|� d}|t
j@ r�|�jO }|t
j@ r�|�jO }���|||� qV�� ��fd d �} ��| � �r�|� |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)r ��watcherr r �wake_rawm � �z#_AsyncPoller.poll.<locals>.wake_rawc s �j � g��R � S r, )r= �r8 )r. r5 r& r r �<lambda>r s z#_AsyncPoller.poll.<locals>.<lambda>c s � D ]}|� � qd S r, )�_clear_io_state)r8 �s)�wrapped_socketsr r �_clear_wrapper_iow s
�z,_AsyncPoller.poll.<locals>._clear_wrapper_io�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 ) rB � cancelled�cancel� RuntimeError� exception�
set_exception�superr0 rN � ExceptionrC )r8 �result�e)� __class__r r& rE r r �
on_poll_ready� s&
����z(_AsyncPoller.poll.<locals>.on_poll_readyc r@ r, rA r rD r r �trigger_timeout� rG z*_AsyncPoller.poll.<locals>.trigger_timeout���MbP?c s$ t �d�r ��� d S � ��� d S )NrR )�hasattrrR Zremove_timeoutrH )r. �timeout_handler r �cancel_timeout� s
z)_AsyncPoller.poll.<locals>.cancel_timeoutc s � � � s
� �� d S d S r, )rB rR rH rD r r �cancel_watcher� s �z)_AsyncPoller.poll.<locals>.cancel_watcher)r" rV rN rW rU rC r' �add_done_callbackr< �
isinstance�_zmq�Socketr1 � from_socket�appendr �_add_recv_eventr �_add_send_eventr3 r4 r: �
call_later)
r& �timeoutrX rY rF rM r6 �maskr7 r[ r\ r` ra �rZ )r r. r5 r& r_ rE rL r rN Y sb ��
�
�
z_AsyncPoller.poll)
r. r r6 r r7 r2 r8 r r# r* )r. r r< r r# r* )r> )r# r? ) r r r r/ r r: r= rN �
__classcell__r r rm r r0 I s
r0 c @ s e Zd Zedd� �ZdS )�_NoTimerc C r+ r, r r r r r rR � s z_NoTimer.cancelN)r r r � staticmethodrR r r r r ro � s ro �T� _AsyncSocket)�boundc s| e Zd ZU dZdZdZded< eZdZ d]d^� fd
d �
Z
e d_d`dd��Z d_da� fdd�
Z
ejj
je
_� fdd�Zejjje_ dbdcd"d#�Z dbddd%d&�Zddd'�ded*d+�Z dbdfd.d/�Z dbdgd2d3�Zd4d5� Zdejfdhd6d7�Zd8d9� Zd:d;� Ze d_d<d=��Z dddd>�didGdH�Z djdIdJ�Z dKdL� Z dMdN� Z!dkdOdP�Z"d_dQdR�Z#dSdT� Z$dUdV� Z%dWdX� Z&d_dYdZ�Z'd[d\� Z(� Z)S )lrr Nr � _zmq.Socket� _shadow_sockr> � _from_socket�_zmq.Socket | Noner# r* 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 )rc rd re rV �__init__Z
underlyingru rx �warnings�warnrZ r �DeprecationWarningr �
_recv_futures�
_send_futures�_stateZFD�_fd)r& �contextZ socket_type�io_looprv r rm r r r{ � s"
�z_AsyncSocket.__init__�cls�type[T]r6 r� r rq c C s | ||d�S )z.Create an async socket from an existing Socket)rv r� r )r� r6 r� r r r rf � s z_AsyncSocket.from_socket�linger�
int | Nonec 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 rB rR rS rJ rV �close)r& r� �
event_list�eventrm r r r� � s �
��z_AsyncSocket.closec s" t � �|�}|tkr| �|� |S r, )rV �getr
�_schedule_remaining_events)r& �keyrX rm r r r� s
z_AsyncSocket.getTF�flagsr2 �copy�bool�track�)Awaitable[list[bytes] | list[_zmq.Frame]]c C � | j dt|||d�d�S )zvReceive a complete multipart zmq message.
Returns a Future whose result will be a multipart message.
�recv_multipart�r� r� r� �r �rh r �r& r� r� r� r r r r� s �z_AsyncSocket.recv_multipart� Awaitable[bytes | _zmq.Frame]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 r� s �z_AsyncSocket.recv��nbytesr� r� �Awaitable[int]c C s | j d|ft||d�d�S )z�Receive a single zmq frame into a pre-allocated buffer.
Returns a Future, whose result will be the number of bytes received.
� recv_intor� )r r r� )r& �bufr� r� r r r r� & s �z_AsyncSocket.recv_into� msg_parts�%Awaitable[_zmq.MessageTracker | None]c 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 )ri )r& r� r� r� r� r r r r r� 1 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 ri )r& r� r� r� r� r r r r r� = s
z_AsyncSocket.sendc s> | � � � � ��fdd�}��|� � �fdd�}� �|� � S )zDeserialize with Futuresc
s� � � � r ��� s��� du rt�d� � d�� d�t� dS ��� r*� ���� � dS ��� }z�|�}W n tyL } z � �|� W Y d}~dS d}~ww � � |� dS )z+Chain result through serialization to recvdNzFuture z completed while awaiting z . A message has been dropped!)
rB rQ rT r| r} �RuntimeWarningrU rX rW rC )�_r� ZloadedrY �r8 �load�recvdr r �_chainU s" � ��z)_AsyncSocket._deserialize.<locals>._chainc s$ �� � rdS � �� r��� dS dS )z"Chain cancellation from f to recvdN)rB rQ rR )r� )r8 r� r r �
_chain_cancelp s
�z0_AsyncSocket._deserialize.<locals>._chain_cancel)r" rb )r& r� r� r� r� r r� r � _deserializeQ s
z_AsyncSocket._deserializec s� �j r t�tj����� }|��|� tt|�|����� � � � ��fdd�}��
� r0|�� n�� |� �fdd�}� � |� � 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 rP )
rB rQ rR rS rT rU r rX rC r� )r8 Zevts�r � poll_futurer& r r �
unwrap_result� s
�� z(_AsyncSocket.poll.<locals>.unwrap_resultc s0 � � � sz� �� W dS ty Y dS w dS )z4Cancel underlying poll if request has been cancelledN)rB rR rS rO )r� r r � cancel_poll� s ��z&_AsyncSocket.poll.<locals>.cancel_poll) r� rd ZZMQError�ENOTSUP�
_poller_class�registerr r rN r" rB rb )r& rk r� �pr� r� r r� r rN { s
z_AsyncSocket.pollc s � fdd�}| � ||�S )z'Add a timeout for a send or recv Futurec s � � � rd S � �t�� � d S r, )rB rU rd �Againr rO r r �future_timeout� s z1_AsyncSocket._add_timeout.<locals>.future_timeout)� _call_later)r& r rk r� r rO 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' rj )r& �delay�callbackr r r r� � s z_AsyncSocket._call_laterc C s. |sdS z|� |� W dS ty Y dS w )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)�remove�
ValueError)r r� r� r r r �_remove_finished_future� s �z$_AsyncSocket._remove_finished_future)r r r r r r � tuple | None�dict[str, Any] | Noner �
Future | Noner c
C s: |p| � � }|du r d}|du ri }|�d�rO|�dd�tj@ rOt| j|�}z ||i |��}W n tyG } z |�|� W Y d}~|S d}~ww |� |� |S t
} t td�rf| jj }
|
dkrf| �
||
d �} t||||d| d�} | j�| � | j�t�t@ r�| �� | jr�| | jv r�|�t| j| j| d �� | �t� |S )
z4Add a recv event, returning the corresponding FutureNr r� r� r ZRCVTIMEOr] �r r r r �r� r� )r" �
startswithr� rd �DONTWAIT�getattrru rW rU rC ro r^ Zrcvtimeor� r r rg r
r � _handle_recvrb r r� �
_add_io_state) r& r r r r r8 r� �rrY r �
timeout_ms�
_future_eventr r r rh � sJ ��
� ��
z _AsyncSocket._add_recv_eventc
C sh |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 �} t||d ||| d
�}| j�|� |�t| j| j|d �� | �t� |S ) z4Add a send event, returning the corresponding Future)r� r� r� r TFN�SNDTIMEOr] r r� r� )r" r� r� r� rd r� r� ru r� rU rW rC r r� ro r^ r� r� r rg rb r r� r� r )r& r r r r r8 r� Z
nowait_kwargsr� Z finish_earlyr� rY r r� r� r r r ri
sR
���
� ��
z _AsyncSocket._add_send_eventc
C s, | j �t�t@ s
dS d}| jr%| j�� \}}}}}}|�� r!d}nn| js| js-| �t� |du r3dS |�� |dkrB|� d� dS |dkrK| j j
}n|dkrT| j j }n|dkr]| j j }nt
d|����|d tjO < z ||i |��}W n ty� } z |�| � W Y d} ~ dS d} ~ ww |� |� dS )zHandle recv eventsNrN r� r� r� zUnhandled recv event type: r� )ru r� r
r r �popleftrB �_drop_io_staterR rC r� r� r� r� rd r� rW rU )
r& r8 r r r r� r r� rX rY r r r r� J s@ �
��z_AsyncSocket._handle_recvc
C s | j �t�t@ s
d S d }| jr%| j�� \}}}}}}|�� r!d }nn| js| js-| �t� |d u r3d S |�� |dkrB|� d � d S |dkrK| j j
}n|dkrT| j j }nt d|����|d t
jO < z
||fi |��}W n ty� } z |�| � W Y d } ~ d S d } ~ ww |� |� d S )NrN r� r� zUnhandled send event type: r� )ru r� r
r r� r� rB r� rR rC r� r� r� rd r� rW rU )
r& r8 r r r r r r� rX rY r r r � _handle_sendu s< �
��z_AsyncSocket._handle_sendc C sH | j jrdS | j �t�}|tj@ r| �� |tj@ r | �� | � � dS )z(Dispatch IO events to _handle_recv, etc.N)
ru r� r� r
rd r r� r r� r� )r& �fd�eventsZ
zmq_eventsr r r �_handle_events� 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� ru 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_handlerru r� r3 r� r- 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)ru r� r� r! Zremove_handler)r& r� r r r rJ � s
�z _AsyncSocket._clear_io_state)Nr> NN)rv rw r# r* r, )r� r� r6 rt r� r r# rq )r� r� r# r* )r TF)r� r2 r� r� r� r� r# r� )r� r2 r� r� r� r� r# r� )r� r2 r� r2 r# r� )r� r r� r2 r� r� r# r� ) r� r r� r2 r� r� r� r� r r r# r� )r# r� )
r r r r� r r� r r� r# r )NNN)r r )*r r r r r� r� r r0 r� r� r{ � classmethodrf r� rd re r/ r� r� r� r� r� r� r� r rN r� r� rp r� rh ri r� r� r� r� r� r� r� r% rJ rn r r rm r rr � sd
� � �� � �*2
�
7=+
)
) r/ �
__future__r r| Zasyncior � collectionsr � functoolsr � itertoolsr �typingr r r r
r r Zzmqrd r
r r r r ZPollerr0 ro rq re rr r r r r �<module> s w