system: handle SIGTERM signals

Cargo.toml

@@ -24,7 +24,7 @@ rand_core = "0.6.4"
 log = "0.4.17"
 log4rs = { version = "1.2.0", features = ["file_appender"] }
 rcgen = { version = "0.13.1", features = ["pem", "x509-parser"] }
-tokio = { version = "1.25.0", features = ["rt", "rt-multi-thread"] }
+tokio = { version = "1.25.0", features = ["rt", "rt-multi-thread", "signal"] }
 tonic = { version = "0.11", features = [ "tls", "transport" ] }
 tonic_lnd = { git = "https://github.com/orbitalturtle/tonic_lnd", rev="18c5a71084886024a6b90307bfb8822288c5daea", package="fedimint-tonic-lnd", features = ["lightningrpc", "routerrpc", "versionrpc"] }
 hex = "0.4.3"

src/main.rs

@@ -23,7 +23,10 @@ use std::fs::create_dir_all;
 use std::path::PathBuf;
 use std::process::exit;
 use std::sync::Arc;
-use tokio::select;
+use tokio::{
+    select,
+    signal::unix::{signal, SignalKind},
+};
 use tonic::transport::{Server, ServerTlsConfig};
 use tonic_lnd::lnrpc::GetInfoRequest;
 
@@ -117,6 +120,10 @@ async fn main() -> Result<(), ()> {
 
     info!("Starting lndk's grpc server at address {grpc_host}:{grpc_port}");
 
+    // Set up signal handlers for SIGTERM and SIGINT
+    let mut sigterm = signal(SignalKind::terminate()).expect("failed to set up SIGTERM handler");
+    let mut sigint = signal(SignalKind::interrupt()).expect("failed to set up SIGINT handler");
+
     select! {
        _ = messenger.run(args, Arc::clone(&handler)) => {
            info!("Onion messenger completed");
@@ -127,6 +134,12 @@ async fn main() -> Result<(), ()> {
                 Err(e) => error!("Error running API: {}", e),
             };
        },
+       _ = sigterm.recv() => {
+            info!("Received SIGTERM, shutting down");
+       },
+       _ = sigint.recv() => {
+            info!("Received SIGINT, shutting down");
+       },
     }
 
     Ok(())