Add shutdown listening (non-graceful)

src/lib.rs

@@ -9,6 +9,7 @@ pub mod keys;
 pub mod serde_utils;
 pub mod server;
 pub mod service;
+pub mod shutdown;
 pub mod task_runner;
 pub mod tasks;
 pub mod types;

src/main.rs

@@ -8,6 +8,7 @@ use tracing_subscriber::util::SubscriberInitExt;
 use tracing_subscriber::EnvFilter;
 use tx_sitter::config::load_config;
 use tx_sitter::service::Service;
+use tx_sitter::shutdown::spawn_await_shutdown_task;
 
 #[derive(Parser)]
 #[command(author, version, about)]
@@ -55,6 +56,8 @@ async fn main() -> eyre::Result<()> {
         )?;
     }
 
+    spawn_await_shutdown_task();
+
     tracing::info!(?config, "Starting service");
     let service = Service::new(config).await?;
     service.wait().await?;

src/shutdown.rs

@@ -0,0 +1,28 @@
+use core::panic;
+
+use tokio::signal::unix::{signal, SignalKind};
+
+pub fn spawn_await_shutdown_task() {
+    tokio::spawn(async {
+        let result = await_shutdown_signal().await;
+        if let Err(err) = result {
+            tracing::error!("Error while waiting for shutdown signal: {}", err);
+            panic!("Error while waiting for shutdown signal: {}", err);
+        }
+
+        tracing::info!("Shutdown complete");
+        std::process::exit(0);
+    });
+}
+
+pub async fn await_shutdown_signal() -> eyre::Result<()> {
+    let mut sigint = signal(SignalKind::interrupt())?;
+    let mut sigterm = signal(SignalKind::terminate())?;
+
+    tokio::select! {
+        _ = sigint.recv() => { tracing::info!("SIGINT received, shutting down"); }
+        _ = sigterm.recv() => { tracing::info!("SIGTERM received, shutting down"); }
+    };
+
+    Ok(())
+}