fix mark-backend-lost cli command (#571)

jamsocket · Jan 23, 2024 · aec1371 · aec1371
1 parent a77959c
commit aec1371
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/cli/src/main.rs b/cli/src/main.rs
@@ -1,4 +1,5 @@
 use anyhow::{Context, Result};
+use chrono::Utc;
 use clap::{Parser, Subcommand};
 use colored::Colorize;
 use plane_core::{
@@ -7,6 +8,7 @@ use plane_core::{
             BackendState, BackendStateMessage, DockerExecutableConfig, ResourceLimits,
             TerminationRequest,
         },
+        drone_state::UpdateBackendStateMessage,
         scheduler::{DrainDrone, ScheduleRequest, ScheduleResponse},
         state::{
             BackendMessage, BackendMessageType, ClusterStateMessage, DroneMessage,
@@ -523,16 +525,32 @@ async fn main() -> Result<()> {
 
                 let mut confirmation = String::new();
                 stdin.read_line(&mut confirmation)?;
+                let confirmation = confirmation.trim();
                 if confirmation != "y" {
                     println!("skipping backend: {backend}, not marking as lost");
                     continue;
                 }
 
-                let lost_state_message =
-                    BackendStateMessage::new(BackendState::Lost, cluster.clone(), backend.clone());
+                let drone = backend_state.drone.clone().unwrap();
+                let message = UpdateBackendStateMessage {
+                    state: BackendState::Lost,
+                    cluster: cluster.clone(),
+                    backend: backend.clone(),
+                    time: Utc::now(),
+                    drone: drone.clone(),
+                };
 
-                nats.publish_jetstream(&lost_state_message).await?;
                 println!("marking backend: {backend} as lost");
+                if let Err(error) = nats.request(&message).await {
+                    tracing::error!(
+                        ?error,
+                        ?state,
+                        ?cluster,
+                        ?backend,
+                        ?drone,
+                        "Failed to update backend state."
+                    );
+                }
             }
         }
     }