docs

Author: sanderegg

services/storage/src/simcore_service_storage/modules/celery/client.py

@@ -61,31 +61,35 @@ def __init__(self, celery_app: Celery):
     def send_task(
         self, task_name: str, *, task_context: TaskContext, **task_params
     ) -> TaskUUID:
-        task_uuid = uuid4()
-        task_id = _build_task_id(task_context, task_uuid)
         with log_context(
             _logger,
             logging.DEBUG,
-            msg=f"Submitting task {task_name}: {task_id=} {task_params=}",
+            msg=f"Submit {task_name=}: {task_context=} {task_params=}",
         ):
+            task_uuid = uuid4()
+            task_id = _build_task_id(task_context, task_uuid)
             self._celery_app.send_task(task_name, task_id=task_id, kwargs=task_params)
             return task_uuid
 
     @staticmethod
     @make_async()
     def abort_task(task_context: TaskContext, task_uuid: TaskUUID) -> None:
-        task_id = _build_task_id(task_context, task_uuid)
-        _logger.info("Aborting task %s", task_id)
-        AbortableAsyncResult(task_id).abort()
+        with log_context(
+            _logger,
+            logging.DEBUG,
+            msg=f"Abort task {task_uuid=}: {task_context=}",
+        ):
+            task_id = _build_task_id(task_context, task_uuid)
+            AbortableAsyncResult(task_id).abort()
 
     @make_async()
     def get_task_result(self, task_context: TaskContext, task_uuid: TaskUUID) -> Any:
-        task_id = _build_task_id(task_context, task_uuid)
         with log_context(
             _logger,
             logging.DEBUG,
-            msg=f"Getting  task {task_id=} result",
+            msg=f"Get task {task_uuid=}: {task_context=} result",
         ):
+            task_id = _build_task_id(task_context, task_uuid)
             return self._celery_app.AsyncResult(task_id).result
 
     def _get_progress_report(
@@ -133,6 +137,7 @@ def _get_completed_task_uuids(self, task_context: TaskContext) -> set[TaskUUID]:
                 for key in keys
             }
         if hasattr(backend_client, "cache"):
+            # NOTE: backend used in testing. It is a dict-like object
             found_keys = set()
             for key in backend_client.cache:
                 str_key = key.decode(_CELERY_TASK_ID_KEY_ENCODING)