main: handle sigterm/sigint signals

Cargo.toml

@@ -32,6 +32,7 @@ configure_me = "0.4.0"
 bytes = "1.4.0"
 triggered = "0.1.2"
 prost = "0.12"
+signal-hook = "0.3.17"
 
 [dev-dependencies]
 bitcoincore-rpc = { package="core-rpc", version = "0.17.0" }

src/main.rs

@@ -19,10 +19,13 @@ use lndk::{
 };
 use lndkrpc::offers_server::OffersServer;
 use log::{error, info};
+use signal_hook::consts::{SIGINT, SIGTERM};
+use signal_hook::iterator::Signals;
 use std::fs::create_dir_all;
 use std::path::PathBuf;
 use std::process::exit;
 use std::sync::Arc;
+use std::thread;
 use tokio::select;
 use tonic::transport::{Server, ServerTlsConfig};
 use tonic_lnd::lnrpc::GetInfoRequest;
@@ -53,13 +56,27 @@ async fn main() -> Result<(), ()> {
     let lnd_args = LndCfg::new(config.address, creds.clone());
 
     let (shutdown, listener) = triggered::trigger();
-    let signals = LifecycleSignals { shutdown, listener };
+    let signals = LifecycleSignals {
+        shutdown: shutdown.clone(),
+        listener: listener.clone(),
+    };
     let args = Cfg {
         lnd: lnd_args,
         signals,
         skip_version_check: config.skip_version_check,
     };
 
+    let mut shutdown_signals = Signals::new([SIGINT, SIGTERM]).map_err(|e| {
+        error!("Error initializing shutdown signals: {e}.");
+    })?;
+
+    thread::spawn(move || {
+        if let Some(sig) = shutdown_signals.forever().next() {
+            info!("Received shutdown signal {:?}. Shutting down.", sig);
+            shutdown.trigger();
+        }
+    });
+
     let response_invoice_timeout = config.response_invoice_timeout;
     if let Some(timeout) = response_invoice_timeout {
         if timeout == 0 {
@@ -113,7 +130,7 @@ async fn main() -> Result<(), ()> {
         .tls_config(ServerTlsConfig::new().identity(identity))
         .expect("couldn't configure tls")
         .add_service(OffersServer::new(server))
-        .serve(addr);
+        .serve_with_shutdown(addr, listener);
 
     info!("Starting lndk's grpc server at address {grpc_host}:{grpc_port}");