fix aio pika

helios · Jun 8, 2023 · 3939d17 · 3939d17
1 parent a679754
commit 3939d17
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...metry-instrumentation-aio-pika/src/opentelemetry/instrumentation/aio_pika/span_builder.py b/...metry-instrumentation-aio-pika/src/opentelemetry/instrumentation/aio_pika/span_builder.py
@@ -49,6 +49,10 @@ def set_destination(self, destination: str):
         self._attributes[SpanAttributes.MESSAGING_DESTINATION] = destination
 
     def set_channel(self, channel: AbstractChannel):
+        if channel.__class__.__name__ == "RobustChannel":
+            # Change introduced in v9 of aio-pika
+            channel = channel.channel
+
         connection = channel.connection
         if getattr(connection, "connection", None):
             # aio_rmq 7