Function set_default_cluster

gpustack/routes/clusters.py:625–665

(session: SessionDep, ctx: TenantContextDep, id: int)

@router.post("/{id}/set-default")
async def set_default_cluster(session: SessionDep, ctx: TenantContextDep, id: int):
    # "Default cluster" is a per-Org concept now: each Org has at most
    # one default, and that's what its members' deploy form falls back
    # to. Writing it follows the standard cluster-write rule (admin
    # always; Org owner only on their own Org's clusters).
    cluster = await Cluster.one_by_id(session, id)
    if not cluster:
        raise NotFoundException(message=f"cluster {id} not found")
    assert_cluster_writable(ctx, cluster)

    try:
        # Unset any existing default in this cluster's Org. Postgres
        # holds this to one via a partial unique index; MySQL/OceanBase
        # have no partial-index equivalent, so we tolerate (and clear)
        # any duplicates instead of relying on the DB.
        existing_defaults = await Cluster.all_by_fields(
            session,
            {
                'is_default': True,
                'deleted_at': None,
                'owner_principal_id': cluster.owner_principal_id,
            },
        )
        for dc in existing_defaults:
            if dc.id != cluster.id:
                await dc.update(
                    session=session,
                    source={"is_default": False},
                    auto_commit=False,
                )
        # Set this cluster as the Org's default.
        await cluster.update(
            session=session,
            source={"is_default": True},
            auto_commit=False,
        )
        await session.commit()
    except Exception as e:
        raise InternalServerErrorException(
            message=f"Failed to set default cluster: {e}"
        )


@router.post("/{id}/worker-pools", response_model=WorkerPoolPublic)