handle restarting with restart_requested shared variable

Author: manio

src/config.rs

@@ -142,6 +142,9 @@ pub struct AppConfig {
     pub ev_battery_logger: Option<PathBuf>,
     pub ev_battery_capacity: u64,
     pub ev_factor: f32,
+
+    #[serde(skip)]
+    pub restart_requested: bool,
 }
 
 impl Default for AppConfig {
@@ -176,6 +179,7 @@ impl Default for AppConfig {
             ev_battery_logger: None,
             ev_battery_capacity: 22000,
             ev_factor: 0.075,
+            restart_requested: false,
         }
     }
 }

src/io_uring.rs

@@ -87,6 +87,7 @@ async fn transfer_monitor(
     usb_bytes_written: Arc<AtomicUsize>,
     tcp_bytes_written: Arc<AtomicUsize>,
     read_timeout: Duration,
+    config: SharedConfig,
 ) -> Result<()> {
     let mut usb_bytes_out_last: usize = 0;
     let mut tcp_bytes_out_last: usize = 0;
@@ -153,6 +154,12 @@ async fn transfer_monitor(
             stall_tcp_bytes_last = tcp_bytes_out;
         }
 
+        // check if we need to restart
+        if config.read().await.restart_requested {
+            config.write().await.restart_requested = false;
+            return Err("config reload request => restarting connection".into());
+        }
+
         sleep(Duration::from_millis(100)).await;
     }
 }
@@ -384,6 +391,7 @@ pub async fn io_loop(
             file_bytes,
             stream_bytes,
             read_timeout,
+            config.clone(),
         ));
 
         // Stop as soon as one of them errors