feat: add graceful shutdown

src/main.rs

@@ -62,10 +62,38 @@ async fn serve(app: Router, port: u16) {
  tracing::debug!("listening on {}", addr);
  axum::Server::bind(&addr)
  .serve(app.layer(TraceLayer::new_for_http()).into_make_service())
+ .with_graceful_shutdown(shutdown_signal())
  .await
  .unwrap();
 }
 
+// graceful shutdown
+async fn 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 => {},
+ }
+
+ println!("signal received, starting graceful shutdown");
+}
+
 
 // we can extract the connection pool with `State`
 async fn using_connection_pool_extractor(