U
|
Z ±d8 ã @ sÒ d dl Z d dlZd dlZd dlmZ d dlmZmZ ddlm Z ddl
|
m Z ddl m Z dd lmZ ervejd
|
ksvtejd d d d G dd dZejd d dG dd dZejd d d dG dd dZdS )é N)Ú defaultdict)ÚDictÚ TYPE_CHECKINGé )Ú_coreé )Ú_public)Úwake_all)ÚWakeupSocketpairÚlinuxTF)ÚslotsÚeqÚfrozenc @ s( e Zd Ze ¡ Ze ¡ ZejddZdS )Ú_EpollStatisticsÚepoll©ÚdefaultN)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibÚtasks_waiting_readÚtasks_waiting_writeÚbackend© r r úKd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_io_epoll.pyr s r )r r c @ s0 e Zd ZejddZejddZejddZdS )Ú EpollWaitersNr r )r r r r r Ú read_taskÚ
|
write_taskÚ current_flagsr r r r r ´ s r )r r Úhashc @ s° e Zd ZejejdZejdd ee e
|
f dZ eje dZ ejddZdd Zd d
|
Zd d Zd d Zdd Zdd Zdd Zdd Zedd Zedd Zedd ZdS ) ÚEpollIOManager)Úfactoryc C s t tS ©N)r r r r r r Ú<lambda>À ó zEpollIOManager.<lambda>)r# ÚtypeNr c C s& | j | jjtj¡ | jj ¡ | _d S r$ )Ú_epollÚregisterÚ _force_wakeupZ wakeup_sockÚselectÚEPOLLINÚfilenoÚ_force_wakeup_fd©Úselfr r r Ú__attrs_post_init__Å s z"EpollIOManager.__attrs_post_init__c C sH d}d}| j ¡ D ](}|jd k r(|d7 }|jd k r|d7 }qt||dS )Nr r )r r )Ú _registeredÚvaluesr r r )r0 r r Úwaiterr r r Ú
|
statisticsÉ s
|
|
|
þzEpollIOManager.statisticsc C s | j ¡ | j ¡ d S r$ )r( Úcloser* r/ r r r r6 Ö s
|
zEpollIOManager.closec C s | j ¡ d S r$ )r* Z wakeup_thread_and_signal_safer/ r r r Ú force_wakeupÚ s zEpollIOManager.force_wakeupc C s t dt| j}| j ||¡S )Nr )ÚmaxÚlenr2 r( Úpoll)r0 ÚtimeoutZ
|
max_eventsr r r Ú
|
get_eventsà s zEpollIOManager.get_eventsc C s |D ]\}}|| j kr"| j ¡ q| j| }d|_|tj @ rZ|jd k rZt |j¡ d |_|tj
|
@ r|j d k rt |j ¡ d |_ | |¡ qd S ©Nr ) r. r* Zdrainr2 r r+ r, r r Z
|
rescheduleÚEPOLLOUTr Ú_update_registrations)r0 ÚeventsÚfdÚflagsÚwaitersr r r Úprocess_eventsç s
|
|
|
z EpollIOManager.process_eventsc
|
C sØ | j | }d}|jd k r"|tjO }|jd k r6|tjO }||jkrÈzLz| j ||tj B ¡ W n( t
|
k
|
r | j ||tj B ¡ Y nX ||_W n: t
|
k
|
rÆ } z | j |= t || W Y ¢d S d }~X Y nX |sÔ| j |= d S r= ) r2 r r+ r, r r> r r( ÚmodifyZ EPOLLONESHOTÚOSErrorr) r )r0 rA rC Z wanted_flagsÚexcr r r r? ú s&
|
|
|
|
|
|
|
|
z$EpollIOManager._update_registrationsc st t ts ¡ j t d k r4t d¡t t ¡ ¡ fdd}t
|
|¡I d H d S )Nz1another task is already reading / writing this fdc s t d ¡ tjjS r$ )Úsetattrr? r ZAbortZ SUCCEEDED)Ú_©Ú attr_namerA r0 rC r r Úabort" s
|
z)EpollIOManager._epoll_wait.<locals>.abort) Ú
|
isinstanceÚintr- r2 Úgetattrr ZBusyResourceErrorrH Z current_taskr? Zwait_task_rescheduled)r0 rA rK rL r rJ r Ú _epoll_wait s
|
|
ÿ
|
zEpollIOManager._epoll_waitc à s | |d¡I d H d S )Nr ©rP ©r0 rA r r r Ú wait_readable) s z EpollIOManager.wait_readablec à s | |d¡I d H d S )Nr rQ rR r r r Ú wait_writable- s z EpollIOManager.wait_writablec C s^ t |ts| ¡ }t| j| t d¡ | j|= z| j |¡ W n t t
|
fk
|
rX Y nX d S )Nzanother task closed this fd) rM rN r- r r2 r ZClosedResourceErrorr( Ú
|
unregisterrF Ú
|
ValueErrorrR r r r Únotify_closing1 s
|
þz EpollIOManager.notify_closing)r r r r r r+ r r( r rN r r2 r
|
r* r. r1 r5 r6 r7 r< rD r? rP r rS rT rW r r r r r" » s*
|
ÿ
|
|
r" )r+ Úsysr Ú collectionsr Útypingr r Ú r Z_runr Z
|
_io_commonr Z_wakeup_socketpairr
|
ÚplatformÚAssertionErrorÚsr r r" r r r r Ú<module> s %
|
