azdle/zero-to-axum
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add graceful shutdown

Browse source
This commit is contained in:
azdle 2025-05-14 13:13:00 -05:00
parent fefb73397c
commit 49c418068e
1 changed files with 30 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
src/server/mod.rs
View file

@ -8,6 +8,7 @@ use std::future::{Future, IntoFuture};
use std::net::SocketAddr; use std::net::SocketAddr;
use std::pin::pin; use std::pin::pin;
use std::pin::Pin; use std::pin::Pin;
use tokio::signal;
use tracing::info; use tracing::info;
use crate::Conf; use crate::Conf;
@ -45,7 +46,7 @@ impl ZeroToAxum {
let listener = tokio::net::TcpListener::bind(&conf.listen).await.unwrap(); let listener = tokio::net::TcpListener::bind(&conf.listen).await.unwrap();
let bound_addr = listener.local_addr().unwrap(); let bound_addr = listener.local_addr().unwrap();
let server = axum::serve(listener, app); let server = axum::serve(listener, app).with_graceful_shutdown(shutdown_signal());
info!("server started, listening on {bound_addr:?}"); info!("server started, listening on {bound_addr:?}");
@ -67,3 +68,31 @@ impl FromRef<AppState> for Key {
state.key.clone() state.key.clone()
} }
} }
async fn shutdown_signal() {
let ctrl_c = async {
signal::ctrl_c()
.await
.expect("failed to install Ctrl+C handler");
info!("got Ctrl+C, shutting down...");
};
#[cfg(unix)]
let terminate = async {
signal::unix::signal(signal::unix::SignalKind::terminate())
.expect("failed to install signal handler")
.recv()
.await;
info!("got SIGTERM, shutting down...");
};
#[cfg(not(unix))]
let terminate = std::future::pending::<()>();
tokio::select! {
_ = ctrl_c => {},
_ = terminate => {},
}
}