feat(sync): gracefully handle SIGTERMs

Author: 0xbrayo

aw-sync/src/main.rs

@@ -15,6 +15,8 @@ extern crate serde_json;
 
 use std::error::Error;
 use std::path::PathBuf;
+use std::sync::atomic::{AtomicBool, Ordering};
+use std::sync::Arc;
 
 use chrono::{DateTime, Utc};
 use clap::{Parser, Subcommand};
@@ -207,7 +209,14 @@ fn main() -> Result<(), Box<dyn Error>> {
 }
 
 fn daemon(client: &AwClient) -> Result<(), Box<dyn Error>> {
-    loop {
+    let running = Arc::new(AtomicBool::new(true));
+    let r = running.clone();
+
+    ctrlc::set_handler(move || {
+        r.store(false, Ordering::SeqCst);
+    })?;
+
+    while running.load(Ordering::SeqCst) {
         if let Err(e) = daemon_sync_cycle(client) {
             error!("Error during sync cycle: {}", e);
             // Re-throw the error
@@ -217,6 +226,9 @@ fn daemon(client: &AwClient) -> Result<(), Box<dyn Error>> {
         info!("Sync pass done, sleeping for 5 minutes");
         std::thread::sleep(std::time::Duration::from_secs(300));
     }
+
+    info!("Termination signal received, shutting down.");
+    Ok(())
 }
 
 fn daemon_sync_cycle(client: &AwClient) -> Result<(), Box<dyn Error>> {