o )i@sddlZddlZddlZddlmZddlmZmZmZej dddkr-ddlm Z dSdee dd fd d Z Gd d d ej ZGdd d ZdS)N) TracebackType)AnyOptionalType) )timeoutdelayreturnTimeoutcCs*t}|dur||nd}t||S)atimeout context manager. Useful in cases when you want to apply timeout logic around block of code or in cases when asyncio.wait_for is not suitable. For example: >>> async with timeout(0.001): ... async with aiohttp.get('https://github.com') as r: ... await r.text() delay - value in seconds or None to disable timeout logic N)asyncioget_running_looptimer )r loopdeadlinere/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/engine/async_timeout.pyasyncio_timeouts  rc@seZdZdZdZdZdZdS)_StateINITENTERTIMEOUTEXITN)__name__ __module__ __qualname__rrrrrrrrr s rc @seZdZdZdeedejddfddZd&dd Z d ee e d ee d ee dee fd dZede fddZedeefddZd'ddZd'ddZdeddfddZdeddfddZd'ddZd'ddZd ee e ddfd d!Zd(d$d%ZdS))r ) _deadline_loop_state_timeout_handlerrrr NcCs4||_tj|_d|_|durd|_dS||dSN)rrrrr rupdate)selfrrrrr__init__:s  zTimeout.__init__cs||Sr!) _do_enterr#rrr __aenter__EszTimeout.__aenter__exc_typeexc_valexc_tbcs||dSr!)_do_exit)r#r(r)r*rrr __aexit__Is zTimeout.__aexit__cCs |jtjkS)z$Is timeout expired during execution?)rrrr&rrrexpiredRs zTimeout.expiredcCs|jSr!)rr&rrrrWszTimeout.deadlinecCs0|jtjtjfvrtd|jj|dS)z Reject scheduled timeout if any.invalid state N)rrrr RuntimeErrorvalue_rejectr&rrrreject[s zTimeout.rejectcCs"|jdur|jd|_dSdSr!)r cancelr&rrrr1cs   zTimeout._rejectr cCs(|j}|dur td|||dS)zAdvance timeout on delay seconds. The delay can be negative. Raise RuntimeError if shift is called when deadline is not scheduled Nz1cannot shift timeout if deadline is not scheduled)rr/r")r#r rrrrshifths z Timeout.shiftcCs^|jtjkr td|jtjkrtd|jdur|j||_|jtjkr-| dSdS)aiSet deadline to absolute value. deadline argument points on the time in the same clock system as loop.time(). If new deadline is in the past the timeout is raised immediately. Please note: it is not POSIX time but a time with undefined starting base, e.g. the time of the system power on. z1cannot reschedule after exit from context managerz!cannot reschedule expired timeoutN) rrrr/rr r3rr _reschedule)r#rrrrr"ss      zTimeout.updatecCs~|jtjksJ|j}|durdS|j}|jdur |jt }||kr3|j |j ||_dS|j ||j ||_dSr!) rrrrrrr r3r current_task call_soon _on_timeoutcall_at)r#rnowtaskrrrr5s     zTimeout._reschedulecCs2|jtjkrtd|jjtj|_|dS)Nr.)rrrr/r0rr5r&rrrr%s  zTimeout._do_entercCs6|tjur|jtjkrd|_tjtj|_|dSr!) r CancelledErrorrrrr TimeoutErrorrr1)r#r(rrrr+s  zTimeout._do_exitr;Optional[asyncio.Task[Any]]cCs|r|tj|_d|_dSr!)r3rrrr )r#r;rrrr8s zTimeout._on_timeout)r r )r N)r;r>r N)rrr __slots__rfloatr AbstractEventLoopr$r'r BaseExceptionrboolr,propertyr-rr2r1r4r"r5r%r+r8rrrrr &s:          )r enumsystypesrtypingrrr version_infor rr@Enumrr rrrrs