diff --git a/instana/instrumentation/asynqp.py b/instana/instrumentation/asynqp.py index a7e2a564..5faeeefc 100644 --- a/instana/instrumentation/asynqp.py +++ b/instana/instrumentation/asynqp.py @@ -8,6 +8,7 @@ try: import asynqp + import asyncio @wrapt.patch_function_wrapper('asynqp.exchange','Exchange.publish') def publish_with_instana(wrapped, instance, argv, kwargs): @@ -41,6 +42,7 @@ def publish_with_instana(wrapped, instance, argv, kwargs): else: return rv + @asyncio.coroutine @wrapt.patch_function_wrapper('asynqp.queue','Queue.get') def get_with_instana(wrapped, instance, argv, kwargs): parent_span = async_tracer.active_span @@ -63,6 +65,7 @@ def get_with_instana(wrapped, instance, argv, kwargs): return msg + @asyncio.coroutine @wrapt.patch_function_wrapper('asynqp.queue','Queue.consume') def consume_with_instana(wrapped, instance, argv, kwargs): def callback_generator(original_callback):