Skip to content

Commit 92115e2

Browse files
committed
add client name in api-server
1 parent 58c8ac1 commit 92115e2

File tree

1 file changed

+9
-5
lines changed
  • services/api-server/src/simcore_service_api_server/api/routes

1 file changed

+9
-5
lines changed

services/api-server/src/simcore_service_api_server/api/routes/tasks.py

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,16 @@
2626
create_route_description,
2727
)
2828

29+
_ASYNC_JOB_CLIENT_NAME = "API_SERVER"
30+
2931
router = APIRouter()
3032
_logger = logging.getLogger(__name__)
3133

3234

3335
def _get_job_id_data(user_id: UserID, product_name: ProductName) -> AsyncJobFilter:
34-
return AsyncJobFilter(user_id=user_id, product_name=product_name)
36+
return AsyncJobFilter(
37+
user_id=user_id, product_name=product_name, client_name=_ASYNC_JOB_CLIENT_NAME
38+
)
3539

3640

3741
_DEFAULT_TASK_STATUS_CODES: dict[int | str, dict[str, Any]] = {
@@ -61,7 +65,7 @@ async def list_tasks(
6165
async_jobs: Annotated[AsyncJobClient, Depends(get_async_jobs_client)],
6266
):
6367
user_async_jobs = await async_jobs.list_jobs(
64-
job_id_data=_get_job_id_data(user_id, product_name),
68+
job_filter=_get_job_id_data(user_id, product_name),
6569
filter_="",
6670
)
6771
app_router = app.router
@@ -102,7 +106,7 @@ async def get_task_status(
102106
):
103107
async_job_rpc_status = await async_jobs.status(
104108
job_id=task_id,
105-
job_id_data=_get_job_id_data(user_id, product_name),
109+
job_filter=_get_job_id_data(user_id, product_name),
106110
)
107111
_task_id = f"{async_job_rpc_status.job_id}"
108112
return TaskStatus(
@@ -134,7 +138,7 @@ async def cancel_task(
134138
):
135139
await async_jobs.cancel(
136140
job_id=task_id,
137-
job_id_data=_get_job_id_data(user_id, product_name),
141+
job_filter=_get_job_id_data(user_id, product_name),
138142
)
139143

140144

@@ -168,6 +172,6 @@ async def get_task_result(
168172
):
169173
async_job_rpc_result = await async_jobs.result(
170174
job_id=task_id,
171-
job_id_data=_get_job_id_data(user_id, product_name),
175+
job_filter=_get_job_id_data(user_id, product_name),
172176
)
173177
return TaskResult(result=async_job_rpc_result.result, error=None)

0 commit comments

Comments
 (0)