#2041 - Patch celery for ddtrace

department-of-veterans-affairs · Nov 19, 2024 · 107d306 · 107d306
1 parent 335c349
commit 107d306
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/app/celery/celery.py b/app/celery/celery.py
@@ -1,9 +1,14 @@
 import time
 
-from celery import Celery, Task
+import celery
+from celery import Task
 from celery.signals import worker_process_shutdown, worker_shutting_down, worker_process_init
 from flask import current_app
 
+from ddtrace import patch
+
+patch(celery=True)
+
 
 @worker_process_init.connect
 def pool_worker_started(
@@ -78,7 +83,7 @@ def __call__(
     return NotifyTask
 
 
-class NotifyCelery(Celery):
+class NotifyCelery(celery.Celery):
     def init_app(
         self,
         app,