Handle to_be_deleted gateways in API

Author: r4victor

src/dstack/_internal/server/models.py

@@ -508,6 +508,7 @@ class GatewayModel(PipelineModelMixin, BaseModel):
     status: Mapped[GatewayStatus] = mapped_column(EnumAsString(GatewayStatus, 100))
     status_message: Mapped[Optional[str]] = mapped_column(Text)
     last_processed_at: Mapped[datetime] = mapped_column(NaiveDateTime)
+    to_be_deleted: Mapped[bool] = mapped_column(Boolean, server_default=false())
 
     project_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("projects.id", ondelete="CASCADE"))
     project: Mapped["ProjectModel"] = relationship(foreign_keys=[project_id])
@@ -521,8 +522,6 @@ class GatewayModel(PipelineModelMixin, BaseModel):
 
     runs: Mapped[List["RunModel"]] = relationship(back_populates="gateway")
 
-    to_be_deleted: Mapped[bool] = mapped_column(Boolean, server_default=false())
-
     __table_args__ = (UniqueConstraint("project_id", "name", name="uq_gateways_project_id_name"),)
 
 

src/dstack/_internal/server/services/gateways/__init__.py

@@ -352,6 +352,8 @@ async def set_default_gateway(
     gateway = await get_project_gateway_model_by_name(session=session, project=project, name=name)
     if gateway is None:
         raise ResourceNotExistsError()
+    if gateway.to_be_deleted:
+        raise ServerClientError("Cannot set gateway marked for deletion as default")
     if project.default_gateway_id == gateway.id:
         return
     previous_gateway = await get_project_default_gateway_model(session, project)
@@ -434,7 +436,10 @@ async def get_project_default_gateway_model(
     session: AsyncSession, project: ProjectModel
 ) -> Optional[GatewayModel]:
     res = await session.execute(
-        select(GatewayModel).where(GatewayModel.id == project.default_gateway_id)
+        select(GatewayModel).where(
+            GatewayModel.id == project.default_gateway_id,
+            GatewayModel.to_be_deleted == False,
+        )
     )
     return res.scalar_one_or_none()
 

src/dstack/_internal/server/services/services/__init__.py

@@ -96,6 +96,10 @@ async def register_service(session: AsyncSession, run_model: RunModel, run_spec:
             raise ResourceNotExistsError(
                 f"Gateway {run_spec.configuration.gateway} does not exist"
             )
+        if gateway.to_be_deleted:
+            raise ResourceNotExistsError(
+                f"Gateway {run_spec.configuration.gateway} was marked for deletion"
+            )
     elif run_spec.configuration.gateway == False:
         gateway = None
     else: