Skip to content
This repository was archived by the owner on Jun 13, 2023. It is now read-only.

Commit d58e848

Browse files
authored
fix(celery.py): fix celery trace creation (#346)
1 parent f5b048e commit d58e848

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

epsagon/events/celery.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def wrap_prerun(*args, **kwargs):
160160
"""
161161
event_key = get_event_key(*args, **kwargs)
162162
if event_key:
163-
trace_factory.prepare()
163+
trace_factory.get_or_create_trace(unique_id=event_key)
164164
runner = CeleryRunner(*args, **kwargs)
165165
ACTIVE_EVENTS[event_key] = runner
166166
trace_factory.set_runner(runner)

epsagon/runners/celery.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(self, *args, **kwargs): # pylint: disable=unused-argument
4545
self.resource['metadata'].update({
4646
'id': task_id,
4747
'state': state,
48-
'hostname': app_conn.hostname,
48+
'hostname': app_conn.hostname or 'localhost',
4949
'virtual_host': app_conn.virtual_host,
5050
'driver': app_conn.transport.driver_type,
5151
})

0 commit comments

Comments
 (0)