feat(sync): gracefully handle SIGTERMs

Author: 0xbrayo

aw-sync/src/main.rs

@@ -15,6 +15,11 @@ extern crate serde_json;
 
 use std::error::Error;
 use std::path::PathBuf;
+use signal_hook::consts::signal::{SIGINT, SIGTERM};
+use signal_hook::iterator::Signals;
+use std::sync::mpsc;
+use std::thread;
+use std::time::Duration;
 
 use chrono::{DateTime, Utc};
 use clap::{Parser, Subcommand};
@@ -207,16 +212,36 @@ fn main() -> Result<(), Box<dyn Error>> {
 }
 
 fn daemon(client: &AwClient) -> Result<(), Box<dyn Error>> {
-    loop {
+    let (tx, rx) = mpsc::channel();
+
+    // Spawn a thread to handle signals
+    let mut signals = Signals::new(&[SIGTERM, SIGINT])?;
+    thread::spawn(move || {
+        for _ in signals.forever() {
+            let _ = tx.send(());
+        }
+    });
+
+    loop{
         if let Err(e) = daemon_sync_cycle(client) {
             error!("Error during sync cycle: {}", e);
             // Re-throw the error
             return Err(e);
         }
 
         info!("Sync pass done, sleeping for 5 minutes");
-        std::thread::sleep(std::time::Duration::from_secs(300));
+
+        // Wait for either the sleep duration or a signal
+        match rx.recv_timeout(Duration::from_secs(300)) {
+            Ok(_) | Err(mpsc::RecvTimeoutError::Disconnected) => {
+                info!("Received termination signal, shutting down");
+                break;
+            }
+            Err(mpsc::RecvTimeoutError::Timeout) => {}
+        }
     }
+
+    Ok(())
 }
 
 fn daemon_sync_cycle(client: &AwClient) -> Result<(), Box<dyn Error>> {