a
=Æ*fvN ã @ sf d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZmZ ee e
eeeehƒZze W n ey¢ i ZY n0 dd„ ZG dd„ deƒZee e fZ dd „ Z d
d „ Z d d
„ Z!dd„ Z"d&dd„Z#d'dd„Z$e$Z%d(dd„Z&G dd„ dƒZ'G dd „ d e'ƒZ(d d „ Z)d)d d „Z*ej+d!krbG d"d#„ d#ƒZ,G d$d%„ d%e'ƒZ-dS )*a Basic infrastructure for asynchronous socket service clients and servers.
There are only two ways to have a program on a single processor do "more
than one thing at a time". Multi-threaded programming is the simplest and
most popular way to do it, but there is another very different technique,
that lets you have nearly all the advantages of multi-threading, without
actually using multiple threads. it's really only practical if your program
is largely I/O bound. If your program is CPU bound, then pre-emptive
scheduled threads are probably what you really need. Network servers are
rarely CPU-bound, however.
If your operating system supports the select() system call in its I/O
library (and nearly all do), then you can use it to juggle multiple
communication channels at once; doing other work while your I/O is taking
place in the "background." Although this strategy can seem strange and
complex, especially at first, it is in many ways easier to understand and
control than multi-threaded programming. The module documented here solves
many of the difficult problems for you, making the task of building
sophisticated high-performance network servers and clients a snap.
é N)
ÚEALREADYÚ EINPROGRESSÚ EWOULDBLOCKÚ
ECONNRESETÚEINVALÚENOTCONNÚ ESHUTDOWNÚEISCONNÚEBADFÚ ECONNABORTEDÚEPIPEÚEAGAINÚ errorcodec
C sF z t | ¡W S tttfy@ | tv r4t| Y S d| Y S 0 d S )NzUnknown error %s)ÚosÚstrerrorÚ
ValueErrorÚ
OverflowErrorÚ NameErrorr )Úerr© r ú ZpollsterrA r! r/ r? r r r Úpoll2¤ s(
rE ç >@Fc C sb |d u r t }|r ttdƒr t}nt}|d u r>|r^|| |ƒ q,n |r^|dkr^|| |ƒ |d }q>d S )NrB r é )r3 Úhasattrr' rE rB )r= Zuse_pollr> ÚcountZpoll_funr r r ÚloopÀ s
rJ c @ s2 e Zd ZdZdZdZdZdZdZe dhƒZ
dAdd„Z dd„ Z dBdd „Z
dCd
d „Zejejfd d
„ZdDdd„Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zd d „ Zd d „ Zd d!„ Z d"d#„ Z d$d%„ Z dEd'd(„Z d)d*„ Z d+d,„ Z!d-d.„ Z"d/d0„ Z#d1d2„ Z$d3d4„ Z%d5d6„ Z&d7d8„ Z'd9d:„ Z(d;d<„ Z)d=d>„ Z*d?d@„ Z+dS )FÚ
dispatcherFNÚwarningc
C sž |d u rt | _n|| _d | _|r”| d¡ | ||¡ d| _z| ¡ | _W qš ty } z.|j t
t fv rpd| _n | |¡ ‚ W Y d }~qšd }~0 0 nd | _
d S )NFT)r3 Ú_mapÚ_filenoÚ setblockingÚ
set_socketÚ connectedZ getpeernameÚaddrr, r- r r Ú del_channelÚsocket)ÚselfÚsockr> r r r r Ú__init__Ü s
zdispatcher.__init__c C s– | j jd | j j g}| jr.| jr.| d¡ n| jr>| d¡ | jd ur€z| d| j ¡ W n" ty~ | t| jƒ¡ Y n0 dd |¡t
| ƒf S )NÚ.Z listeningrQ z%s:%dz <%s at %#x>ú ) Ú __class__r r r9 rR r8 rQ Ú TypeErrorÚreprÚjoinÚid)rU Ústatusr r r Ú__repr__ü s
zdispatcher.__repr__c C s |d u r| j }| || j< d S r )rM rN )rU r> r r r Ú add_channel s zdispatcher.add_channelc C s, | j }|d u r| j}||v r"||= d | _ d S r )rN rM )rU r> rA r r r rS s zdispatcher.del_channelc C s. ||f| _ t ||¡}| d¡ | |¡ d S ©NF)Zfamily_and_typerT rO rP )rU ZfamilyÚtyperV r r r Ú
create_socket s
zdispatcher.create_socketc C s || _ | ¡ | _| |¡ d S r )rT ÚfilenorN ra ©rU rV r> r r r rP s
zdispatcher.set_socketc C sB z*| j t jt j| j t jt j¡dB ¡ W n ty< Y n0 d S )NrG )rT Z
setsockoptÚ
SOL_SOCKETZ SO_REUSEADDRÚ
getsockoptr, ©rU r r r Úset_reuse_addr# s
ÿÿþ zdispatcher.set_reuse_addrc C s dS ©NTr ri r r r r6 4 s zdispatcher.readablec C s dS rk r ri r r r r7 7 s zdispatcher.writablec C s( d| _ tjdkr |dkr d}| j |¡S )NTÚnté )r9 r ÚnamerT Úlisten)rU Znumr r r ro > s zdispatcher.listenc C s || _ | j |¡S r )rR rT Úbind)rU rR r r r rp D s zdispatcher.bindc C sp d| _ d| _| j |¡}|tttfv s8|tkrBtj dkrB|| _
d S |dt fv r^|| _
| ¡ nt
|t| ƒ‚d S )NFTrl r )rQ Ú
connectingrT Z
connect_exr r r r r rn rR r Úhandle_connect_eventr, r )rU Úaddressr r r r ÚconnectH s ÿÿ
zdispatcher.connectc
C sv z| j ¡ \}}W nV ty& Y d S tyh } z,|jtttfv rRW Y d }~d S ‚ W Y d }~nd }~0 0 ||fS d S r )rT Úacceptr[ r, r- r r r
)rU ZconnrR Úwhyr r r ru V s zdispatcher.acceptc
C sv z| j |¡}|W S typ } zF|jtkr:W Y d }~dS |jtv rZ| ¡ W Y d }~dS ‚ W Y d }~n
d }~0 0 d S ©Nr )rT Úsendr, r- r r. r+ )rU ÚdataÚresultrv r r r rx d s
zdispatcher.sendc
C st z(| j |¡}|s | ¡ W dS |W S W nF tyn } z.|jtv rX| ¡ W Y d }~dS ‚ W Y d }~n
d }~0 0 d S ©Nó )rT Úrecvr+ r, r- r. )rU Z buffer_sizery rv r r r r} q s
zdispatcher.recvc
C sl d| _ d| _d| _| ¡ | jd urhz| j ¡ W n4 tyf } z |jtt fvrR‚ W Y d }~n
d }~0 0 d S rb )
rQ r9 rq rS rT Úcloser, r- r r
)rU rv r r r r~ ƒ s
zdispatcher.closec C s t j dt|ƒ ¡ d S )Nzlog: %s
)ÚsysÚstderrr$ Ústr)rU Úmessager r r Úlog“ s zdispatcher.logÚinfoc C s || j vrtd||f ƒ d S )Nz%s: %s)Úignore_log_typesÚprint)rU r‚ rc r r r Úlog_info– s
zdispatcher.log_infoc C s: | j r| ¡ n&| js.| jr$| ¡ | ¡ n| ¡ d S r )r9 Ú
handle_acceptrQ rq rr Ú handle_readri r r r r š s
z dispatcher.handle_read_eventc C s@ | j t jt j¡}|dkr(t|t|ƒƒ‚| ¡ d| _d| _d S )Nr TF) rT rh rg ÚSO_ERRORr, r Úhandle_connectrQ rq ©rU r r r r rr ¦ s z dispatcher.handle_connect_eventc C s* | j r
d S | js | jr | ¡ | ¡ d S r )r9 rQ rq rr Ú handle_writeri r r r r# ® s z dispatcher.handle_write_eventc C s0 | j t jt j¡}|dkr$| ¡ n| ¡ d S rw )rT rh rg rŠ r+ Ú handle_exptrŒ r r r r% ¹ s
z dispatcher.handle_expt_eventc C sX t ƒ \}}}}z t| ƒ}W n dt| ƒ }Y n0 | d||||f d¡ | ¡ d S )Nz)<__repr__(self) failed for object at %0x>z:uncaptured python exception, closing channel %s (%s:%s %s)Úerror)Úcompact_tracebackr\ r^ r‡ r+ )rU ZnilÚtÚvÚtbinfoZ self_reprr r r r È s üù zdispatcher.handle_errorc C s | dd¡ d S )Nz!unhandled incoming priority eventrL ©r‡ ri r r r rŽ Ü s zdispatcher.handle_exptc C s | dd¡ d S )Nzunhandled read eventrL r” ri r r r r‰ ß s zdispatcher.handle_readc C s | dd¡ d S )Nzunhandled write eventrL r” ri r r r r â s zdispatcher.handle_writec C s | dd¡ d S )Nzunhandled connect eventrL r” ri r r r r‹ å s zdispatcher.handle_connectc C s | ¡ }|d ur| j|Ž d S r )ru Úhandle_accepted)rU Zpairr r r rˆ è s zdispatcher.handle_acceptc C s | ¡ | dd¡ d S )Nzunhandled accepted eventrL )r~ r‡ )rU rV rR r r r r• í s zdispatcher.handle_acceptedc C s | dd¡ | ¡ d S )Nzunhandled close eventrL )r‡ r~ ri r r r r+ ñ s zdispatcher.handle_close)NN)N)N)N)r„ ),r r r ÚdebugrQ r9 rq ÚclosingrR Ú frozensetr… rW r` ra rS rT ZAF_INETZ SOCK_STREAMrd rP rj r6 r7 ro rp rt ru rx r} r~ rƒ r‡ r rr r# r% r rŽ r‰ r r‹ rˆ r• r+ r r r r rK Ò sJ
rK c @ s6 e Zd Zd dd„Zdd„ Zdd„ Zdd „ Zd
d „ ZdS )
Údispatcher_with_sendNc C s t | ||¡ d| _d S r{ )rK rW Ú
out_bufferrf r r r rW ü s z dispatcher_with_send.__init__c C s. d}t | | jd d… ¡}| j|d … | _d S )Nr i )rK rx rš )rU Znum_sentr r r Ú
initiate_send s z"dispatcher_with_send.initiate_sendc C s | ¡ d S r )r› ri r r r r s z!dispatcher_with_send.handle_writec C s | j pt| jƒS r )rQ Úlenrš ri r r r r7 s z dispatcher_with_send.writablec C s0 | j r| dt|ƒ ¡ | j| | _| ¡ d S )Nz
sending %s)r– r‡ r\ rš r› )rU ry r r r rx s zdispatcher_with_send.send)NN)r r r rW r› r r7 rx r r r r r™ ú s
r™ c C s‚ t ¡ \} }}g }|s tdƒ‚|rL| |jjj|jjjt|j ƒf¡ |j
}q ~|d \}}}d dd„ |D ƒ¡}|||f| ||fS )Nztraceback does not existéÿÿÿÿrY c S s g | ] }d | ‘qS )z
[%s|%s|%s]r )Ú.0Úxr r r Ú
& r| z%compact_traceback..) r Úexc_infoÚAssertionErrorr8 Útb_frameÚf_codeÚ co_filenameÚco_namer Ú tb_linenoÚtb_nextr] )r‘ r’ Útbr“ ÚfileÚfunctionÚliner„ r r r r s ýr c C s | d u r t } t| ¡ ƒD ]j}z | ¡ W q ty^ } z |jtkrDn|sJ‚ W Y d }~qd }~0 typ ‚ Y q |s|‚ Y q0 q| ¡ d S r ) r3 r4 Úvaluesr~ r, r- r
r Úclear)r> Z
ignore_allrŸ r r r Ú close_all) s
r¯ Úposixc @ sN e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zdd
d „ZeZeZ d d
„ Z
dd„ Z d S )Ú file_wrapperc C s t |¡| _d S r )r ÚduprA ©rU rA r r r rW N s zfile_wrapper.__init__c C s* | j dkr tjd| t| d | ¡ d S )Nr zunclosed file %r)Úsource)rA ÚwarningsÚwarnÚResourceWarningr~ ri r r r Ú__del__Q s
ÿzfile_wrapper.__del__c G s t j| jg|¢R Ž S r )r r" rA ©rU Úargsr r r r} W s zfile_wrapper.recvc G s t j| jg|¢R Ž S r )r r$ rA r¹ r r r rx Z s zfile_wrapper.sendNc C s( |t jkr |t jkr |s dS tdƒ‚d S )Nr z-Only asyncore specific behaviour implemented.)rT rg rŠ ÚNotImplementedError)rU ÚlevelZoptnameÚbuflenr r r rh ] s
ÿþzfile_wrapper.getsockoptc C s( | j dk rd S | j }d| _ t |¡ d S )Nr r )rA r r~ r³ r r r r~ h s
zfile_wrapper.closec C s | j S r )rA ri r r r re o s zfile_wrapper.fileno)N) r r r rW r¸ r} rx rh r" r$ r~ re r r r r r± I s
r± c @ s e Zd Zddd„Zdd„ ZdS )Úfile_dispatcherNc C sN t | d |¡ d| _z | ¡ }W n ty2 Y n0 | |¡ t |d¡ d S )NTF)rK rW rQ re ÚAttributeErrorÚset_filer Ú set_blocking)rU rA r> r r r rW t s
zfile_dispatcher.__init__c C s" t |ƒ| _| j ¡ | _| ¡ d S r )r± rT re rN ra r³ r r r rÀ s
zfile_dispatcher.set_file)N)r r r rW rÀ r r r r r¾ r s
r¾ )r2 N)r2 N)rF FNN)NF).Ú__doc__r' rT r r: rµ r r- r r r r r r r r r
r r r
r r˜ r. r3 r r Ú Exceptionr ÚKeyboardInterruptÚ
SystemExitr r" r$ r& r1 rB rE Zpoll3rJ rK r™ r r¯ rn r± r¾ r r r r Ú sD < ÿ
'
*
)