j6UdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZdd lmZdd lmZej6Zej:Zej>Z ejBZ"ej6Z#ej:Z$ej>Z%ejLe'Z(ejRd k7rdd Z*nddZ*d d!dZ+d d"dZ,d d!dZ-d d"dZ.d d!dZ/e0edr[eejbejdzeejbejfze ejbejdzejfziZ4niZ4d d!dZ5e0edrNeejleejne ejlejnziZ8ejlejnzZ9niZ8d d!dZ:d#dZ;er ejxZB%%B.*B1.B11 C: C C Cc K| tdt t}d fd } t|} |tz}|t z}|s|st jd| jd |r|j||t|r|j||t t j|d{|r|j!||r|j#| |j% }7=#t$rYFwxYw#|r|j!||r|j#|wwxYw#t&$r}t j(d|d}~wt*$r}|j,} | cYd}~Sd}~wwxYww)a Coroutine waiting for a generator to complete. :param gen: a generator performing database operations and yielding `Ready` values when it would block. :param fileno: the file descriptor to wait on. :param interval: interval (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. :return: whatever `!gen` returns on completion. Behave like in `wait()`, but exposing an `asyncio` interface. Nr#c0|zjyrsetstateevr3s rwakeupzwait_async..wakeups  r!bad poll status: rrr>rreturnNone)r&rrr'WAIT_RWAIT_Wr InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerr*OSErrorrr,r-) r.rr/loopr@r0readerwriterrr4r?r3s @@r wait_asyncrT~sw@AA B  D  IZFZFfoo(9!&=>> HHJE88 /"2779h777&&v.&&v.A+8#&&v.&&v. E  !;<"D  s})F A>E.D D  D 9E D DDDD)EE F E%% F 1 F>F ?F F  F c K| tdt t}d fd } t|\}} |tz}|t z}|s|st jd| jd |r|j||t|r|j||t |r$ t j|d{n jd{|r|j!||r|j#| |j% \}}7Y#t$rYIwxYw7Q#|r|j!||r|j#|wwxYw#t&$r}|j(} | cYd}~Sd}~wwxYww)a Coroutine waiting for a connection generator to complete. :param gen: a generator performing database operations and yielding (fd, `Ready`) pairs when it would block. :param interval: interval (in seconds) to check for other interrupt, e.g. to allow Ctrl-C. :return: whatever `!gen` returns on completion. Behave like in `wait()`, but take the fileno to wait from the generator itself, which might change during processing. Nr#c*|jyrr;r=s rr@zwait_conn_async..wakeups r!rArrB)r&rrr'rErFrrGrHrIrJrKrLrrMrrNrOr*r,r-) r.r/rQr@rr0rRrSrr4r?r3s @@rwait_conn_asyncrWsu@AA B  D I ZFZFfoo(9!&=>> HHJE88 /&rwwy(;;;'')OO&&v.&&v.IFA1<'$&&v.&&v.  s)F BE)0D=3D,D*D,D=)D;*D=.= 1024. On Windows it's fine. Nr#r rr)r&r'r)rErFrrrrJrLr*rPr,r-) r.rr/r0emptyfnlistrlwlxlr3rr4s r wait_selectr^s @AA If*%f*% JBB  ((()CDDE A#& E  !;<"D  s*B B C$B:: C CCC EpollSelectorcD| td t|}|dkrd}tj5}t|}|j || |j |x}s!t||jt5|dd}d}|tjzrt}|tjzr |tz}|j|}t|}|j||#1swYyxYw#t$r} | j } | cYd} ~ Sd} ~ wwxYw)a$ Wait for a generator using epoll where supported. Parameters are like for `wait()`. If it is detected that the best selector strategy is `epoll` then this function will be used instead of `wait`. See also: https://linux.die.net/man/2/epoll_ctl BUG: if the connection FD is closed, `epoll.poll()` hangs. Same for EpollSelector. For this reason, wait_poll() is currently preferable. To reproduce the bug: export PSYCOPG_WAIT_FUNC=wait_epoll pytest tests/test_concurrency.py::test_concurrent_close Nr#rr )r&r'r)epoll_epoll_evmasksr(pollrr*r+EPOLLINrJEPOLLOUTrLmodifyr,r-) r.rr/r0rbevmaskfileevsr?r3rr4s r wait_epollrj-s  @AA I a<H \\^u#A&F NN66 *#(::h#777$V,HHZ(QZ]&#E'W$EHHUO'* VV,^$  s5&C>B=C22C;7C>;C>> D DDD PollSelectorc| td t|}|dkrd}nt|dz}tj}t |}|j || |j |x}s|jt*|dd}d}|tjzrt}|tjzr |tz}|s)|tzr t|tj d|j|}t |}|j#||#t$$r} | j&} | cYd} ~ Sd} ~ wwxYw)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nr#rg@@r r)r&r'intr)rd _poll_evmasksr(r*r+POLLINrJPOLLOUTrLPOLL_BADrrrrgr,r-) r.rr/r0rdrhrir?r3rr4s r wait_pollrrgs. @AA# I a<H8f,-H{{}q! ff%#yy22G2$ABEFMM!FNN" R(] ((()CDDA"1%F KK '),  sDD E) D<6E<Ectjjdx}r |jdry yy#t$r }t j d|Yd}~yd}~wwxYw)z Detect if some greenlet library has patched the select library. If this is the case, avoid to use the wait_c function as it doesn't behave in a collaborative way. Currently supported: gevent. z gevent.monkeyr)Tz+failed to detect gevent monkey-patching: %sNF)sysmodulesgetis_module_patchedrloggerwarning)mrs r_is_select_patchedr{sf KKOOO ,,q, N""8,-  N NNH" M M  Ns8 A!AA!rrMPSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got SelectSelector)rrmrCrD)ra)r.z PQGen[RV]rrmr/floatrCr )r.z PQGenConn[RV]r/rrCr )rCbool)E__doc__ __future__rrrtr)logging selectorsasynciorrrrrr rabcr r r r_enumsrr_cmodulerRrEWrFRWWAIT_RWNONEr+rJrLREADY_RW getLogger__name__rxplatformrr5r8rTrWr^hasattr EPOLLONESHOTrerfrcrjrorprnrqrrr{wait_c__annotations__environfname startswithglobals ImportErrorrMgetattrr r!rrsG# AA%//"    '' ZZ '' '' 88   8 $<<7I!HD:z:@%P 9o&##fnn4##foo5$$v~~5GN N,^ 9n% /M /0HM+\( __F"**$ JJ* +E   G $WY(> 9    9U D #,,')2D2F D')5Et"LL D Y' D Dr!