shutdown signal

Cargo.toml

@@ -14,7 +14,7 @@ code0-flow = { version = "0.0.44", features = ["flow_telemetry"] }
 jsonwebtoken = { version = "11.0.0", features = ["rust_crypto"] }
 log = "0.4.33"
 serde = { version = "1.0.228", features = ["derive"] }
-tokio = { version = "1.52.3", features = ["macros", "rt-multi-thread", "sync"] }
+tokio = { version = "1.52.3", features = ["macros", "rt-multi-thread", "signal", "sync"] }
 tokio-stream = "0.1.18"
 tonic = "0.14.6"
 tonic-health = "0.14.6"

gateway/src/main.rs

@@ -114,12 +114,38 @@ async fn run(config: Config) -> anyhow::Result<()> {
     }
 
     log::info!("Sagittarius gateway listening on {}", address);
-    server_builder.serve(address).await?;
+    server_builder
+        .serve_with_shutdown(address, shutdown_signal())
+        .await?;
     log::info!("Sagittarius gateway stopped");
 
     Ok(())
 }
 
+/// Resolves once a Ctrl+C (SIGINT) or SIGTERM is received, so the caller can
+/// pass it to `serve_with_shutdown` for a graceful stop.
+async fn shutdown_signal() {
+    #[cfg(unix)]
+    let sigterm = async {
+        use tokio::signal::unix::{SignalKind, signal};
+
+        let mut term = signal(SignalKind::terminate()).expect("failed to install SIGTERM handler");
+        term.recv().await;
+    };
+
+    #[cfg(not(unix))]
+    let sigterm = std::future::pending::<()>();
+
+    tokio::select! {
+        _ = tokio::signal::ctrl_c() => {
+            log::info!("Ctrl+C/Exit signal received, shutting down");
+        }
+        _ = sigterm => {
+            log::info!("SIGTERM received, shutting down");
+        }
+    }
+}
+
 fn install_panic_logging() {
     std::panic::set_hook(Box::new(move |panic_info| {
         let message = if let Some(message) = panic_info.payload().downcast_ref::<&str>() {