From 0319dead604c4cfa852f21f24e2fa390e4d9e808 Mon Sep 17 00:00:00 2001 From: Peter Giacomo Lombardo Date: Tue, 26 Feb 2019 10:25:06 +0100 Subject: [PATCH] Mark instrumented functions as coroutines --- instana/instrumentation/asynqp.py | 3 +++ 1 file changed, 3 insertions(+) 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):