From b3c53bcee2e2b75bc285de26be19ef47d4f788aa Mon Sep 17 00:00:00 2001 From: MasterPtato Date: Mon, 18 Nov 2024 21:51:52 +0000 Subject: [PATCH] fix: update ds state when failed to allocate --- .../services/ds/src/workflows/server/nomad/mod.rs | 2 ++ .../services/ds/src/workflows/server/pegboard/mod.rs | 11 +++++++---- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/packages/services/ds/src/workflows/server/nomad/mod.rs b/packages/services/ds/src/workflows/server/nomad/mod.rs index 7bab51689b..72bfeb9998 100644 --- a/packages/services/ds/src/workflows/server/nomad/mod.rs +++ b/packages/services/ds/src/workflows/server/nomad/mod.rs @@ -106,6 +106,8 @@ pub(crate) async fn ds_server_nomad(ctx: &mut WorkflowCtx, input: &Input) -> Glo }) .output() .await?; + + bail!("failed to evaluate"); } } Init::Destroy(sig) => { diff --git a/packages/services/ds/src/workflows/server/pegboard/mod.rs b/packages/services/ds/src/workflows/server/pegboard/mod.rs index e571cdd5fc..2bca8f2ea4 100644 --- a/packages/services/ds/src/workflows/server/pegboard/mod.rs +++ b/packages/services/ds/src/workflows/server/pegboard/mod.rs @@ -422,10 +422,13 @@ async fn wait_actor_ready(ctx: &mut WorkflowCtx, server_id: Uuid) -> GlobalResul Init::ActorStateUpdate(sig) => match sig.state { pp::ActorState::Allocated { client_id } => client_id, pp::ActorState::FailedToAllocate => { - ctx.msg(CreateFailed {}) - .tag("server_id", server_id) - .send() - .await?; + ctx.workflow(destroy::Input { + server_id, + override_kill_timeout_ms: None, + signal_actor: false, + }) + .output() + .await?; bail!("failed to allocate actor"); }