From 8a5f959084c1c67fe7775492b44131a6fc7b1716 Mon Sep 17 00:00:00 2001
From: Mike Pond <32133001+P0NDER0SA@users.noreply.github.com>
Date: Fri, 2 Aug 2024 10:25:15 -0400
Subject: [PATCH] attempting to get celery to show up in xray (#2244)

---
 run_celery.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/run_celery.py b/run_celery.py
index e43328ae3f..3a5cf9aa95 100644
--- 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()