feat(agent/grpc): add ungraceful shutdown

virt-do · Apr 25, 2024 · 91a5e54 · 91a5e54
1 parent 96af7c9
commit 91a5e54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/agent/src/workload/service.rs b/src/agent/src/workload/service.rs
@@ -1,4 +1,4 @@
-use std::{process::exit, sync::Arc};
+use std::{process, sync::Arc};
 
 use crate::agent::{self, ExecuteRequest, ExecuteResponse, SignalRequest};
 
@@ -62,6 +62,6 @@ impl WorkloadRunner for WorkloadRunnerService {
     }
 
     async fn signal(&self, _: Request<SignalRequest>) -> Result<()> {
-        unreachable!();
+        process::exit(0)
     }
 }