fix

Author: matusdrobuliak66

packages/postgres-database/src/simcore_postgres_database/models/comp_runs_collections.py

@@ -35,4 +35,7 @@
         "ix_comp_runs_collections_client_or_system_generated_id",
         "client_or_system_generated_id",
     ),
+    sa.UniqueConstraint(
+        "client_or_system_generated_id", name="client_or_system_generated_id_uniqueness"
+    ),
 )

services/web/server/src/simcore_service_webserver/director_v2/_comp_runs_collections_repository.py

@@ -5,6 +5,7 @@
 from pydantic import TypeAdapter
 from simcore_postgres_database.models.comp_runs_collections import comp_runs_collections
 from sqlalchemy import func
+from sqlalchemy.dialects.postgresql import insert as pg_insert
 
 from ._comp_runs_collections_models import CompRunCollectionDBGet
 
@@ -63,3 +64,32 @@ async def get_comp_run_collection_or_none_by_client_generated_id(
     if row is None:
         return None
     return CompRunCollectionDBGet.model_validate(row)
+
+
+async def upsert_comp_run_collection(
+    conn,
+    client_or_system_generated_id: str,
+    client_or_system_generated_display_name: str,
+    is_generated_by_system: bool,
+) -> CollectionRunID:
+    """Upsert a computational run collection. If it exists, only update the modified time."""
+    insert_stmt = pg_insert(comp_runs_collections).values(
+        client_or_system_generated_id=client_or_system_generated_id,
+        client_or_system_generated_display_name=client_or_system_generated_display_name,
+        is_generated_by_system=is_generated_by_system,
+        created=func.now(),
+        modified=func.now(),
+    )
+    on_update_stmt = insert_stmt.on_conflict_do_update(
+        index_elements=[
+            comp_runs_collections.c.client_or_system_generated_display_name
+        ],
+        set_={
+            "modified": func.now(),
+        },
+    )
+    result = await conn.stream(
+        on_update_stmt.returning(comp_runs_collections.c.collection_run_id)
+    )
+    collection_id_tuple: tuple[UUID] = result.one()
+    return TypeAdapter(CollectionRunID).validate_python(collection_id_tuple[0])

services/web/server/src/simcore_service_webserver/director_v2/_comp_runs_collections_service.py

@@ -27,6 +27,22 @@ async def create_comp_run_collection(
         )
 
 
+async def upsert_comp_run_collection(
+    app: web.Application,
+    *,
+    client_or_system_generated_id: str,
+    client_or_system_generated_display_name: str,
+    is_generated_by_system: bool,
+) -> CollectionRunID:
+    async with transaction_context(get_asyncpg_engine(app)) as conn:
+        return await _comp_runs_collections_repository.upsert_comp_run_collection(
+            conn=conn,
+            client_or_system_generated_id=client_or_system_generated_id,
+            client_or_system_generated_display_name=client_or_system_generated_display_name,
+            is_generated_by_system=is_generated_by_system,
+        )
+
+
 async def get_comp_run_collection_or_none_by_id(
     app: web.Application, *, collection_run_id: CollectionRunID
 ) -> CompRunCollectionDBGet | None:

services/web/server/src/simcore_service_webserver/director_v2/_controller/rest.py

@@ -88,7 +88,7 @@ async def start_computation(request: web.Request) -> web.Response:
     comp_run_collection: CompRunCollectionDBGet | None = None
     if group_id_or_none:
         comp_run_collection = await _comp_runs_collections_service.get_comp_run_collection_or_none_by_client_generated_id(
-            request.app, client_or_system_generated_id=group_id_or_none  # type: ignore
+            request.app, client_or_system_generated_id=str(group_id_or_none)
         )
     if comp_run_collection is not None:
         created_at: datetime = comp_run_collection.created
@@ -111,10 +111,10 @@ async def start_computation(request: web.Request) -> web.Response:
         client_or_system_generated_id = f"{group_id_or_none}"
     group_name = custom_metadata.get("group_name", "No Group Name")
 
-    collection_run_id = await _comp_runs_collections_service.create_comp_run_collection(
+    collection_run_id = await _comp_runs_collections_service.upsert_comp_run_collection(
         request.app,
         client_or_system_generated_id=client_or_system_generated_id,
-        client_or_system_generated_display_name=group_name,  # type: ignore
+        client_or_system_generated_display_name=str(group_name),
         is_generated_by_system=is_generated_by_system,
     )