fix: correct RabbitExchange processing by OTEL in broker.publish case

airtai · Jun 12, 2024 · ddab603 · ddab603
1 parent bb6997f
commit ddab603
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/faststream/rabbit/opentelemetry/provider.py b/faststream/rabbit/opentelemetry/provider.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Union
 
 from opentelemetry.semconv.trace import SpanAttributes
 
@@ -9,6 +9,7 @@
     from aio_pika import IncomingMessage
 
     from faststream.broker.message import StreamMessage
+    from faststream.rabbit.schemas.exchange import RabbitExchange
     from faststream.types import AnyDict
 
 
@@ -44,9 +45,12 @@ def get_publish_attrs_from_kwargs(
         self,
         kwargs: "AnyDict",
     ) -> "AnyDict":
+        exchange: Union[None, str, RabbitExchange] = kwargs.get("exchange")
         return {
             SpanAttributes.MESSAGING_SYSTEM: self.messaging_system,
-            SpanAttributes.MESSAGING_DESTINATION_NAME: kwargs.get("exchange") or "",
+            SpanAttributes.MESSAGING_DESTINATION_NAME: getattr(
+                exchange, "name", exchange or ""
+            ),
             SpanAttributes.MESSAGING_RABBITMQ_DESTINATION_ROUTING_KEY: kwargs[
                 "routing_key"
             ],