ó L]c@sdZddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl Z ddlmZdd lmZeed ƒrÆejejfZndZd „Zd e fd „ƒYZdefd„ƒYZdS(sjThis module wraps the :class:`Socket` and :class:`Context` found in :mod:`pyzmq ` to be non blocking iÿÿÿÿ(tprint_functionN(tContext(tSocketi(t_Poller(t AsyncResult(tget_hubtRCVTIMEOcCs2y|jƒWntk r-}|jƒnXdS(sNsimple wrapper for stopping an Event, allowing for method rename in gevent 1.0N(tstoptAttributeErrortcancel(tevtte((s-lib/python2.7/site-packages/zmq/green/core.pyt_stop#st_SocketcBs×eZdZeZeZdZdZdZ dZ eZ e Z d„Zd„Zdd„Zd„Zd„Zddd„Zd„Zd „Zd eed „Zd eed „Zd „Zd„Zd„Zd„ZRS(sOGreen version of :class:`zmq.Socket` The following methods are overridden: * send * recv To ensure that the ``zmq.NOBLOCK`` flag is set and that sending or receiving is deferred to the hub if a ``zmq.EAGAIN`` (retry) error is raised. The `__state_changed` method is triggered when the zmq.FD for the socket is marked as readable and triggers the necessary read and write events (which are waited for in the recv and send methods). Some double underscore prefixes are used to minimize pollution of :class:`zmq.Socket`'s namespace. g333333'@cOs9tt|ƒj||Žt|_t|_|jƒdS(N(tsuperR t__init__tFalset_Socket__in_send_multipartt_Socket__in_recv_multipartt_Socket__setup_events(tselftatkw((s-lib/python2.7/site-packages/zmq/green/core.pyRFs  cCs|jƒdS(N(tclose(R((s-lib/python2.7/site-packages/zmq/green/core.pyt__del__MscCs$tt|ƒj|ƒ|jƒdS(N(RR Rt_Socket__cleanup_events(Rtlinger((s-lib/python2.7/site-packages/zmq/green/core.pyRPscCsIt|ddƒr+t|jƒd|_n|jjƒ|jjƒdS(Nt _state_event(tgetattrtNoneR Rt_Socket__writabletsett_Socket__readable(R((s-lib/python2.7/site-packages/zmq/green/core.pyt__cleanup_eventsTs    cCs¿tƒ|_tƒ|_|jjƒ|jjƒy>tƒjj|jtj ƒdƒ|_ |j j |j ƒWnHt k rºddlm}||jtj ƒ|j dtƒ|_ nXdS(Niiÿÿÿÿ(t read_eventtpersist(RR RRRtlooptiot getsockopttzmqtFDRtstartt_Socket__state_changedRt gevent.coreR"tTrue(RR"((s-lib/python2.7/site-packages/zmq/green/core.pyt__setup_events]s    ' cCs­|jr|jƒdSytt|ƒjtjƒ}Wn6tjk rn}|jj |ƒ|j j |ƒn;X|tj @rŒ|jj ƒn|tj @r©|j j ƒndS(N(tclosedRRR R&R'tEVENTStZMQErrorRt set_exceptionR tPOLLOUTRtPOLLIN(Rteventt_evtypeteventstexc((s-lib/python2.7/site-packages/zmq/green/core.pyt__state_changedks    cCs?|jjƒstdƒ‚tƒ|_tjƒ}|j}|rWtjd|ƒ}nd}zºy*|rv|j ƒn|jj dt ƒWn‰tjk r}||k r±‚ntjƒ}|j r|r|||kr|j tjƒtj@rtd|jdtjƒqnXWd|r-|jƒn|jjƒXdS(Ns.Only one greenlet can be waiting on this eventtsecondstblocks6BUG: gevent may have missed a libzmq send event on %i!tfile(RtreadytAssertionErrorRttimet_gevent_bug_timeouttgeventtTimeoutRR)tgetR,t _debug_geventR&R'R/R2tprintR(tsyststderrR R(Rttictdtttimeoutttttoc((s-lib/python2.7/site-packages/zmq/green/core.pyt _wait_write{s,      % cCs?|jjƒstdƒ‚tƒ|_tjƒ}|j}|rWtjd|ƒ}nd}zºy*|rv|j ƒn|jj dt ƒWn‰tjk r}||k r±‚ntjƒ}|j r|r|||kr|j tjƒtj@rtd|jdtjƒqnXWd|r-|jƒn|jjƒXdS(Ns.Only one greenlet can be waiting on this eventR9R:s6BUG: gevent may have missed a libzmq recv event on %i!R;(R R<R=RR>R?R@RARR)RBR,RCR&R'R/R3RDR(RERFR R(RRGRHRIRJRK((s-lib/python2.7/site-packages/zmq/green/core.pyt _wait_read˜s,      % icKs|tj@rSz(tt|ƒj|||||}Wd|jsN|jƒnX|S|tjO}xštrüy%tt|ƒj||||ƒ}WnDtjk rÔ}|j tj krï|jsË|jƒn‚qïnX|jsë|jƒn|S|j ƒqcWdS(s£send, which will only block current greenlet state_changed always fires exactly once (success or fail) at the end of this method. N( R'tNOBLOCKRR tsendRR*R,R0terrnotEAGAINRL(RtdatatflagstcopyttracktkwargstmsgR ((s-lib/python2.7/site-packages/zmq/green/core.pyRO·s& (   %     cCsø|tj@rMz"tt|ƒj|||ƒ}Wd|jsH|jƒnX|S|tjO}x—tróy"tt|ƒj|||ƒ}WnDtjk rË}|j tj kræ|jsÂ|jƒn‚qænX|jsâ|jƒn|S|j ƒq]WdS(s£recv, which will only block current greenlet state_changed always fires exactly once (success or fail) at the end of this method. N( R'RNRR trecvRR*R,R0RPRQRM(RRSRTRURWR ((s-lib/python2.7/site-packages/zmq/green/core.pyRXÙs& "   "     cOsCt|_ztt|ƒj||Ž}Wdt|_|jƒX|S(sAwrap send_multipart to prevent state_changed on each partial sendN(R,RRR tsend_multipartRR*(RtargsRVRW((s-lib/python2.7/site-packages/zmq/green/core.pyRYös    cOsCt|_ztt|ƒj||Ž}Wdt|_|jƒX|S(sAwrap recv_multipart to prevent state_changed on each partial recvN(R,RRR trecv_multipartRR*(RRZRVRW((s-lib/python2.7/site-packages/zmq/green/core.pyR[s    cCsW|tkrtjdtƒntt|ƒj|ƒ}|tjkrS|j ƒn|S(s+trigger state_changed on getsockopt(EVENTS)s0TIMEO socket options have no effect in zmq.green( tTIMEOStwarningstwarnt UserWarningRR RBR'R/R*(Rtopttoptval((s-lib/python2.7/site-packages/zmq/green/core.pyRB s   cCs8|tkrtjdtƒntt|ƒj||ƒS(sset socket options0TIMEO socket options have no effect in zmq.green(R\R]R^R_RR R(RR`tval((s-lib/python2.7/site-packages/zmq/green/core.pyRs N(t__name__t __module__t__doc__RRRRRR RR?RCRt _poller_classRRRRRR*RLRMR,RORXRYR[RBR(((s-lib/python2.7/site-packages/zmq/green/core.pyR +s.      " t_ContextcBseZdZeZRS(srReplacement for :class:`zmq.Context` Ensures that the greened Socket above is used in calls to `socket`. (RcRdReR t _socket_class(((s-lib/python2.7/site-packages/zmq/green/core.pyRgs((Ret __future__RRER>R]R'Rt_original_ContextRt_original_SockettpollRR@t gevent.eventRt gevent.hubRthasattrRtSNDTIMEOR\R R Rg(((s-lib/python2.7/site-packages/zmq/green/core.pyt s"      ï