Add queue name to logs (#201)

* Log queue name * Update tasktiger_processor's docstring * Check that queue name is logged * Fix test comment
closeio · Dec 2, 2021 · a090d00 · a090d00
1 parent f3b3e24
commit a090d00
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tasktiger/logging.py b/tasktiger/logging.py
@@ -5,10 +5,12 @@ def tasktiger_processor(logger, method_name, event_dict):
     """
     TaskTiger structlog processor.
 
-    Inject the current task id for non-batch tasks.
+    Inject the current task ID and queue for non-batch tasks.
     """
 
     if g['current_tasks'] is not None and not g['current_task_is_batch']:
-        event_dict['task_id'] = g['current_tasks'][0].id
+        current_task = g['current_tasks'][0]
+        event_dict['task_id'] = current_task.id
+        event_dict['queue'] = current_task.queue
 
     return event_dict
diff --git a/tests/test_logging.py b/tests/test_logging.py
@@ -16,8 +16,9 @@
 
 def logging_task():
     log = logger.info("simple task")
-    # Confirm tasktiger_processor injected task id
+    # Confirm tasktiger_processor injected task id and queue name
     assert log[1]["task_id"] == tiger.current_task.id
+    assert log[1]["queue"] == tiger.current_task.queue
 
 
 class TestLogging(BaseTestCase):