Rust scheduler: termination flag

v2/rust/Cargo.toml

@@ -10,6 +10,7 @@ anyhow = { version = "*", features = ["backtrace"] }
 atomicwrites = "*"
 chrono = "0.4.31"
 clap = { version = "*", features = ["derive"] }
+ctrlc = "*"
 flexi_logger = "*"
 log = "*"
 quick-xml = { version = "0.30.0", features = ["serialize"] }

v2/rust/src/main.rs

@@ -7,11 +7,15 @@ mod logging;
 pub mod parse_xml;
 mod results;
 mod setup;
+mod termination;
 
 use anyhow::{Context, Result};
 use clap::Parser;
-use log::{debug, info};
+use log::{debug, info, warn};
 use logging::log_and_return_error;
+use std::process::exit;
+use std::thread::sleep;
+use std::time::Duration;
 
 fn main() -> Result<()> {
     let args = cli::Args::parse();
@@ -28,5 +32,16 @@ fn main() -> Result<()> {
         .map_err(log_and_return_error)?;
     debug!("Setup completed");
 
-    Ok(())
+    let termination_flag = termination::start_termination_control()
+        .context("Failed to set up termination control")
+        .map_err(log_and_return_error)?;
+    debug!("Termination control set up");
+
+    loop {
+        if termination_flag.should_terminate() {
+            warn!("Termination signal received, shutting down");
+            exit(1);
+        }
+        sleep(Duration::from_millis(100))
+    }
 }

v2/rust/src/termination.rs

@@ -0,0 +1,22 @@
+use anyhow::{Context, Result};
+use std::sync::atomic::{AtomicBool, Ordering};
+use std::sync::Arc;
+
+pub fn start_termination_control() -> Result<TerminationFlag> {
+    let raw_flag = Arc::new(AtomicBool::new(false));
+    let raw_flag_clone = raw_flag.clone();
+    ctrlc::set_handler(move || {
+        raw_flag_clone.store(true, Ordering::Relaxed);
+    })
+    .context("Failed to register signal handler for CTRL+C")?;
+    Ok(TerminationFlag(raw_flag))
+}
+
+#[derive(Clone)]
+pub struct TerminationFlag(Arc<AtomicBool>);
+
+impl TerminationFlag {
+    pub fn should_terminate(&self) -> bool {
+        self.0.load(Ordering::Relaxed)
+    }
+}