File: //usr/lib/python3/dist-packages/twisted/internet/__pycache__/asyncioreactor.cpython-38.pyc
U
��W[9) � @ s� d Z ddlmZmZ ddlZddlmZ ddlmZ ddl m
Z
ddl m Z m
Z
mZ ddlmZ dd lmZ zdd
lmZ W n ek
r� ed ��Y nX dd lmZmZ G d
d� de�Zee�G dd� de ��Zddd�Z dS )z'
asyncio-based reactor implementation.
� )�absolute_import�divisionN)� implementer)�Logger)� DelayedCall)�PosixReactorBase� _NO_FILEDESC�_ContinuousPolling)�callWithLogger)�
IReactorFDSet)�get_event_loopzRequires asyncio.)�PermissionError�BrokenPipeErrorc @ s e Zd ZdZdd� Zdd� ZdS )� _DCHandlez�
Wraps ephemeral L{asyncio.Handle} instances. Callbacks can close
over this and use it as a mutable reference to asyncio C{Handles}.
@ivar handle: The current L{asyncio.Handle}
c C s
|| _ d S �N)�handle)�selfr � r �A/usr/lib/python3/dist-packages/twisted/internet/asyncioreactor.py�__init__' s z_DCHandle.__init__c C s | j �� dS )z5
Cancel the inner L{asyncio.Handle}.
N)r �cancel�r r r r r + s z_DCHandle.cancelN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c s� e Zd ZdZdZe� Zd)� fdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd*d d �Z� fd d �Z� fd!d"�Zd#d$� Zd%d&� Zd'd(� Z� ZS )+�AsyncioSelectorReactorzA
Reactor running on top of L{asyncio.SelectorEventLoop}.
FNc s@ |d krt � }|| _i | _i | _t� | _t| �| _t� � � d S r )
r �_asyncioEventloop�_writers�_readers�set�
_delayedCallsr �_continuousPolling�superr )r � eventloop�� __class__r r r ; s
z AsyncioSelectorReactor.__init__c C s$ z| j j�|� W n Y nX dS )am
Compensate for a bug in asyncio where it will not unregister a FD that
it cannot handle in the epoll loop. It touches internal asyncio code.
A description of the bug by markrwilliams:
The C{add_writer} method of asyncio event loops isn't atomic because
all the Selector classes in the selector module internally record a
file object before passing it to the platform's selector
implementation. If the platform's selector decides the file object
isn't acceptable, the resulting exception doesn't cause the Selector to
un-track the file object.
The failing/hanging stdio test goes through the following sequence of
events (roughly):
* The first C{connection.write(intToByte(value))} call hits the asyncio
reactor's C{addWriter} method.
* C{addWriter} calls the asyncio loop's C{add_writer} method, which
happens to live on C{_BaseSelectorEventLoop}.
* The asyncio loop's C{add_writer} method checks if the file object has
been registered before via the selector's C{get_key} method.
* It hasn't, so the KeyError block runs and calls the selector's
register method
* Code examples that follow use EpollSelector, but the code flow holds
true for any other selector implementation. The selector's register
method first calls through to the next register method in the MRO
* That next method is always C{_BaseSelectorImpl.register} which
creates a C{SelectorKey} instance for the file object, stores it under
the file object's file descriptor, and then returns it.
* Control returns to the concrete selector implementation, which asks
the operating system to track the file descriptor using the right API.
* The operating system refuses! An exception is raised that, in this
case, the asyncio reactor handles by creating a C{_ContinuousPolling}
object to watch the file descriptor.
* The second C{connection.write(intToByte(value))} call hits the
asyncio reactor's C{addWriter} method, which hits the C{add_writer}
method. But the loop's selector's get_key method now returns a
C{SelectorKey}! Now the asyncio reactor's C{addWriter} method thinks
the asyncio loop will watch the file descriptor, even though it won't.
N)r Z _selectorZ
unregister)r �fdr r r �_unregisterFDInAsyncioH s 2z-AsyncioSelectorReactor._unregisterFDInAsyncioc
C s� |r
|j n|j}|�� dkr.| �|t|� d S z
|� }W n2 tk
rj } z|}| j�d � W 5 d }~X Y nX |r~| �|||� d S �N���)ZdoReadZdoWrite�filenoZ_disconnectSelectabler � Exception�_logZfailure)r Z
selectable�read�methodZwhy�er r r � _readOrWrite� s
z#AsyncioSelectorReactor._readOrWritec
C s� || j �� ks|| jj kr d S |�� }z&| j�|t|| j|d� || j |< W nH tk
r� } z*| � |� |j
t
j kr�| j� |� n� W 5 d }~X Y nX d S �NT)
r �keysr" r+ r Z
add_readerr
r1 �IOErrorr( �errnoZEPERM� addReader)r �readerr' r0 r r r r6 � s"
� �
z AsyncioSelectorReactor.addReaderc C s� || j �� ks|| jj kr d S |�� }z&| j�|t|| j|d� || j |< W n\ tk
rv | � |� | j�
|� Y n4 t k
r� | � |� Y n | � |� � Y nX d S )NF) r r3 r" r+ r Z
add_writerr
r1 r
r( � addWriterr �r �writerr' r r r r8 � s(
� �
z AsyncioSelectorReactor.addWriterc C st || j �� ks | j�|�s d S | j�|�r:| j�|� d S |�� }|dkrX| j �|�}n | j �|� | j�|� d S r) ) r r3 r" Z isReading� removeReaderr+ �popr Z
remove_reader)r r7 r' r r r r; � s
� z#AsyncioSelectorReactor.removeReaderc C st || j �� ks | j�|�s d S | j�|�r:| j�|� d S |�� }|dkrX| j �|�}n | j �|� | j�|� d S r) ) r r3 r" Z isWriting� removeWriterr+ r< r Z
remove_writerr9 r r r r= � s
� z#AsyncioSelectorReactor.removeWriterc C s" | � | j�� | j�� �| j�� S r )Z
_removeAllr r3 r r" � removeAllr r r r r> � s �z AsyncioSelectorReactor.removeAllc C s t | j�� �| j�� S r )�listr r3 r" �
getReadersr r r r r@ � s �z!AsyncioSelectorReactor.getReadersc C s t | j�� �| j�� S r )r? r r3 r" �
getWritersr r r r rA � s �z!AsyncioSelectorReactor.getWritersc C s
t | j�S r )r? r! r r r r �getDelayedCalls� s z&AsyncioSelectorReactor.getDelayedCallsc C s$ | j �|d | j j� | j �� d S )Ng{�G�z�?)r �
call_later�stop� run_forever)r Ztimeoutr r r �iterate s �z AsyncioSelectorReactor.iterateTc C s& | j |d� | j�� | jr"d| _d S )N)�installSignalHandlersF)Z startRunningr rE Z _justStopped)r rG r r r �run s
zAsyncioSelectorReactor.runc s t � �� | �d| jd� d S )Nr Zshutdown)r# rD � callLaterZfireSystemEventr r% r r rD s
zAsyncioSelectorReactor.stopc s t � �� | j�� d S r )r# �crashr rD r r% r r rJ s
z AsyncioSelectorReactor.crashc C s
| j �� S r )r �timer r r r �seconds s z AsyncioSelectorReactor.secondsc sx � ����fdd���j �|��}t|����fdd�}���fdd�}t��� | �di ||�jd���j��� �S ) Nc s d�_ �j��� �� �� d S r2 )Zcalledr! �remover )�args�dc�f�kwargsr r r rH s z-AsyncioSelectorReactor.callLater.<locals>.runc s �j �| � � �� d S r )r! rM r �rO )�dchandler r r r & s z0AsyncioSelectorReactor.callLater.<locals>.cancelc s �j �| j��� _d S r )r Zcall_atrK r rR )rS rH r r r �reset* s z/AsyncioSelectorReactor.callLater.<locals>.resetr )rL )r rC r r rL r! �add)r rL rP rN rQ r r rT r )rN rO rS rP rQ rH r r rI s � z AsyncioSelectorReactor.callLaterc s" � ���fdd�}�j �|� d S )Nc s �j d�f� ���S )Nr )rI r �rN rP rQ r r r �<lambda>4 � z7AsyncioSelectorReactor.callFromThread.<locals>.<lambda>)r Zcall_soon_threadsafe)r rP rN rQ �gr rV r �callFromThread3 s z%AsyncioSelectorReactor.callFromThread)N)T)r r r r Z _asyncClosedr r- r r( r1 r6 r8 r; r= r> r@ rA rB rF rH rD rJ rL rI rZ �
__classcell__r r r% r r 3 s*
8
r c C s t | �}ddlm} ||� dS )z�
Install an asyncio-based reactor.
@param eventloop: The asyncio eventloop to wrap. If default, the global one
is selected.
r )�installReactorN)r Ztwisted.internet.mainr\ )r$ Zreactorr\ r r r �install9 s r] )N) r Z
__future__r r r5 Zzope.interfacer Ztwisted.loggerr Ztwisted.internet.baser Ztwisted.internet.posixbaser r r Ztwisted.python.logr
Ztwisted.internet.interfacesr Zasyncior � ImportError�builtinsr
r �objectr r r] r r r r �<module> s&