use log::info;
use tokio::signal;
pub async fn shutdown_signal() {
info!("Waiting for shutdown signal (Ctrl+C or SIGTERM)...");
async_shutdown_signal().await;
info!("Received shutdown signal");
}
pub async fn async_shutdown_signal() {
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;
};
#[cfg(not(unix))]
let terminate = std::future::pending::<()>();
tokio::select! {
_ = ctrl_c => {},
_ = terminate => {},
}
info!("shutdown signal received, starting graceful shutdown");
}