avoid exception when property name changes

open-telemetry · Dec 20, 2021 · 68f9783 · 68f9783
1 parent fdba606
commit 68f9783
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...elemetry-instrumentation-pika/src/opentelemetry/instrumentation/pika/pika_instrumentor.py b/...elemetry-instrumentation-pika/src/opentelemetry/instrumentation/pika/pika_instrumentor.py
@@ -55,7 +55,9 @@ def _instrument_blocking_channel_consumers(
     ) -> Any:
         for consumer_tag, consumer_info in channel._consumer_infos.items():
             callback_attr = PikaInstrumentor.CONSUMER_CALLBACK_ATTR
-            consumer_callback = getattr(consumer_info, callback_attr)
+            consumer_callback = getattr(consumer_info, callback_attr, None)
+            if consumer_callback is None:
+                continue
             decorated_callback = utils._decorate_callback(
                 consumer_callback,
                 tracer,
@@ -142,7 +144,7 @@ def uninstrument_channel(channel: BlockingChannel) -> None:
 
         for consumers_tag, client_info in channel._consumer_infos.items():
             callback_attr = PikaInstrumentor.CONSUMER_CALLBACK_ATTR
-            consumer_callback = getattr(client_info, callback_attr)
+            consumer_callback = getattr(client_info, callback_attr, None)
             if hasattr(consumer_callback, "_original_callback"):
                 channel._consumer_infos[
                     consumers_tag