attempting to get celery to show up in xray (#2244)

cds-snc · Aug 2, 2024 · 8a5f959 · 8a5f959
1 parent bc1086e
commit 8a5f959
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/run_celery.py b/run_celery.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python
 import newrelic.agent  # See https://bit.ly/2xBVKBH
+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,4 +14,8 @@
 
 application = Flask("celery")
 create_app(application)
+
+xray_recorder.configure(service='celery')
+XRayMiddleware(application, xray_recorder)
+
 application.app_context().push()