feat: Capture spans for Celery's apply_async

dcramer · dcramer · commit bb4c1ade6572 · 2019-08-07T12:19:09.000-07:00
diff --git a/sentry_sdk/integrations/celery.py b/sentry_sdk/integrations/celery.py
@@ -77,7 +77,12 @@ def apply_async(*args, **kwargs):
                 headers[key] = value
             if headers is not None:
                 kwargs["headers"] = headers
-        return f(*args, **kwargs)
+
+                from sentry_sdk import Hub
+            with hub.span(op="celery", description=task.name):
+                return f(*args, **kwargs)
+        else:
+            return f(*args, **kwargs)
 
     return apply_async
 
