U Tg" @sdZddlZddlZddlmZddlmZddlmZddl m Z z6e edZ e edZ e ed Ze ed Ze ed ZWn*ek rZz eeW5dZ[XYnXeeGd d d ejejZddZd dgZdS)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() N) implementer) posixbase) IReactorFDSet)logepollEPOLLHUPEPOLLERREPOLLINEPOLLOUTc@s|eZdZdZeeBZeZe Z ddZ ddZ ddZ dd Zd d Zd d ZddZddZddZddZddZeZdS) 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)}. cCs<td|_t|_t|_i|_t||_tj |dS)zm Initialize epoll object, file descriptor tracking dictionaries, and the base class. iN) r_pollerset_reads_writes _selectablesrZ_ContinuousPolling_continuousPollingPosixReactorBase__init__selfrA/tmp/pip-unpacked-wheel-1_7di8ic/twisted/internet/epollreactor.pyrGs   zEPollReactor.__init__c CsX|}||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 modifyregisteradd) rxerprimaryother selectablesevent antieventfdflagsrrr_addVs zEPollReactor._addc Csbz|||j|j|jttWn>tk r\}z |jtjkrJ|j |nW5d}~XYnXdS)zR Add a FileDescriptor for notification of data available to read. N) r$rrrr r OSErrorerrnoEPERMr addReader)rreadererrrr(ps zEPollReactor.addReaderc Csbz|||j|j|jttWn>tk r\}z |jtjkrJ|j |nW5d}~XYnXdS)zS Add a FileDescriptor for notification of data available to write. N) r$rrrr r r%r&r'r addWriter)rwriterr*rrrr+s zEPollReactor.addWriterc Csv|}|dkr2|D]\}}||krq2qdS||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)ritemsr r unregisterremove) rrrrrr r!r"Zfdesr#rrr_removes zEPollReactor._removecCs:|j|r|j|dS|||j|j|jttdS)zQ Remove a Selectable for notification of data available to read. N) rZ isReading removeReaderr1rrrr r )rr)rrrr2s  zEPollReactor.removeReadercCs:|j|r|j|dS|||j|j|jttdS)zR Remove a Selectable for notification of data available to write. N) rZ isWriting removeWriterr1rrrr r )rr,rrrr3s  zEPollReactor.removeWritercs6fddjDfddjDjS)zD Remove all selectables, and return a list of them. csg|]}j|qSrr.0r"rrr sz*EPollReactor.removeAll..csg|]}j|qSrr4r5rrrr7s)Z _removeAllrrr removeAllrrrrr8szEPollReactor.removeAllcsfddjDjS)Ncsg|]}j|qSrr4r5rrrr7sz+EPollReactor.getReaders..)rr getReadersrrrrr9s  zEPollReactor.getReaderscsfddjDjS)Ncsg|]}j|qSrr4r5rrrr7sz+EPollReactor.getWriters..)rr getWritersrrrrr:s  zEPollReactor.getWritersc Cs|dkr d}z|j|t|j}Wn:tk r^}z|jtjkrLWY dSW5d}~XYnX|j}|D]>\}}z|j|}Wntk rYqjXt |||||qjdS)z1 Poll the poller for new events. Nr-) r polllenrr%r&ZEINTRZ_doReadOrWriteKeyErrorrZcallWithLogger)rtimeoutlerrZ_drdwr"r Z selectablerrrdoPolls   zEPollReactor.doPollN)__name__ __module__ __qualname____doc__rrZ_POLL_DISCONNECTEDr Z_POLL_INr Z _POLL_OUTrr$r(r+r1r2r3r8r9r:rAZ doIterationrrrrr %s    r cCst}ddlm}||dS)z& Install the epoll() reactor. r)installReactorN)r Ztwisted.internet.mainrF)prFrrrinstalls rH)rEr&selectZzope.interfacerZtwisted.internetrZtwisted.internet.interfacesrZtwisted.pythonrgetattrrrrr r AttributeErrorr* ImportErrorrZ_PollLikeMixinr rH__all__rrrrs&         T