o `j@@s@ddlZddlZddlZddlmZeeZGdddZdS)N)RetriableErrorc@steZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZdS)Future)is_donevalue exception _callbacks _errbacks_lockFcCs,d|_d|_d|_g|_g|_t|_dS)NF)rrrrr threadingLockr selfrJ/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/future.py__init__s zFuture.__init__cCs|jo|jduSNrrr rrr succeededzFuture.succeededcCs|jo|jduSrrr rrrfailedrz Future.failedcCs|jot|jtSr)r isinstancerrr rrr retriableszFuture.retriablec Cs|j}|||_d|_|j}d|_d|_||rE|j}|D]#}z||Wq!tyD}zt d|r:|WYd}~q!d}~ww|S)NTError processing callback) r acquirerrrrreleaseerror_on_callbacks Exceptionlogr)r rlock callbacksrferrrsuccesss*   zFuture.successc Cst|tur|n|}t|tstd|j}|||_d|_|j}d|_ d|_| |rY|j }|D]#}z||Wq5t yX}zt d|rN|WYd}~q5d}~ww|S)Nz"future failed without an exceptionTError processing errback)typer BaseException TypeErrorr rrrrrrrrr)r r!rrerrbacksrr errrrrfailure:s0    zFuture.failurec Os|s|rtj|g|Ri|}|j}||js&|j|||S||jdurUz||j W|St yT}zt d|j rI|WYd}~|Sd}~ww|S)Nr) functoolspartialr rrrappendrrrrrrr r argskwargsrr!rrr add_callbackQ,     zFuture.add_callbackc Os|s|rtj|g|Ri|}|j}||js&|j|||S||jdurUz||jW|St yT}zt d|j rI|WYd}~|Sd}~ww|S)Nr#) r*r+r rrrr,rrrrrr-rrr add_errbackdr1zFuture.add_errbackc Cs|j}||js|j||j|||S||jdurKz||jW|St yJ}zt d|j r?|WYd}~|Sd}~wwz||jW|St yp}zt d|j re|WYd}~|Sd}~ww)aRegister a (callback, errback) pair under a single lock acquire. Fast path for call sites that always register both a plain callback and errback with no ``*args``/``**kwargs``. Used on the producer hot path (``FutureRecordMetadata`` -> ``FutureProduceResult``) to halve the per-record lock-acquire count vs. calling ``add_callback()`` + ``add_errback()`` separately. Nrr#) r rrrr,rrrrrrr)r cbebrr!rrr _add_cb_ebws>            zFuture._add_cb_ebcOs4|j|g|Ri||j|g|Ri||Sr)r0r2)r r r.r/rrradd_bothszFuture.add_bothcCs||j|j|Sr)r5r"r))r futurerrrchainsz Future.chainccs |js|V|jr |j|jSr)rrrr rrr __await__s zFuture.__await__N)__name__ __module__ __qualname__ __slots__rrrrrr"r)r0r2r5r6r8r9rrrrr s! r) r*loggingr kafka.errorsr getLoggerr:rrrrrrs