Skip to content

BatchKafkaTelemetrySettingsProvider

faststream.kafka.opentelemetry.provider.BatchKafkaTelemetrySettingsProvider #

BatchKafkaTelemetrySettingsProvider()

Bases: BaseKafkaTelemetrySettingsProvider[tuple['ConsumerRecord', ...]]

Source code in faststream/kafka/opentelemetry/provider.py 
23
24
def __init__(self) -> None:
    self.messaging_system = "kafka"

messaging_system instance-attribute #

messaging_system = 'kafka'

get_consume_attrs_from_message #

get_consume_attrs_from_message(msg)
Source code in faststream/kafka/opentelemetry/provider.py 
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
def get_consume_attrs_from_message(
    self,
    msg: "StreamMessage[tuple[ConsumerRecord, ...]]",
) -> dict[str, Any]:
    raw_message = msg.raw_message[0]

    return {
        SpanAttributes.MESSAGING_SYSTEM: self.messaging_system,
        SpanAttributes.MESSAGING_MESSAGE_ID: msg.message_id,
        SpanAttributes.MESSAGING_MESSAGE_CONVERSATION_ID: msg.correlation_id,
        SpanAttributes.MESSAGING_MESSAGE_PAYLOAD_SIZE_BYTES: len(
            bytearray().join(cast("Sequence[bytes]", msg.body)),
        ),
        SpanAttributes.MESSAGING_BATCH_MESSAGE_COUNT: len(msg.raw_message),
        SpanAttributes.MESSAGING_KAFKA_DESTINATION_PARTITION: raw_message.partition,
        MESSAGING_DESTINATION_PUBLISH_NAME: raw_message.topic,
    }

get_consume_destination_name #

get_consume_destination_name(msg)
Source code in faststream/kafka/opentelemetry/provider.py 
101
102
103
104
105
def get_consume_destination_name(
    self,
    msg: "StreamMessage[tuple[ConsumerRecord, ...]]",
) -> str:
    return cast("str", msg.raw_message[0].topic)

get_publish_attrs_from_cmd #

get_publish_attrs_from_cmd(cmd)
Source code in faststream/kafka/opentelemetry/provider.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
def get_publish_attrs_from_cmd(
    self,
    cmd: "KafkaPublishCommand",
) -> dict[str, Any]:
    attrs: dict[str, Any] = {
        SpanAttributes.MESSAGING_SYSTEM: self.messaging_system,
        SpanAttributes.MESSAGING_DESTINATION_NAME: cmd.destination,
        SpanAttributes.MESSAGING_MESSAGE_CONVERSATION_ID: cmd.correlation_id,
    }

    if cmd.partition is not None:
        attrs[SpanAttributes.MESSAGING_KAFKA_DESTINATION_PARTITION] = cmd.partition

    if cmd.key is not None:
        attrs[SpanAttributes.MESSAGING_KAFKA_MESSAGE_KEY] = cmd.key

    return attrs

get_publish_destination_name #

get_publish_destination_name(cmd)
Source code in faststream/kafka/opentelemetry/provider.py 
44
45
46
47
48
def get_publish_destination_name(
    self,
    cmd: "PublishCommand",
) -> str:
    return cmd.destination