Hmd :ddlZddlZddlZddlZddlmZ ddlZn#e$rddlZYnwxYw ddl m Z n#e$r ddl m Z YnwxYw ddl Z n#e$rddl Z YnwxYwddl mZmZmZdZdZdZdZd ZGd d eZGd d eZGddeZdS)N) Condition)Queue)Threadselectlfg?ct|trtj|d}t |dr|}||fS)Nrbfileno) isinstanceintosfdopenhasattrr )ios >/usr/lib/x86_64-linux-gnu/glusterfs/python/syncdaemon/repce.pyioparser#sO!S Ia  q( HHJJ q6Mc`tj|tj|tdS)zpickle args and write out wholly in one syscall ie. not use the ability of pickle to dump directly to a stream, as that would potentially mess up messages by interleaving them N)rwritepickledumps pickle_proto)outargss rsendr.s(HS&,t\2233333rc|t|drtj|jStj|S)zload an object from input stream python2 and python3 compatibility, inf is sys.stdin and is opened as text stream by default. Hence using the buffer attribute in python3 buffer)rrloadr)infs rrecvr"8s8 sH {3:&&&{3rc&eZdZdZddZdZdZdS) RepceServerzRePCe is Hungarian for canola, http://hu.wikipedia.org/wiki/Repce ... also our homebrewed RPC backend where the transport layer is reduced to a pair of filehandles. This is the server component. c||_t||\|_|_||_t |_dS)z}register a backend object .obj to which incoming messages are dispatched, also incoming/outcoming streams N)objrr!rwnumrq)selfr'rrr(s r__init__zRepceServer.__init__Ns7$Q]]$( rc*t|jD]+}t|j}|, |jt|j-#t$rtj dYdSwxYw)z>*** + 9 9 9 L7 8 8 8 8 8 8 9s.A00BBcX |jd}|d}|d}d}|dkrt}nX t|j||dd}n6#t jd}d}tjdYnxYwt|j |||) aMlife of a worker Get message, extract its id, method name and arguments (kwargs not supported), call method on .obj. Send back message id + return value. If method call throws an exception, rescue it, and send back the exception as result (with flag marking it as exception). TrF__repce_version__rNz call failed: ) r)get repce_versiongetattrr'sysexc_infor4 exceptionrr)r*in_dataridrmethexcress rr0zRepceServer.workerbs *fjj&&G!*CAJEC+++#72'$(E22GABBK@CC7,..+CC%o66666 3S ) ) ) *s A1BN)r%)__name__ __module__ __qualname____doc__r+r7r0rrr$r$DsP 9 9 9*****rr$c*eZdZdZdZdZdZdZdS)RepceJobzEclass representing message status we can use for waiting on replyctjtjt jf|_||_t|_d|_ dS)zb - .rid: (process-wise) unique id - .cbk: what we do upon receiving reply FN) rgetpidthread get_identtimerBcbkrleverdone)r*rRs rr+zRepceJob.__init__sD IKK!1!3!3TY[[A[[  rcJdd|jDS)N:c,g|]}t|SrJ)str).0xs r z%RepceJob.__repr__..s222AQ222r)joinrBr*s r__repr__zRepceJob.__repr__s%xx22222333rc|j|js|j|j|jSN)rSacquirerTwaitreleaseresultr]s rrbz RepceJob.waitsN y  JOO    {rc||_|jd|_|j|jdS)NT)rdrSrarTnotifyrc)r*datas rwakeupzRepceJob.wakeupsR     rN)rFrGrHrIr+r^rbrhrJrrrLrL}sZ444rrLcReZdZdZdZdZdZdZGddeZ dZ d Z d S) RepceClientzRePCe is Hungarian for canola, http://hu.wikipedia.org/wiki/Repce ... also our homebrewed RPC backend where the transport layer is reduced to a pair of filehandles. This is the client component. ct||\|_|_i|_t |j}|dS)Nr-)rr!rjtabrlistenr1)r*rrr6s rr+zRepceClient.__init__sB$Q]]$( $+ & & &  rc t|jfddt|j\}}}|j|}|jr||||gi)NTrJ)rr!r"rlpoprR)r*rBrDrErjobs rrmzRepceClient.listensj + DH;B ' ' ' NNMCc9==%%Dx +Sz***  +rc |d}|sd}t|}||j|j<t jdt |d|t |dt|j|j|g|R|S)zwrap arguments in a RepceJob, send them to server and return the RepceJob @cbk to pass on RepceJob can be given as kwarg. rRc&|dr|ddS)Nrr9rJrjrEs rrRzRepceClient.push..cbks!q6!a&L!!rcall  z ...) r;rLrlrBr4debugreprrr)r*methrkwrRrps rpushzRepceClient.pushs ffUmm ! ! ! !}}" $( DJJJJd4jjjjIJJJ TXtx----- rc t|j|g|Riddi}|\}}|rStjt dt ||t t|j|tj dt |d|dt ||S)zRePCe client is callabe, calling it implements a synchronous remote call. We do a .push with a cbk which does a wakeup upon receiving answer, then wait on the RepceJob. rRc,||Sr`)rhrss rz&RepceClient.__call__..s299S>>rz call failed)callmethoderrorrurvz -> ) r{rbr4rrrxrXtyperFrw)r*ryrrprDrEs r__call__zRepceClient.__call__sty DDDD!#A#ABDD99;;S   M"]"&t**$(#&tCyy'9#:#:<<< = = =I DJJJJd3iiiHIII rceZdZdZdZdZdS)RepceClient.mprxzmethod proxy, standard trick to implement rubyesque method_missing in Python A class is a closure factory, you know what I mean, or go read some SICP. c"||_||_dSr`insry)r*rrys rr+zRepceClient.mprx.__init__sDHDIIIrc(|j|jg|RSr`r)r*as rrzRepceClient.mprx.__call__s48DI**** *rN)rFrGrHrIr+rrJrrmprxrs<      + + + + +rrc.|||S)aAthis implements transparent method dispatch to remote object, so that you don't need to call the RepceClient instance like rclient('how_old_are_you_if_born_in', 1979) but you can make it into an ordinary method call like rclient.how_old_are_you_if_born_in(1979) )r)r*rys r __getattr__zRepceClient.__getattr__syyt$$$rc`d|di} |d|d<n#t$rYnwxYw|S)z)used in handshake to verify compatibilityprotor:versionobject)AttributeError)r*ds r __version__zRepceClient.__version__sT dd.// 0 $y//AhKK    D s  ++N) rFrGrHrIr+rmr{rrrrrrJrrrjrjs +++"&+++++v+++ % % %rrj)rr>rQr4 threadingr_threadrO ImportErrorqueuercPickler syncdutilsrrrrr<rrr"rr$rLrjrJrrrs MMMMMMMMMM*)))))))))  444    6*6*6*6*6*&6*6*6*rvB_____&_____s- ))4 AAA AA