Skip to content

Commit 1588ca9

Browse files
committed
fix: Function type has been updated
Signed-off-by: Cagri Yonca <cagri@ibm.com>
1 parent e4ce09e commit 1588ca9

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/instana/instrumentation/aioamqp.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -33,14 +33,14 @@ async def basic_publish_with_instana(
3333
logger.debug(f"aioamqp basic_publish_with_instana error: {exc}")
3434

3535
@wrapt.patch_function_wrapper("aioamqp.channel", "Channel.basic_consume")
36-
def basic_consume_with_instana(
36+
async def basic_consume_with_instana(
3737
wrapped: Callable[..., aioamqp.connect],
3838
instance: object,
3939
argv: Tuple[object, Tuple[object, ...]],
4040
kwargs: Dict[str, Any],
4141
) -> object:
4242
if tracing_is_off():
43-
return wrapped(*argv, **kwargs)
43+
return await wrapped(*argv, **kwargs)
4444

4545
callback = argv[0]
4646
tracer, parent_span, _ = get_tracer_tuple()
@@ -70,7 +70,7 @@ async def callback_wrapper(
7070
wrapped_callback = callback_wrapper(callback)
7171
argv = (wrapped_callback,) + argv[1:]
7272

73-
return wrapped(*argv, **kwargs)
73+
return await wrapped(*argv, **kwargs)
7474

7575
logger.debug("Instrumenting aioamqp")
7676

0 commit comments

Comments
 (0)