From 5a903ea305c529df68773ad537e9562acab70710 Mon Sep 17 00:00:00 2001 From: Jim Yeh Date: Wed, 22 Jan 2020 11:01:45 +0800 Subject: [PATCH] Add celery task_id into scope tags --- sentry_sdk/integrations/celery.py | 2 ++ tests/integrations/celery/test_celery.py | 1 + 2 files changed, 3 insertions(+) diff --git a/sentry_sdk/integrations/celery.py b/sentry_sdk/integrations/celery.py index da0ee5c5e3..38c2452618 100644 --- a/sentry_sdk/integrations/celery.py +++ b/sentry_sdk/integrations/celery.py @@ -169,6 +169,8 @@ def event_processor(event, hint): # type: (Event, Hint) -> Optional[Event] with capture_internal_exceptions(): + tags = event.setdefault("tags", {}) + tags["celery_task_id"] = uuid extra = event.setdefault("extra", {}) extra["celery-job"] = { "task_name": task.name, diff --git a/tests/integrations/celery/test_celery.py b/tests/integrations/celery/test_celery.py index c2b01082ef..956f6869c3 100644 --- a/tests/integrations/celery/test_celery.py +++ b/tests/integrations/celery/test_celery.py @@ -83,6 +83,7 @@ def dummy_task(x, y): assert event["contexts"]["trace"]["trace_id"] == span.trace_id assert event["contexts"]["trace"]["span_id"] != span.span_id assert event["transaction"] == "dummy_task" + assert "celery_task_id" in event["tags"] assert event["extra"]["celery-job"] == dict( task_name="dummy_task", **expected_context )