Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: ensure server process is properly terminated and reaped

Browse Source
This commit is contained in:
Akarshan 2025-06-11 19:05:31 +05:30 committed by Louis
parent f463008362
commit 2eeabf8ae6
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src-tauri/src/core/utils/extensions/inference_llamacpp_extension/server.rs
View File

@ -166,33 +166,37 @@ pub async fn unload_llama_model(
let mut process_map = state.llama_server_process.lock().await;
match process_map.remove(&pid) {
Some(mut child) => {
log::info!("Attempting to terminate server process with PID: {}", pid);
log::info!("Terminating server process with PID: {}", pid);
match child.start_kill() {
Ok(_) => {
log::info!("Server process termination signal sent successfully");
// 1. Send the kill signal
if let Err(e) = child.kill().await {
log::error!("Failed to send kill to server process: {}", e);
return Ok(UnloadResult {
success: false,
error: Some(format!("kill failed: {}", e)),
});
}
// 2. Await its exit so the OS can reap it
match child.wait().await {
Ok(exit_status) => {
log::info!("Server exited with: {}", exit_status);
Ok(UnloadResult {
success: true,
error: None,
})
}
Err(e) => {
log::error!("Failed to kill server process: {}", e);
log::error!("Error waiting for server process: {}", e);
Ok(UnloadResult {
success: false,
error: Some(format!("Failed to kill server process: {}", e)),
error: Some(format!("wait failed: {}", e)),
})
}
}
}
None => {
log::warn!(
"Attempted to unload server with PID '{}', but no such process exists",
pid
);
log::warn!("No server with PID '{}' found to unload", pid);
Ok(UnloadResult {
success: true,
error: None,