Skip to content

Commit 3bf0292

Browse files
author
Andrei Neagu
committed
extend interface
1 parent 65ee01c commit 3bf0292

File tree

2 files changed

+17
-3
lines changed

2 files changed

+17
-3
lines changed

services/dynamic-scheduler/src/simcore_service_dynamic_scheduler/api/rpc/_services.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,8 @@
2323
async def list_tracked_dynamic_services(
2424
app: FastAPI, *, user_id: UserID | None = None, project_id: ProjectID | None = None
2525
) -> list[DynamicServiceGet]:
26-
director_v2_client = DirectorV2Client.get_from_app_state(app)
27-
return await director_v2_client.list_tracked_dynamic_services(
28-
user_id=user_id, project_id=project_id
26+
return await scheduler_interface.list_tracked_dynamic_services(
27+
app, user_id=user_id, project_id=project_id
2928
)
3029

3130

services/dynamic-scheduler/src/simcore_service_dynamic_scheduler/services/scheduler_interface.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,28 @@
55
DynamicServiceStop,
66
)
77
from models_library.api_schemas_webserver.projects_nodes import NodeGet, NodeGetIdle
8+
from models_library.projects import ProjectID
89
from models_library.projects_nodes_io import NodeID
10+
from models_library.users import UserID
911

1012
from ..core.settings import ApplicationSettings
1113
from .director_v2 import DirectorV2Client
1214
from .service_tracker import set_request_as_running, set_request_as_stopped
1315

1416

17+
async def list_tracked_dynamic_services(
18+
app: FastAPI, *, user_id: UserID | None = None, project_id: ProjectID | None = None
19+
) -> list[DynamicServiceGet]:
20+
settings: ApplicationSettings = app.state.settings
21+
if settings.DYNAMIC_SCHEDULER_USE_INTERNAL_SCHEDULER:
22+
raise NotImplementedError
23+
24+
director_v2_client = DirectorV2Client.get_from_app_state(app)
25+
return await director_v2_client.list_tracked_dynamic_services(
26+
user_id=user_id, project_id=project_id
27+
)
28+
29+
1530
async def get_service_status(
1631
app: FastAPI, *, node_id: NodeID
1732
) -> NodeGet | DynamicServiceGet | NodeGetIdle:

0 commit comments

Comments
 (0)