Skip to content

Commit 341e283

Browse files
committed
convert to OwnerMetadata baseclass before submitting rpc requests
1 parent 4ba288c commit 341e283

File tree

2 files changed

+26
-14
lines changed
  • services
    • api-server/src/simcore_service_api_server/services_rpc
    • web/server/src/simcore_service_webserver/tasks

2 files changed

+26
-14
lines changed

services/api-server/src/simcore_service_api_server/services_rpc/storage.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from models_library.api_schemas_webserver.storage import PathToExport
88
from models_library.products import ProductName
99
from models_library.users import UserID
10+
from servicelib.celery.models import OwnerMetadata
1011
from servicelib.rabbitmq._client_rpc import RabbitMQRPCClient
1112
from servicelib.rabbitmq.rpc_interfaces.storage import simcore_s3 as storage_rpc
1213

@@ -33,8 +34,10 @@ async def start_data_export(
3334
self._rpc_client,
3435
paths_to_export=paths_to_export,
3536
export_as="download_link",
36-
owner_metadata=ApiServerOwnerMetadata(
37-
user_id=self._user_id, product_name=self._product_name
37+
owner_metadata=OwnerMetadata.model_validate(
38+
ApiServerOwnerMetadata(
39+
user_id=self._user_id, product_name=self._product_name
40+
).model_dump()
3841
),
3942
user_id=self._user_id,
4043
)

services/web/server/src/simcore_service_webserver/tasks/_rest.py

Lines changed: 21 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
parse_request_path_parameters_as,
2828
)
2929
from servicelib.aiohttp.rest_responses import create_data_response
30+
from servicelib.celery.models import OwnerMetadata
3031
from servicelib.long_running_tasks import lrt_api
3132
from servicelib.rabbitmq.rpc_interfaces.async_jobs import async_jobs
3233

@@ -69,9 +70,11 @@ async def get_async_jobs(request: web.Request) -> web.Response:
6970
user_async_jobs = await async_jobs.list_jobs(
7071
rabbitmq_rpc_client=rabbitmq_rpc_client,
7172
rpc_namespace=STORAGE_RPC_NAMESPACE,
72-
owner_metadata=WebServerOwnerMetadata(
73-
user_id=_req_ctx.user_id,
74-
product_name=_req_ctx.product_name,
73+
owner_metadata=OwnerMetadata.model_validate(
74+
WebServerOwnerMetadata(
75+
user_id=_req_ctx.user_id,
76+
product_name=_req_ctx.product_name,
77+
).model_dump()
7578
),
7679
)
7780
return create_data_response(
@@ -118,9 +121,11 @@ async def get_async_job_status(request: web.Request) -> web.Response:
118121
rabbitmq_rpc_client=rabbitmq_rpc_client,
119122
rpc_namespace=STORAGE_RPC_NAMESPACE,
120123
job_id=async_job_get.task_id,
121-
owner_metadata=WebServerOwnerMetadata(
122-
user_id=_req_ctx.user_id,
123-
product_name=_req_ctx.product_name,
124+
owner_metadata=OwnerMetadata.model_validate(
125+
WebServerOwnerMetadata(
126+
user_id=_req_ctx.user_id,
127+
product_name=_req_ctx.product_name,
128+
).model_dump()
124129
),
125130
)
126131
_task_id = f"{async_job_rpc_status.job_id}"
@@ -154,9 +159,11 @@ async def cancel_async_job(request: web.Request) -> web.Response:
154159
rabbitmq_rpc_client=rabbitmq_rpc_client,
155160
rpc_namespace=STORAGE_RPC_NAMESPACE,
156161
job_id=async_job_get.task_id,
157-
owner_metadata=WebServerOwnerMetadata(
158-
user_id=_req_ctx.user_id,
159-
product_name=_req_ctx.product_name,
162+
owner_metadata=OwnerMetadata.model_validate(
163+
WebServerOwnerMetadata(
164+
user_id=_req_ctx.user_id,
165+
product_name=_req_ctx.product_name,
166+
).model_dump()
160167
),
161168
)
162169

@@ -182,9 +189,11 @@ class _PathParams(BaseModel):
182189
rabbitmq_rpc_client=rabbitmq_rpc_client,
183190
rpc_namespace=STORAGE_RPC_NAMESPACE,
184191
job_id=async_job_get.task_id,
185-
owner_metadata=WebServerOwnerMetadata(
186-
user_id=_req_ctx.user_id,
187-
product_name=_req_ctx.product_name,
192+
owner_metadata=OwnerMetadata.model_validate(
193+
WebServerOwnerMetadata(
194+
user_id=_req_ctx.user_id,
195+
product_name=_req_ctx.product_name,
196+
).model_dump()
188197
),
189198
)
190199

0 commit comments

Comments
 (0)