Mark instrumented functions as coroutines (#134)

Author: pglombardo

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):