Jędrzej Stuczyński
827c13b69e
dont build netstack in CI additional rust 2024 fixes fixes removed temp.rs first round of cleanup removed duplicated NS types moved gateway probe to the monorepo
75 lines
2.1 KiB
Rust
75 lines
2.1 KiB
Rust
use crate::manager::SentError;
|
|
|
|
#[cfg(unix)]
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn wait_for_signal() {
|
|
use tokio::signal::unix::{SignalKind, signal};
|
|
let mut sigterm = signal(SignalKind::terminate()).expect("Failed to setup SIGTERM channel");
|
|
let mut sigquit = signal(SignalKind::quit()).expect("Failed to setup SIGQUIT channel");
|
|
|
|
tokio::select! {
|
|
_ = tokio::signal::ctrl_c() => {
|
|
log::info!("Received SIGINT");
|
|
},
|
|
_ = sigterm.recv() => {
|
|
log::info!("Received SIGTERM");
|
|
}
|
|
_ = sigquit.recv() => {
|
|
log::info!("Received SIGQUIT");
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(not(unix))]
|
|
pub async fn wait_for_signal() {
|
|
tokio::select! {
|
|
_ = tokio::signal::ctrl_c() => {
|
|
log::info!("Received SIGINT");
|
|
},
|
|
}
|
|
}
|
|
|
|
#[allow(deprecated)]
|
|
#[cfg(unix)]
|
|
#[allow(clippy::expect_used)]
|
|
pub async fn wait_for_signal_and_error(shutdown: &mut crate::TaskManager) -> Result<(), SentError> {
|
|
use tokio::signal::unix::{SignalKind, signal};
|
|
|
|
let mut sigterm = signal(SignalKind::terminate()).expect("Failed to setup SIGTERM channel");
|
|
let mut sigquit = signal(SignalKind::quit()).expect("Failed to setup SIGQUIT channel");
|
|
|
|
tokio::select! {
|
|
_ = tokio::signal::ctrl_c() => {
|
|
log::info!("Received SIGINT");
|
|
Ok(())
|
|
},
|
|
_ = sigterm.recv() => {
|
|
log::info!("Received SIGTERM");
|
|
Ok(())
|
|
}
|
|
_ = sigquit.recv() => {
|
|
log::info!("Received SIGQUIT");
|
|
Ok(())
|
|
}
|
|
Some(msg) = shutdown.wait_for_error() => {
|
|
log::info!("Task error: {msg:?}");
|
|
Err(msg)
|
|
}
|
|
}
|
|
}
|
|
|
|
#[allow(deprecated)]
|
|
#[cfg(not(unix))]
|
|
pub async fn wait_for_signal_and_error(shutdown: &mut crate::TaskManager) -> Result<(), SentError> {
|
|
tokio::select! {
|
|
_ = tokio::signal::ctrl_c() => {
|
|
log::info!("Received SIGINT");
|
|
Ok(())
|
|
},
|
|
Some(msg) = shutdown.wait_for_error() => {
|
|
log::info!("Task error: {msg:?}");
|
|
Err(msg)
|
|
}
|
|
}
|
|
}
|