U  W[-!ã@sšdZddlmZmZddlmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZe eƒGd 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|eZdZdZeeBZeZe 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)}. cCs<tdƒ|_tƒ|_tƒ|_i|_t |¡|_tj   |¡dS)zm Initialize epoll object, file descriptor tracking dictionaries, and the base class. iN) 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 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ÚmodifyÚregisterÚadd) rÚxerÚprimaryÚotherÚ selectablesÚeventÚ antieventÚfdÚflagsrrrÚ_addMs zEPollReactor._addc Csbz| ||j|j|jtt¡Wn>tk r\}z |jtjkrJ|j   |¡n‚W5d}~XYnXdS)zR Add a FileDescriptor for notification of data available to read. N) r&rrrrrÚIOErrorÚerrnoÚEPERMrÚ addReader)rÚreaderÚerrrr*hsÿ zEPollReactor.addReaderc Csbz| ||j|j|jtt¡Wn>tk r\}z |jtjkrJ|j   |¡n‚W5d}~XYnXdS)zS Add a FileDescriptor for notification of data available to write. N) r&rrrrrr'r(r)rÚ addWriter)rÚwriterr,rrrr-ysÿ 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)rÚitemsrrZ unregisterÚremove) rrrr r!r"r#r$Zfdesr%rrrÚ_removeŠs zEPollReactor._removecCs:|j |¡r|j |¡dS| ||j|j|jtt¡dS)zQ Remove a Selectable for notification of data available to read. N) rZ isReadingÚ removeReaderr2rrrrr)rr+rrrr3¤s  ÿzEPollReactor.removeReadercCs:|j |¡r|j |¡dS| ||j|j|jtt¡dS)zR Remove a Selectable for notification of data available to write. N) rZ isWritingÚ removeWriterr2rrrrr)rr.rrrr4¯s  ÿzEPollReactor.removeWritercs6ˆ ‡fdd„ˆjDƒ‡fdd„ˆjDƒ¡ˆj ¡S)zD Remove all selectables, and return a list of them. csg|]}ˆj|‘qSr©r©Ú.0r$rrrÚ ¿sz*EPollReactor.removeAll..csg|]}ˆj|‘qSrr5r6rrrr8Às)Z _removeAllrrrÚ removeAllrrrrr9ºs þýzEPollReactor.removeAllcs‡fdd„ˆjDƒˆj ¡S)Ncsg|]}ˆj|‘qSrr5r6rrrr8Åsz+EPollReactor.getReaders..)rrÚ getReadersrrrrr:ÄsÿzEPollReactor.getReaderscs‡fdd„ˆjDƒˆj ¡S)Ncsg|]}ˆj|‘qSrr5r6rrrr8Êsz+EPollReactor.getWriters..)rrÚ getWritersrrrrr;ÉsÿzEPollReactor.getWritersc Cs®|dkr d}z|j |t|jƒ¡}Wn:tk r^}z|jtjkrLWY¢ dS‚W5d}~XYnX|j}|D]>\}}z|j|}Wntk r”YqjXt   |||||¡qjdS)z1 Poll the poller for new events. Nr/) rZpollÚlenrr'r(ZEINTRZ_doReadOrWriteÚKeyErrorr ZcallWithLogger)rZtimeoutÚlÚerrZ_drdwr$r"Z selectablerrrÚdoPollÎs   zEPollReactor.doPollN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrZ_POLL_DISCONNECTEDrZ_POLL_INrZ _POLL_OUTrr&r*r-r2r3r4r9r:r;r@Z doIterationrrrrr s    r cCstƒ}ddlm}||ƒdS)z& Install the epoll() reactor. r)ÚinstallReactorN)r Ztwisted.internet.mainrE)ÚprErrrÚinstallðs rG)rDZ __future__rrZselectrrrrrr(Zzope.interfacer Ztwisted.internet.interfacesr Ztwisted.pythonr Ztwisted.internetr rZ_PollLikeMixinr rGÚ__all__rrrrÚs     U