Alice Hau
37 lines
906 B
Rust
37 lines
906 B
Rust
use std::future::Future;
|
|
use std::pin::Pin;
|
|
use tokio::signal;
|
|
|
|
#[cfg(unix)]
|
|
pub fn shutdown_signal() -> Pin<Box<dyn Future<Output = ()> + Send>> {
|
|
Box::pin(async move {
|
|
let ctrl_c = async {
|
|
signal::ctrl_c()
|
|
.await
|
|
.expect("failed to install Ctrl+C handler");
|
|
};
|
|
|
|
#[cfg(unix)]
|
|
let terminate = async {
|
|
signal::unix::signal(signal::unix::SignalKind::terminate())
|
|
.expect("failed to install signal handler")
|
|
.recv()
|
|
.await;
|
|
};
|
|
|
|
tokio::select! {
|
|
_ = ctrl_c => {},
|
|
_ = terminate => {},
|
|
}
|
|
})
|
|
}
|
|
|
|
#[cfg(not(unix))]
|
|
pub fn shutdown_signal() -> Pin<Box<dyn Future<Output = ()> + Send>> {
|
|
Box::pin(async move {
|
|
signal::ctrl_c()
|
|
.await
|
|
.expect("failed to install Ctrl+C handler");
|
|
})
|
|
}
|