feat: Support shutdown with SIGINT

Author: jan-auer

Cargo.lock

@@ -4,7 +4,8 @@ version = "0.1.0"
 edition = "2024"
 
 [dependencies]
-tokio = { version = "1.45.1", features = ["full"] }
-tonic = { version = "0.13.1" }
+elegant-departure = { version = "0.3.1", features = ["tokio"] }
 proto-codegen = { path = "../proto-codegen" }
 service = { path = "../service" }
+tokio = { version = "1.45.1", features = ["full"] }
+tonic = { version = "0.13.1" }

server/Cargo.toml

@@ -5,11 +5,12 @@
 //! an `HTTP` layer which can serve files directly to *external clients*.
 
 use proto_codegen::storage::{AllocateBlobRequest, AllocateBlobResponse, StorageId};
+use tokio::signal::unix::SignalKind;
 use tonic::{Request, Response, Status, transport::Server};
 
 use proto_codegen::storage::storage_server::{Storage, StorageServer};
 
-#[derive(Default)]
+#[derive(Debug, Default)]
 pub struct MockStorage {}
 
 #[tonic::async_trait]
@@ -37,13 +38,19 @@ impl Storage for MockStorage {
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let addr = "0.0.0.0:50051".parse().unwrap();
-    let greeter = MockStorage::default();
+    let storage = MockStorage::default();
 
     println!("Server listening on {addr}");
 
+    let shutdown = elegant_departure::tokio::depart()
+        .on_termination()
+        .on_sigint()
+        .on_signal(SignalKind::hangup())
+        .on_signal(SignalKind::quit());
+
     Server::builder()
-        .add_service(StorageServer::new(greeter))
-        .serve(addr)
+        .add_service(StorageServer::new(storage))
+        .serve_with_shutdown(addr, shutdown)
         .await?;
 
     Ok(())