(aiohttp_client) Add support for async functions for request and response hooks.

Author: D-Sokol

instrumentation/opentelemetry-instrumentation-aiohttp-client/src/opentelemetry/instrumentation/aiohttp_client/__init__.py

@@ -88,8 +88,10 @@ def response_hook(span: Span, params: typing.Union[
 ---
 """
 
+import inspect
 import types
 import typing
+from collections.abc import Awaitable
 from timeit import default_timer
 from typing import Collection
 from urllib.parse import urlparse
@@ -139,7 +141,10 @@ def response_hook(span: Span, params: typing.Union[
 
 _UrlFilterT = typing.Optional[typing.Callable[[yarl.URL], str]]
 _RequestHookT = typing.Optional[
-    typing.Callable[[Span, aiohttp.TraceRequestStartParams], None]
+    typing.Callable[
+        [Span, aiohttp.TraceRequestStartParams],
+        typing.Union[None, Awaitable[None]],
+    ]
 ]
 _ResponseHookT = typing.Optional[
     typing.Callable[
@@ -150,7 +155,7 @@ def response_hook(span: Span, params: typing.Union[
                 aiohttp.TraceRequestExceptionParams,
             ],
         ],
-        None,
+        typing.Union[None, Awaitable[None]],
     ]
 ]
 
@@ -367,7 +372,9 @@ async def on_request_start(
         )
 
         if callable(request_hook):
-            request_hook(trace_config_ctx.span, params)
+            call = request_hook(trace_config_ctx.span, params)
+            if inspect.isawaitable(call):
+                await call
 
         trace_config_ctx.token = context_api.attach(
             trace.set_span_in_context(trace_config_ctx.span)
@@ -384,7 +391,10 @@ async def on_request_end(
             return
 
         if callable(response_hook):
-            response_hook(trace_config_ctx.span, params)
+            call = response_hook(trace_config_ctx.span, params)
+            if inspect.isawaitable(call):
+                await call
+
         _set_http_status_code_attribute(
             trace_config_ctx.span,
             params.response.status,
@@ -414,7 +424,9 @@ async def on_request_exception(
             trace_config_ctx.span.record_exception(params.exception)
 
         if callable(response_hook):
-            response_hook(trace_config_ctx.span, params)
+            call = response_hook(trace_config_ctx.span, params)
+            if inspect.isawaitable(call):
+                await call
 
         _end_trace(trace_config_ctx)