o `šj ã@snddlZddlZddlmZddlmZddlmZGdd„deƒZ Gdd„deƒZ e  d gd ¢¡Z d e _ dS) éN)Úerrors)ÚFuture)ÚTimercsBeZdZdZ‡fdd„Z‡fdd„Z‡fdd„Zd d d „Z‡ZS) ÚFutureProduceResult)Útopic_partitionÚ_latchcstƒ ¡||_t ¡|_dS©N)ÚsuperÚ__init__rÚ threadingÚEventr)Úselfr©Ú __class__©úS/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/producer/future.pyr s zFutureProduceResult.__init__cótƒ |¡}|j ¡|Sr)r ÚsuccessrÚset)r ÚvalueÚretrrrró  zFutureProduceResult.successcrr)r Úfailurerr)r ÚerrorrrrrrrzFutureProduceResult.failureNcCs|j |¡p |j ¡Sr)rÚwaitÚis_set)r ÚtimeoutrrrrszFutureProduceResult.waitr) Ú__name__Ú __module__Ú __qualname__Ú __slots__r rrrÚ __classcell__rrrrr s    rcs>eZdZdZdZ‡fdd„Zdd„Zdd„Zd d d „Z‡Z S) ÚFutureRecordMetadataaÊAn asynchronous handle to the result of a single :meth:`~kafka.KafkaProducer.send`. :meth:`~kafka.KafkaProducer.send` returns one of these immediately, before the record has been transmitted to the broker. Call :meth:`get` to block until the record is acknowledged and obtain its :class:`RecordMetadata`, or register callbacks via :meth:`~kafka.future.Future.add_callback` / :meth:`~kafka.future.Future.add_errback` to be notified without blocking. The future resolves successfully once the containing batch is acknowledged according to the producer's ``acks`` configuration, or fails with the relevant exception (for example :class:`~kafka.errors.KafkaTimeoutError`). )Ú_produce_futureÚargscs6tƒ ¡||_||||||f|_| |j|j¡dSr)r r r#r$Ú _add_cb_ebÚ_produce_successr)r Úproduce_futureÚ batch_indexÚ timestamp_msÚchecksumÚserialized_key_sizeÚserialized_value_sizeÚserialized_header_sizerrrr /s zFutureRecordMetadata.__init__c Cs˜|\}}}|j\}}}}} } |dur| ||ƒ¡dS|dur%|dkr%|}|dkr1|dur1||7}|jj} t| d| d| ||||| | ƒ } | | ¡dS)Néÿÿÿÿré)r$rr#rÚRecordMetadatar) r ÚresultÚoffsetÚproduce_timestamp_msÚrecord_exceptions_fnr(r)r*r+r,r-ÚtpÚmetadatarrrr&6s ÿþz%FutureRecordMetadata._produce_successc CsN|j}||_|j\}}}}}} |||||| f|_| |j|j¡|j ¡dS)aÎRebind this future to a new produce future with a new batch index. Used when a batch is split due to MESSAGE_TOO_LARGE. The original FutureRecordMetadata is rebound to the new (smaller) batch's future. This must be called from the sender thread while the old produce_future has not been completed. Any user thread blocked in get() on the old produce_future's latch will be woken and will re-wait on the new one. N)r#r$r%r&rrr) r Únew_produce_futureÚnew_batch_indexÚold_produce_futureÚ_r)r*ÚskÚsvÚshrrrÚrebindLs  zFutureRecordMetadata.rebindNcCsbt|dur |dndƒ}|js'|js'|j |j¡s!t d|f¡‚|js'|jr| ¡r.|j ‚|j S)z6Wait for up to timeout seconds for future to complete.Nièz"Timeout after waiting for %s secs.) rÚis_doneÚexpiredr#rÚ timeout_secsÚErrorsÚKafkaTimeoutErrorÚfailedÚ exceptionr)r rÚtimerrrrÚget`s ÿ þzFutureRecordMetadata.getr) rrrÚ__doc__r r r&r>rGr!rrrrr" s r"r0) ÚtopicÚ partitionrr2Ú timestampr*r+r,r-a4Metadata about a record that has been acknowledged by the broker. Returned by :meth:`FutureRecordMetadata.get`, which resolves once the batch containing the record has been acknowledged according to the producer's ``acks`` configuration. Keyword Arguments: topic (str): The topic the record was appended to. partition (int): The partition the record was appended to. topic_partition (TopicPartition): The ``(topic, partition)`` the record was appended to. offset (int): The offset of the record in the topic partition, or -1 if the broker did not assign one (e.g. ``acks=0``). timestamp (int): The timestamp of the record, in milliseconds since the epoch (UTC). For CreateTime this is the producer-supplied timestamp; for LogAppendTime it is the broker-assigned timestamp. checksum (int): Deprecated. The CRC32 checksum of the record, or None. Removed in message format v2 (Kafka 0.11+). serialized_key_size (int): The size of the serialized, uncompressed key in bytes, or -1 if the key is None. serialized_value_size (int): The size of the serialized, uncompressed value in bytes, or -1 if the value is None. serialized_header_size (int): The size of the serialized, uncompressed headers in bytes, or -1 if there are no headers. )Ú collectionsr ÚkafkarrBÚ kafka.futurerÚ kafka.utilrrr"Ú namedtupler0rHrrrrÚs   Pÿ