MindWorkAI/AI-Studio
MindWorkAI/AI-Studio
1
0
Fork 0
mirror of https://github.com/MindWorkAI/AI-Studio.git synced 2025-04-28 11:39:48 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Refactored the shutdown creation

Browse Source
This commit is contained in:
Thorsten Sommer 2025-02-26 20:38:33 +01:00
parent dcc616f5f0
commit fa4071222e
Signed by: tsommer
GPG Key ID: 371BBA77A02C0108
1 changed files with 18 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
runtime/src/runtime_api.rs
View File

@ -1,6 +1,4 @@
#[cfg(unix)]
use std::collections::HashSet;
use log::info;
use once_cell::sync::Lazy;
use rocket::config::Shutdown;
@ -28,24 +26,8 @@ pub fn start_runtime_api() {
let api_port = *API_SERVER_PORT;
info!("Try to start the API server on 'http://localhost:{api_port}'...");
// The shutdown configuration for the runtime API server:
let shutdown = Shutdown {
// We do not want to use the Ctrl+C signal to stop the server:
ctrlc: false,
// Everything else is set to default for now:
..Shutdown::default()
};
#[cfg(unix)]
{
// Now, shutdown needs to be mutable:
let mut shutdown = shutdown;
// We do not want to use the termination signal to stop the server.
// This option, however, is only available on Unix systems:
shutdown.signals = HashSet::new();
}
// Get the shutdown configuration:
let shutdown = create_shutdown(cfg!(windows));
// Configure the runtime API server:
let figment = Figment::from(rocket::Config::release_default())
@ -101,3 +83,19 @@ pub fn start_runtime_api() {
.launch().await.unwrap();
});
}
fn create_shutdown(for_windows: bool) -> Shutdown {
let mut shutdown = Shutdown {
// We do not want to use the Ctrl+C signal to stop the server:
ctrlc: false,
// Everything else is set to default for now:
..Shutdown::default()
};
if for_windows {
shutdown.signals = HashSet::new();
}
shutdown
}