File: //usr/lib64/python3.6/site-packages/zmq/green/__pycache__/core.cpython-36.pyc
3
VS�_O) � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZ ddlm Z
ddl m Z ddl
Z
ddlmZ dd lmZ eed
�r�ejejfZnf Zd d � ZG d
d� de
�ZG dd� de�ZdS )zjThis module wraps the :class:`Socket` and :class:`Context` found in :mod:`pyzmq <zmq>` to be non blocking
� )�print_functionN)�Context)�Socket� )�_Poller)� AsyncResult)�get_hub�RCVTIMEOc C s< y | j � W n* tk
r6 } z| j� W Y dd}~X nX dS )zNsimple wrapper for stopping an Event, allowing for method rename in gevent 1.0N)�stop�AttributeError�cancel)Zevt�e� r � /usr/lib64/python3.6/core.py�_stop# s r c s� e Zd ZdZdZdZdZdZdZdZ dZ
e Z � fdd�Z
dd� Zd#� fd d
� Zd d � Zd
d� Zd$� fdd� Zdd� Zdd� Zd%� fdd� Zd&� fdd� Z� fdd �Z� fd d �Z� fd d �Z� fd!d"�Z� ZS )'�_SocketaO Green version of :class:`zmq.Socket`
The following methods are overridden:
* send
* recv
To ensure that the ``zmq.NOBLOCK`` flag is set and that sending or receiving
is deferred to the hub if a ``zmq.EAGAIN`` (retry) error is raised.
The `__state_changed` method is triggered when the zmq.FD for the socket is
marked as readable and triggers the necessary read and write events (which
are waited for in the recv and send methods).
Some double underscore prefixes are used to minimize pollution of
:class:`zmq.Socket`'s namespace.
FNg333333'@c s* t t| �j||� d| _d| _| j� d S )NF)�superr �__init__�_Socket__in_send_multipart�_Socket__in_recv_multipart�_Socket__setup_events)�self�a�kw)� __class__r r r F s z_Socket.__init__c C s | j � d S )N)�close)r r r r �__del__M s z_Socket.__del__c s t t| �j|� | j� d S )N)r r r �_Socket__cleanup_events)r Zlinger)r r r r P s z
_Socket.closec C s4 t | dd �r t| j� d | _| jj� | jj� d S )N� _state_event)�getattrr r �_Socket__writable�set�_Socket__readable)r r r r Z__cleanup_eventsT s
z_Socket.__cleanup_eventsc C s� t � | _t � | _| jj� | jj� y,t� jj| jtj �d�| _
| j
j | j � W n: t
k
r� ddlm} || jtj �| j dd�| _
Y nX d S )Nr r )�
read_eventT)Zpersist)r r" r r! r Zloop�io�
getsockopt�zmq�FDr �start�_Socket__state_changedr Z gevent.corer# )r r# r r r Z__setup_events] s
z_Socket.__setup_eventsc s� | j r| j� d S ytt| �jtj�}W n< tjk
rd } z | jj |� | j
j |� W Y d d }~X n*X |tj @ rz| jj � |tj
@ r�| j
j � d S )N)�closedr r r r% r& �EVENTS�ZMQErrorr Z
set_exceptionr" �POLLOUTr! �POLLIN)r ZeventZ_evtypeZevents�exc)r r r Z__state_changedk s
z_Socket.__state_changedc C s� | j j� std��t� | _ tj� }| j}|r:tj|d�}nd }z�y |rN|j� | j j dd� W nr tjk
r� } zT||k r|� tj� }| j
r�|r�|| |kr�| j t j
�t j@ r�td| j tjd� W Y d d }~X nX W d |r�|j� | j j� X d S )Nz.Only one greenlet can be waiting on this event)�secondsT)�blockz6BUG: gevent may have missed a libzmq send event on %i!)�file)r �ready�AssertionErrorr �time�_gevent_bug_timeout�gevent�Timeoutr( �get�
_debug_geventr% r& r+ r- �printr' �sys�stderrr r! )r �tic�dt�timeout�t�tocr r r � _wait_write{ s* *z_Socket._wait_writec C s� | j j� std��t� | _ tj� }| j}|r:tj|d�}nd }z�y |rN|j� | j j dd� W nr tjk
r� } zT||k r|� tj� }| j
r�|r�|| |kr�| j t j
�t j@ r�td| j tjd� W Y d d }~X nX W d |r�|j� | j j� X d S )Nz.Only one greenlet can be waiting on this event)r0 T)r1 z6BUG: gevent may have missed a libzmq recv event on %i!)r2 )r" r3 r4 r r5 r6 r7 r8 r( r9 r: r% r& r+ r. r; r' r< r= r r! )r r> r? r@ rA rB r r r �
_wait_read� s* *z_Socket._wait_readr Tc s� |t j@ r>z tt| �j||||f|�}W d| js8| j� X |S |t jO }xzytt| �j||||�}W n@ t jk
r� } z"|jt j kr�| js�| j� � W Y dd}~X nX | js�| j� |S | j
� qJW dS )z�send, which will only block current greenlet
state_changed always fires exactly once (success or fail) at the
end of this method.
N) r& �NOBLOCKr r �sendr r) r, �errno�EAGAINrC )r �data�flags�copy�track�kwargs�msgr
)r r r rF � s&
z _Socket.sendc s� |t j@ r8ztt| �j|||�}W d| js2| j� X |S |t jO }xxytt| �j|||�}W n@ t jk
r� } z"|jt j kr�| js�| j� � W Y dd}~X nX | js�| j� |S | j
� qDW dS )z�recv, which will only block current greenlet
state_changed always fires exactly once (success or fail) at the
end of this method.
N) r& rE r r �recvr r) r, rG rH rD )r rJ rK rL rN r
)r r r rO � s&
z _Socket.recvc s2 d| _ ztt| �j||�}W dd| _ | j� X |S )zAwrap send_multipart to prevent state_changed on each partial sendTNF)r r r �send_multipartr) )r �argsrM rN )r r r rP � s
z_Socket.send_multipartc s2 d| _ ztt| �j||�}W dd| _ | j� X |S )zAwrap recv_multipart to prevent state_changed on each partial recvTNF)r r r �recv_multipartr) )r rQ rM rN )r r r rR s
z_Socket.recv_multipartc s: |t krtjdt� tt| �j|�}|tjkr6| j � |S )z+trigger state_changed on getsockopt(EVENTS)z0TIMEO socket options have no effect in zmq.green)
�TIMEOS�warnings�warn� UserWarningr r r9 r& r+ r) )r �optZoptval)r r r r9
s
z _Socket.getc s& |t krtjdt� tt| �j||�S )zset socket optionz0TIMEO socket options have no effect in zmq.green)rS rT rU rV r r r! )r rW �val)r r r r! s z _Socket.set)N)NN)r TF)r TF) �__name__�
__module__� __qualname__�__doc__r r r r" r r6 r: r Z
_poller_classr r r r r r) rC rD rF rO rP rR r9 r! �
__classcell__r r )r r r + s. "
r c @ s e Zd ZdZeZdZdS )�_ContextzrReplacement for :class:`zmq.Context`
Ensures that the greened Socket above is used in calls to `socket`.
N)rY rZ r[ r\ r Z
_socket_classZ _instancer r r r r^ s r^ )r\ Z
__future__r r< r5 rT r& r Z_original_Contextr Z_original_SocketZpollr r7 Z gevent.eventr Z
gevent.hubr �hasattrr ZSNDTIMEOrS r r r^ r r r r �<module> s$
p