U
|
Z ±d ã @ s¾ d dl Z d dlZd dlmZ d dlZd dlmZ d dlZd dlZddl m
|
Z
|
ddl m Z ddl mZ erejd kr|ejd
|
kstejd d d d G dd dZejd d dG dd dZdS )é N)Ú TYPE_CHECKING)Úcontextmanageré )Ú_coreé )Ú_public)ÚWakeupSocketpairÚlinuxÚwin32TF)ÚslotsÚeqÚfrozenc @ s( e Zd Ze ¡ Ze ¡ ZejddZdS )Ú_KqueueStatisticsÚkqueue©ÚdefaultN)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibÚ tasks_waitingÚmonitorsÚbackend© r r úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_io_kqueue.pyr s r )r r c @ s e Zd ZejejdZejedZ eje
|
dZ ejddZ dd Z dd Zdd Zd
|
d Zd d Zdd Zedd Zeedd Zedd Zdd Zedd Zedd Zed d ZdS ) ÚKqueueIOManager)ÚfactoryNr c C s8 t | jjt jt j¡}| j |gd¡ | jj ¡ | _ d S ©Nr )
|
ÚselectÚkeventÚ _force_wakeupZ wakeup_sockÚKQ_FILTER_READÚ KQ_EV_ADDÚ_kqueueÚcontrolÚfilenoÚ_force_wakeup_fd)ÚselfZforce_wakeup_eventr r r Ú__attrs_post_init__ s ÿz#KqueueIOManager.__attrs_post_init__c C sD d}d}| j ¡ D ]$}t|tjkr.|d7 }q|d7 }qt||dS )Nr r )r r )Ú _registeredÚvaluesÚtyper ÚTaskr )r( r r Úreceiverr r r Ú
|
statistics' s
|
|
zKqueueIOManager.statisticsc C s | j ¡ | j ¡ d S ©N)r$ Úcloser! ©r( r r r r1 1 s
|
zKqueueIOManager.closec C s | j ¡ d S r0 )r! Z wakeup_thread_and_signal_safer2 r r r Ú force_wakeup5 s z KqueueIOManager.force_wakeupc C sD t | jd }g }| j g ||¡}||7 }t ||k r:q@qd}q|S )Nr r )Úlenr* r$ r% )r( ÚtimeoutZ
|
max_eventsÚeventsÚbatchr r r Ú
|
get_events8 s zKqueueIOManager.get_eventsc C s| |D ]r}|j |jf}|j | jkr,| j ¡ q| j| }|jtj@ rJ| j|= t |t
|
j krlt
|
|t |¡¡ q| |¡ qd S r0 )ÚidentÚfilterr' r! Zdrainr* Úflagsr Ú KQ_EV_ONESHOTr, r r- Ú
|
rescheduleÚoutcomeÚValueÚ
|
put_nowait)r( r6 ÚeventÚkeyr. r r r Úprocess_eventsI s
|
|
z KqueueIOManager.process_eventsc C s | j S r0 )r$ r2 r r r Úcurrent_kqueueb s z KqueueIOManager.current_kqueuec c sH ||f}|| j kr t d¡t ¡ }|| j |< z
|
|V W 5 | j |= X d S )NúAattempt to register multiple listeners for same ident/filter pair)r* r ÚBusyResourceErrorZUnboundedQueue)r( r9 r: rB Úqr r r Úmonitor_keventf s
|
ÿ
|
|
z KqueueIOManager.monitor_keventc sJ ||fj kr t d¡t ¡ j < fdd}t |¡I d H S )NrE c s | }|t jjkr j= |S r0 )r ÚAbortÚ SUCCEEDEDr* )Z raise_cancelÚr©Ú
|
abort_funcrB r( r r Úabort~ s z*KqueueIOManager.wait_kevent.<locals>.abort)r* r rF Z current_taskZwait_task_rescheduled)r( r9 r: rM rN r rL r Ú wait_keventu s
|
ÿzKqueueIOManager.wait_keventc sd t ts ¡ tjtjB }t |¡}j |gd¡ fdd} |¡I d H d S )Nr c
|
sf t t j¡}zj |gd¡ W n8 tk
|
r\ } z|jtjtjfkrJn W 5 d }~X Y nX t j
|
j S r ) r r Ú KQ_EV_DELETEr$ r% ÚOSErrorÚerrnoÚEBADFÚENOENTr rI rJ )Ú_rA Úexc©Úfdr: r( r r rN s z+KqueueIOManager._wait_common.<locals>.abort)
|
Ú
|
isinstanceÚintr& r r# r< r r$ r% rO )r( rX r: r; rA rN r rW r Ú _wait_common s
|
z KqueueIOManager._wait_commonc à s | |tj¡I d H d S r0 )r[ r r" ©r( rX r r r Ú wait_readable¥ s z KqueueIOManager.wait_readablec à s | |tj¡I d H d S r0 )r[ r ÚKQ_FILTER_WRITEr\ r r r Ú wait_writable© s z KqueueIOManager.wait_writablec C s¢ t |ts| ¡ }tjtjfD ]~}||f}| j |¡}|d kr@q t|t j
|
krt ||tj ¡}| j |gd¡ t d¡}t |t |¡¡ | j|= q tdq d S )Nr zanother task closed this fdz.can't close an fd that monitor_kevent is using)rY rZ r& r r" r^ r* Úgetr, r r- r rP r$ r% ZClosedResourceErrorr= r> ÚErrorÚNotImplementedError)r( rX r: rB r. rA rV r r r Únotify_closing s
|
|
|
ÿz KqueueIOManager.notify_closing) r r r r r r r r$ Údictr* r r! r' r) r/ r1 r3 r8 rC r rD r rH rO r[ r] r_ rc r r r r r s0
|
|
|
|
|
r )r ÚsysÚtypingr r> Ú
|
contextlibr r rR Ú r Z_runr Z_wakeup_socketpairr ÚplatformÚAssertionErrorÚsr r r r r r Ú<module> s
|
