fix: graceful shutdown in unix

ElaBosak233 · Nov 15, 2024 · d15147a · d15147a
1 parent 597e733
commit d15147a
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 83 deletions.
diff --git a/deploys/compose.dev.yml b/deploys/compose.dev.yml
diff --git a/deploys/compose.yml b/deploys/compose.yml
diff --git a/src/main.rs b/src/main.rs
@@ -68,12 +68,17 @@ async fn bootstrap() {
 
 async fn shutdown_signal() {
     let ctrl_c = async {
-        tokio::signal::ctrl_c().await.expect("Failed to install Ctrl+C handler");
+        tokio::signal::ctrl_c()
+            .await
+            .expect("Failed to install Ctrl+C handler");
     };
 
     #[cfg(unix)]
     let terminate = async {
-        tokio::signal::unix::signal(tokio::signal::unix::SignalKind::terminate());
+        tokio::signal::unix::signal(tokio::signal::unix::SignalKind::terminate())
+            .expect("Failed to install signal handler")
+            .recv()
+            .await;
     };
 
     #[cfg(not(unix))]

diff --git a/src/web/router/mod.rs b/src/web/router/mod.rs
@@ -1,11 +1,7 @@
-use axum::{
-    middleware::from_fn,
-    response::IntoResponse,
-    Router,
-};
-use tower_http::{trace::TraceLayer};
+use axum::{middleware::from_fn, response::IntoResponse, Router};
+use tower_http::trace::TraceLayer;
 
-use crate::{web::middleware};
+use crate::web::middleware;
 
 pub mod api;
 pub mod metric;