mirror of https://github.com/ospab/ostp.git
39 lines
913 B
Rust
39 lines
913 B
Rust
use anyhow::Result;
|
|
|
|
#[cfg(unix)]
|
|
pub async fn wait_for_shutdown_signal() -> Result<()> {
|
|
use tokio::signal::unix::{signal, SignalKind};
|
|
|
|
let mut sigterm = signal(SignalKind::terminate())?;
|
|
let mut sigint = signal(SignalKind::interrupt())?;
|
|
|
|
tokio::select! {
|
|
_ = sigterm.recv() => {}
|
|
_ = sigint.recv() => {}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(not(unix))]
|
|
pub async fn wait_for_shutdown_signal() -> Result<()> {
|
|
#[cfg(target_os = "windows")]
|
|
{
|
|
use tokio::signal::windows::{ctrl_break, ctrl_c, ctrl_close};
|
|
let mut c_c = ctrl_c()?;
|
|
let mut c_close = ctrl_close()?;
|
|
let mut c_break = ctrl_break()?;
|
|
|
|
tokio::select! {
|
|
_ = c_c.recv() => {}
|
|
_ = c_close.recv() => {}
|
|
_ = c_break.recv() => {}
|
|
}
|
|
}
|
|
#[cfg(not(target_os = "windows"))]
|
|
{
|
|
tokio::signal::ctrl_c().await?;
|
|
}
|
|
Ok(())
|
|
}
|