Stop signal and shutdown node on Ctl-C

bennyhodl · Dec 17, 2024 · 1d3e21a · 1d3e21a
1 parent 1588263
commit 1d3e21a
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/ddk-node/src/lib.rs b/ddk-node/src/lib.rs
@@ -90,13 +90,18 @@ impl DdkNode {
         let ddk: Ddk = builder.finish().await?;
 
         ddk.start()?;
-
         let node = DdkNode::new(ddk);
-
-        Server::builder()
+        let node_stop = node.node.clone();
+        let server = Server::builder()
             .add_service(DdkRpcServer::new(node))
-            .serve(opts.grpc_host.parse()?)
-            .await?;
+            .serve_with_shutdown(opts.grpc_host.parse()?, async {
+                tokio::signal::ctrl_c()
+                    .await
+                    .expect("Failed to install Ctrl+C signal handler");
+                let _ = node_stop.stop();
+            });
+
+        server.await?;
 
         Ok(())
     }