File: //usr/lib/python3/dist-packages/twisted/internet/__pycache__/epollreactor.cpython-38.pyc
U
W[-! � @ s� d Z ddlmZmZ ddlmZmZmZmZm Z ddl
Z
ddl m Z ddl
mZ ddlmZ ddlmZ e e�G d d
� d
ejej��Zd d � Zd
d gZdS )
z�
An epoll() based implementation of the twisted main loop.
To install the event loop (and you should do this before any connections,
listeners or connectors are added)::
from twisted.internet import epollreactor
epollreactor.install()
� )�division�absolute_import)�epoll�EPOLLHUP�EPOLLERR�EPOLLIN�EPOLLOUTN)� implementer)�
IReactorFDSet)�log)� posixbasec @ s| e Zd ZdZeeB ZeZe Z
dd� Z dd� Z dd� Z
dd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZeZdS )� EPollReactora�
A reactor that uses epoll(7).
@ivar _poller: A C{epoll} which will be used to check for I/O
readiness.
@ivar _selectables: A dictionary mapping integer file descriptors to
instances of C{FileDescriptor} which have been registered with the
reactor. All C{FileDescriptors} which are currently receiving read or
write readiness notifications will be present as values in this
dictionary.
@ivar _reads: A set containing integer file descriptors. Values in this
set will be registered with C{_poller} for read readiness notifications
which will be dispatched to the corresponding C{FileDescriptor}
instances in C{_selectables}.
@ivar _writes: A set containing integer file descriptors. Values in this
set will be registered with C{_poller} for write readiness
notifications which will be dispatched to the corresponding
C{FileDescriptor} instances in C{_selectables}.
@ivar _continuousPolling: A L{_ContinuousPolling} instance, used to handle
file descriptors (e.g. filesystem files) that are not supported by
C{epoll(7)}.
c C s< t d�| _t� | _t� | _i | _t�| �| _tj �
| � dS )zm
Initialize epoll object, file descriptor tracking dictionaries, and the
base class.
i N) r �_poller�set�_reads�_writes� _selectablesr Z_ContinuousPolling�_continuousPolling�PosixReactorBase�__init__��self� r �?/usr/lib/python3/dist-packages/twisted/internet/epollreactor.pyr = s
zEPollReactor.__init__c C sX |� � }||krT|}||kr4||O }| j�||� n| j�||� |�|� |||< dS )z�
Private method for adding a descriptor from the event loop.
It takes care of adding it if new or modifying it if already added
for another state (read -> read/write for example).
N)�filenor �modify�register�add) r �xer�primary�other� selectables�event� antievent�fd�flagsr r r �_addM s
zEPollReactor._addc
C sb z | � || j| j| jtt� W n> tk
r\ } z |jtjkrJ| j �
|� n� W 5 d}~X Y nX dS )zR
Add a FileDescriptor for notification of data available to read.
N) r& r r r r r �IOError�errno�EPERMr � addReader)r �reader�er r r r* h s � zEPollReactor.addReaderc
C sb z | � || j| j| jtt� W n> tk
r\ } z |jtjkrJ| j �
|� n� W 5 d}~X Y nX dS )zS
Add a FileDescriptor for notification of data available to write.
N) r& r r r r r r' r( r) r � addWriter)r �writerr, r r r r- y s � zEPollReactor.addWriterc
C sv |� � }|dkr2|�� D ]\}}||kr q2qdS ||krr||krV|} | j�|| � n||= | j�|� |�|� dS )z�
Private method for removing a descriptor from the event loop.
It does the inverse job of _add, and also add a check in case of the fd
has gone away.
���N)r �itemsr r Z
unregister�remove)
r r r r r! r" r# r$ Zfdesr% r r r �_remove� s zEPollReactor._removec C s: | j �|�r | j �|� dS | �|| j| j| jtt� dS )zQ
Remove a Selectable for notification of data available to read.
N) r Z isReading� removeReaderr2 r r r r r )r r+ r r r r3 � s �zEPollReactor.removeReaderc C s: | j �|�r | j �|� dS | �|| j| j| jtt� dS )zR
Remove a Selectable for notification of data available to write.
N) r Z isWriting� removeWriterr2 r r r r r )r r. r r r r4 � s �zEPollReactor.removeWriterc s6 � � � fdd�� jD �� fdd�� jD ��� j�� S )zD
Remove all selectables, and return a list of them.
c s g | ]}� j | �qS r �r ��.0r$ r r r �
<listcomp>� s z*EPollReactor.removeAll.<locals>.<listcomp>c s g | ]}� j | �qS r r5 r6 r r r r8 � s )Z
_removeAllr r r � removeAllr r r r r9 � s ��zEPollReactor.removeAllc s � fdd�� j D �� j�� S )Nc s g | ]}� j | �qS r r5 r6 r r r r8 � s z+EPollReactor.getReaders.<locals>.<listcomp>)r r �
getReadersr r r r r: � s �zEPollReactor.getReadersc s � fdd�� j D �� j�� S )Nc s g | ]}� j | �qS r r5 r6 r r r r8 � s z+EPollReactor.getWriters.<locals>.<listcomp>)r r �
getWritersr r r r r; � s �zEPollReactor.getWritersc
C s� |dkr d}z| j �|t| j��}W n: tk
r^ } z |jtjkrLW Y �
dS � W 5 d}~X Y nX | j}|D ]>\}}z| j| }W n tk
r� Y qjX t �
|||||� qjdS )z1
Poll the poller for new events.
Nr/ ) r Zpoll�lenr r' r( ZEINTRZ_doReadOrWrite�KeyErrorr ZcallWithLogger)r Ztimeout�l�errZ_drdwr$ r" Z
selectabler r r �doPoll� s
zEPollReactor.doPollN)�__name__�
__module__� __qualname__�__doc__r r Z_POLL_DISCONNECTEDr Z_POLL_INr Z _POLL_OUTr r& r* r- r2 r3 r4 r9 r: r; r@ Z doIterationr r r r r
s
r
c C s t � } ddlm} || � dS )z&
Install the epoll() reactor.
r )�installReactorN)r
Ztwisted.internet.mainrE )�prE r r r �install� s rG )rD Z
__future__r r Zselectr r r r r r( Zzope.interfacer Ztwisted.internet.interfacesr
Ztwisted.pythonr Ztwisted.internetr r Z_PollLikeMixinr
rG �__all__r r r r �<module> s
U