mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-04 13:41:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

krata: implement fast exit code notification

Browse Source
This commit is contained in:
Alex Zenla
2024-03-15 17:36:26 +00:00
parent e3aa54edea
commit 48123166fe
5 changed files with 189 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
crates/kratactl/src/console.rs
View File

@ -9,7 +9,7 @@ use krata::{
common::GuestStatus,
control::{watch_events_reply::Event, ConsoleDataReply, ConsoleDataRequest},
};
use log::{debug, warn};
use log::debug;
use termion::raw::IntoRawMode;
use tokio::{
fs::File,
@ -61,7 +61,10 @@ impl StdioConsoleStream {
Ok(())
}
pub async fn guest_exit_hook(id: String, events: EventStream) -> Result<JoinHandle<()>> {
pub async fn guest_exit_hook(
id: String,
events: EventStream,
) -> Result<JoinHandle<Option<i32>>> {
Ok(tokio::task::spawn(async move {
let mut stream = events.subscribe();
while let Ok(event) = stream.recv().await {
@ -80,16 +83,17 @@ impl StdioConsoleStream {
}
if let Some(exit_info) = state.exit_info {
std::process::exit(exit_info.code);
return Some(exit_info.code);
}
if state.status() == GuestStatus::Destroy {
warn!("attached guest was destroyed");
std::process::exit(1);
let status = state.status();
if status == GuestStatus::Destroy || status == GuestStatus::Destroyed {
return Some(10);
}
}
}
}
None
}))
}
}