Merge pull request #52 from dynatrace-oss/aiohttp-client-support

Add initial support for aiohttp client

Author: dlopes7

README.md

@@ -22,6 +22,7 @@ For most technologies, just import it in your code.
 
 ### Technologies supported:
 
+- **aiohttp** (client)
 - **celery**
 - **concurrent.futures**
 - **confluent_kafka**

autodynatrace/__init__.py

@@ -47,6 +47,7 @@
     "tornado": True,
     "fastapi": True,
     "starlette": True,
+    "aiohttp": True,
 }
 
 

autodynatrace/wrappers/aiohttp/__init__.py

@@ -0,0 +1 @@
+from .wrapper import instrument

autodynatrace/wrappers/aiohttp/wrapper.py

@@ -0,0 +1,28 @@
+import aiohttp
+import wrapt
+
+from oneagent.common import DYNATRACE_HTTP_HEADER_NAME
+from ...log import logger
+from ...sdk import sdk
+
+
+def instrument():
+    @wrapt.patch_function_wrapper("aiohttp.client", "ClientSession._request")
+    async def dynatrace_request(wrapped, instance, args, kwargs):
+
+        method = args[0]
+        url = args[1]
+        headers = dict(kwargs.get("headers", {}))
+
+        with sdk.trace_outgoing_web_request(url, method, headers) as tracer:
+            tag = tracer.outgoing_dynatrace_string_tag.decode()
+            logger.debug("dynatrace - tracing {} '{}' with tag '{}'".format(method, url, tag))
+            headers.update({DYNATRACE_HTTP_HEADER_NAME: tag})
+            kwargs["headers"] = headers
+
+            response = await wrapped(*args, **kwargs)
+
+            tracer.set_status_code(response.status)
+            tracer.add_response_headers(dict(response.headers))
+
+            return response

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="autodynatrace",
-    version="1.0.80",
+    version="1.0.81",
     packages=find_packages(),
     package_data={"autodynatrace": ["wrappers/*"]},
     install_requires=["wrapt>=1.11.2", "oneagent-sdk>=1.3.0", "six>=1.10.0", "autowrapt>=1.0"],