bootstrapping the xray sdk/middleware (#1923)

* bootstrapping the xray sdk/middleware * fixing formatting * adding blank line * fixing * attempting Custom TEmplate loader * just make a segment * Update notify-dev-entrypoint.sh * format fixes * fix * Trying this with a different initialization order --------- Co-authored-by: Stephen Astels <[email protected]>
cds-snc · Aug 7, 2024 · a6f3abb · a6f3abb
1 parent 01f5548
commit a6f3abb
Show file tree

Hide file tree

Showing 4 changed files with 104 additions and 3 deletions.
diff --git a/application.py b/application.py
@@ -1,6 +1,8 @@
 import os
 
 from apig_wsgi import make_lambda_handler
+from aws_xray_sdk.core import xray_recorder
+from aws_xray_sdk.ext.flask.middleware import XRayMiddleware
 from dotenv import load_dotenv
 from flask import Flask
 
@@ -12,7 +14,8 @@
 
 application = Flask("app")
 application.wsgi_app = ProxyFix(application.wsgi_app)  # type: ignore
-
+xray_recorder.configure(service='admin')
+XRayMiddleware(application, xray_recorder)
 create_app(application)
 
 apig_wsgi_handler = make_lambda_handler(application, binary_support=True)

diff --git a/mypy.ini b/mypy.ini
@@ -59,4 +59,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-newrelic.*]
+ignore_missing_imports = True
+
+[mypy-aws_xray_sdk.*]
 ignore_missing_imports = True
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -63,6 +63,7 @@ xlrd = "1.2.0" # this is pinned so that we can continue to support .xslm files;
 # Putting upgrade on hold due to v1.0.0 using sha512 instead of sha1 by default
 itsdangerous = "2.2.0"  # pyup: <1.0.0
 newrelic = "8.10.0" # Pinned to 8.10.0 for now, 8.11.0 caused a performance regression: https://gcdigital.slack.com/archives/C012W5K734Y/p1709668046344929
+aws-xray-sdk = "^2.14.0"
 
 [tool.poetry.group.test.dependencies]
 isort = "5.13.2"