Skip to content

Commit 780d34c

Browse files
committed
Add example illustrating the use of the celery signal handler
1 parent b9cde7a commit 780d34c

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

  • examples/server_with_celery/example_app
Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,8 @@
1-
from celery import Celery
2-
from flask_log_request_id.extras.celery import RequestIDAwareTask
1+
from celery import Celery, signals
2+
from flask_log_request_id.extras.celery import add_request_id_header, RequestIDAwareTask
33

4+
# Either sublcass all tasks from RequestIDAwareTask
45
celery = Celery(task_cls=RequestIDAwareTask)
6+
7+
# Or alternatively register the provided signal handler
8+
signals.before_task_publish.connect(add_request_id_header)

0 commit comments

Comments
 (0)