o `j @s&ddlZddlmZGdddZdS)N)Futurec@s2eZdZdZddZddZd ddZd d ZdS) WakeupNotifieraawait wakeup(timeout_secs) when either ``timeout_secs`` elapses or notify() is called -- whichever first. The notifier is safe to call from any thread (it routes through call_soon_threadsafe). Level-triggered: notify() arriving while no one is awaiting is latched and consumed by the next ``__call__``. This closes a lost-wakeup race where a caller's state mutation (e.g. ``cluster._need_update = True``) and its ``notify()`` happen between another task's pre-await state check and its ``await self._wakeup(...)``. Without latching, the notification arrives at the IO thread before the task has registered a future to signal, and the task would sleep for the full timeout despite work being ready. Used by the metadata refresh loop to sleep on its TTL while remaining interruptible by external callers (e.g. KafkaProducer / KafkaConsumer invoking cluster.request_update() from another thread). cCst||_d|_d|_dS)NF)weakrefproxy_net_fut_pending)selfnetr W/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/net/wakeup_notifier.py__init__s  zWakeupNotifier.__init__cCs,|jdur|jjs|jddSd|_dS)NT)ris_donesuccessrr r r r _wakeup s zWakeupNotifier._wakeupNc s|jdur tdt|_|jrd|_|jd|dur3z |j||j}Wn ty2YdSwd}z|jIdHWd|_|durL|j |dSdSd|_|dur\|j |ww)Nz$Concurrent access to WakeupNotifier!F) r RuntimeErrorrrrr call_laterrReferenceErrorcancel)r timeout_secstimerr r r __call__&s0   zWakeupNotifier.__call__cCs*z |j|jWdStyYdSwN)rcall_soon_threadsaferrrr r r notify;s  zWakeupNotifier.notifyr)__name__ __module__ __qualname____doc__r rrrr r r r rs   r)r kafka.futurerrr r r r s