o DDi6@sUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddlm Z ddlmZddlmZmZmZmZdd lmZdd lmZdd lmZejZejZejZ ej!Z"ejZ#ejZ$ejZ%e&e'Z(ej)d krxd4ddZ*nd4ddZ*d5d6ddZ+d5d7ddZ,d5d6dd Z-d5d7d!d"Z.d5d6d#d$Z/e0ed%reej1ej2Beej1ej3Be ej1ej2Bej3BiZ4niZ4d5d6d&d'Z5e0ed(reej6eej7e ej6ej7BiZ8ej6ej7BZ9niZ8d5d6d)d*Z:d8d,d-Z;erejvrej>d0Z?e?@d1re?eAvreBd2e?eAe?ZCdSer'ej)d kr'e;s'et|}t,}||| |j|d}s*t||tq|dd}||}|||q1s?wYWdSty]}z |j }|WYd}~Sd}~ww)a Wait for a generator using the best strategy available. :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. Consume `!gen`, scheduling `fileno` for completion when it is reported to block. Once ready again send the ready state back to `!gen`. N%indefinite wait not supported anymoreTtimeoutrr) ValueErrornextrregisterselectrsend READY_NONE StopIterationvalue) r"rr$sselrlistreadyrrvrrr wait_selector8s*      r6 PQGenConn[RV]c Cs|durtdzCt|\}}t/}||| |j|d}s(|tq|||dd}||\}}|||q1sDwYWdStyb}z |j }|WYd}~Sd}~ww)a Wait for a connection generator using the best strategy available. :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&Tr'rr) r)r*rr+r,r-r. unregisterr/r0) r"r$rr1r2r3r4rr5rrr wait_conn\s*        r9c s\|dur tdtt}d fdd }znt|} |t@}|t@}|s1|s1td |d |r@| ||t |rI| ||t z&z t |IdHWn ty_YnwW|rh|||ro||n|rw|||r||ww|}qty}ztd |d}~wty}z |j} | WYd}~Sd}~ww)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&staterrrcs|OdSr setr:evr4rrwakeups zwait_async..wakeupTbad poll status: rrr:rrr)r)rrr*WAIT_RWAIT_Wr InternalErrorclear add_readerREADY_R add_writerREADY_Wrwaitr remove_reader remove_writerr-OSErrorrr/r0) r"rr$loopr@r1readerwriterrr5rr>r wait_async~s\        rRc sT|dur tdtt}d fdd }z{t|\}} |t@}|t@}|s3|s3td |d |rB| ||t |rK| ||t z/|rdz t |IdHWntycYnwIdHW|rs|||rz||n|r|||r||ww|\}}qty}z |j} | WYd}~Sd}~ww) 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&r:rrrcs|dSr r;r=r>rrr@s zwait_conn_async..wakeupTrArrB)r)rrr*rCrDrrErFrGrHrIrJrrKrrLrMr-r/r0) r"r$rOr@rr1rPrQrr5rr>rwait_conn_asyncsZ       rSc Cs|durtdz>t|}d}|f} t|t@r|n||t@r"|n|||\}}}|r5t|tdd} |r;t} |rA| t O} | | }qt yX} ztd| d} ~ wt yn} z | j } | WYd} ~ Sd} ~ ww)z Wait for a generator using select where supported. BUG: on Linux, can't select on FD >= 1024. On Windows it's fine. Nr&rTrr)r)r*r,rCrDrrrrHrJr-rNr/r0) r"rr$r1emptyfnlistrlwlxlr4rr5rrr wait_selects>    rY EpollSelectorc Cs|durtdz^t|}|dkrd}tE}t|}||| ||}s4t||t q"|dd}d}|tj @rCt }|tj @rL|t O}||}t|}|||q#1s_wYWdSty}} z | j} | WYd} ~ Sd} ~ ww)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!Tr)r)r*r,epoll_epoll_evmasksr+pollrr-r.EPOLLINrHEPOLLOUTrJmodifyr/r0) r"rr$r1r[evmaskfileevsr?r4rr5rrr wait_epoll-s<        rc PollSelectorc Cs|durtdzdt|}|dkrd}nt|d}t}t|}||| ||}s6|tq(|dd}d}|tj @rEt }|tj @rN|t O}|s]|t @r]t|td||}t|}|||q)ty} z | j} | WYd} ~ Sd} ~ ww)za Wait for a generator using poll where supported. Parameters are like for `wait()`. Nr&rg@@Trr)r)r*rr,r] _poll_evmasksr+r-r.POLLINrHPOLLOUTrJPOLL_BADrrrr`r/r0) r"rr$r1r]rarbr?r4rr5rrr wait_pollgs@          riboolc Cs^tjd}r-z |drWdSWdSty,}z td|WYd}~dSd}~wwdS)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)mrrrr_is_select_patcheds  rrr rKPSYCOPG_WAIT_FUNCwait_zHPSYCOPG_WAIT_FUNC should be the name of an available wait function; got SelectSelector)rrrr)r!)r"r#rrr$r%rr )r"r7r$r%rr )rrj)E__doc__ __future__rrrkr,logging selectorsasynciorrrrrr rabcr r r r _enumsrr_cmodulerRrCWrDRWWAIT_RWNONEr.rHrJREADY_RW getLogger__name__roplatformrr6r9rRrSrYhasattr EPOLLONESHOTr^r_r\rcrfrgrerhrirrwait_c__annotations__environfname startswithglobals ImportErrorrKgetattrrrrrs           $ " = @ (    /   .