Skip to content

Commit 4c88efb

Browse files
rename
1 parent 9d0e894 commit 4c88efb

File tree

1 file changed

+8
-6
lines changed
  • services/storage/src/simcore_service_storage/modules/celery

1 file changed

+8
-6
lines changed

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

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
class CeleryTaskClient:
3636
_celery_app: Celery
3737
_celery_settings: CelerySettings
38-
_task_store: TaskInfoStore
38+
_task_info_store: TaskInfoStore
3939

4040
async def submit_task(
4141
self,
@@ -63,7 +63,9 @@ async def submit_task(
6363
if task_metadata.ephemeral
6464
else self._celery_settings.CELERY_RESULT_EXPIRES
6565
)
66-
await self._task_store.create_task(task_id, task_metadata, expiry=expiry)
66+
await self._task_info_store.create_task(
67+
task_id, task_metadata, expiry=expiry
68+
)
6769
return task_uuid
6870

6971
@make_async()
@@ -96,9 +98,9 @@ async def get_task_result(
9698
async_result = self._celery_app.AsyncResult(task_id)
9799
result = async_result.result
98100
if async_result.ready():
99-
task_metadata = await self._task_store.get_task_metadata(task_id)
101+
task_metadata = await self._task_info_store.get_task_metadata(task_id)
100102
if task_metadata is not None and task_metadata.ephemeral:
101-
await self._task_store.remove_task(task_id)
103+
await self._task_info_store.remove_task(task_id)
102104
await self._forget_task(task_id)
103105
return result
104106

@@ -107,7 +109,7 @@ async def _get_task_progress_report(
107109
) -> ProgressReport:
108110
if task_state in (TaskState.STARTED, TaskState.RETRY, TaskState.ABORTED):
109111
task_id = build_task_id(task_context, task_uuid)
110-
progress = await self._task_store.get_task_progress(task_id)
112+
progress = await self._task_info_store.get_task_progress(task_id)
111113
if progress is not None:
112114
return progress
113115
if task_state in (
@@ -153,4 +155,4 @@ async def list_tasks(self, task_context: TaskContext) -> list[Task]:
153155
logging.DEBUG,
154156
msg=f"Listing tasks: {task_context=}",
155157
):
156-
return await self._task_store.list_tasks(task_context)
158+
return await self._task_info_store.list_tasks(task_context)

0 commit comments

Comments
 (0)