Skip to content

KafkaAckableMessage

faststream.kafka.message.KafkaAckableMessage #

KafkaAckableMessage(*args: Any, **kwargs: Any)

Bases: KafkaMessage

Source code in faststream/kafka/message.py
83
84
85
def __init__(self, *args: Any, **kwargs: Any) -> None:
    super().__init__(*args, **kwargs)
    self.committed = None

committed instance-attribute #

committed = None

raw_message instance-attribute #

raw_message = raw_message

body instance-attribute #

body = body

reply_to instance-attribute #

reply_to = reply_to

content_type instance-attribute #

content_type = content_type

source_type instance-attribute #

source_type = source_type

headers instance-attribute #

headers = headers or {}

batch_headers instance-attribute #

batch_headers = batch_headers or []

path instance-attribute #

path = path or {}

correlation_id instance-attribute #

correlation_id = correlation_id or str(uuid4())

message_id instance-attribute #

message_id = message_id or self.correlation_id

processed instance-attribute #

processed = False

consumer instance-attribute #

consumer = consumer

raw_headers property #

raw_headers: dict[str, bytes | None]

Original header bytes, or the first message's headers for a batch.

batch_raw_headers property #

batch_raw_headers: list[dict[str, bytes | None]]

Original header bytes for every batch message; empty for a single message.

ack async #

ack() -> None
Source code in faststream/kafka/message.py
87
88
89
90
async def ack(self) -> None:
    if not self.committed:
        await self.consumer.commit()
    await super().ack()

reject async #

reject() -> None
Source code in faststream/kafka/message.py
92
93
async def reject(self) -> None:
    await self.ack()

nack async #

nack() -> None
Source code in faststream/kafka/message.py
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
async def nack(self) -> None:
    if not self.committed:
        # A batch may span several partitions: rewind each one to its earliest
        # record, so none of them is committed past an unprocessed message.
        records = (
            self.raw_message
            if isinstance(self.raw_message, tuple)
            else (self.raw_message,)
        )
        earliest: dict[AIOKafkaTopicPartition, int] = {}
        for record in records:
            topic_partition = AIOKafkaTopicPartition(record.topic, record.partition)
            offset = earliest.get(topic_partition)
            if offset is None or record.offset < offset:
                earliest[topic_partition] = record.offset
        for topic_partition, offset in earliest.items():
            self.consumer.seek(partition=topic_partition, offset=offset)
    await super().nack()

set_decoder #

set_decoder(decoder: AsyncCallable) -> None
Source code in faststream/message/message.py
69
70
def set_decoder(self, decoder: "AsyncCallable") -> None:
    self.__decoder = decoder

clear_cache #

clear_cache() -> None
Source code in faststream/message/message.py
72
73
def clear_cache(self) -> None:
    self.__decoded_caches.clear()

decode async #

decode() -> Optional[Any]

Serialize the message by lazy decoder.

Returns a cache after first usage. To prevent such behavior, please call message.clear_cache() after message.body changes.

Source code in faststream/message/message.py
102
103
104
105
106
107
108
109
110
111
112
113
114
115
async def decode(self) -> Optional["Any"]:
    """Serialize the message by lazy decoder.

    Returns a cache after first usage. To prevent such behavior, please call
    `message.clear_cache()` after `message.body` changes.
    """
    assert self.__decoder, "You should call `set_decoder()` method first."

    if (
        result := self.__decoded_caches.get(self.__decoder, _NOT_CACHED)
    ) is _NOT_CACHED:
        result = self.__decoded_caches[self.__decoder] = await self.__decoder(self)

    return result