feat: graceful shutdown

frectonz · Aug 7, 2024 · 9f44ef8 · 9f44ef8
1 parent d83b0b4
commit 9f44ef8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -152,7 +152,15 @@ async fn main() -> color_eyre::Result<()> {
     }
 
     let address = args.address.parse::<std::net::SocketAddr>()?;
-    warp::serve(routes).run(address).await;
+    let (_, fut) = warp::serve(routes).bind_with_graceful_shutdown(address, async move {
+        tokio::signal::ctrl_c()
+            .await
+            .expect("failed to listen to shutdown signal");
+    });
+
+    fut.await;
+    println!();
+    tracing::info!("shutting down...");
 
     Ok(())
 }