Add keep_scanning atomic bool to interrupt scanning

src/scanner/scanner.rs

@@ -1,5 +1,6 @@
 use std::{
     collections::{HashMap, HashSet},
+    sync::atomic::AtomicBool,
     time::{Duration, Instant},
 };
 
@@ -48,6 +49,7 @@ impl SpScanner {
         start: Height,
         end: Height,
         dust_limit: Amount,
+        keep_scanning: &AtomicBool,
     ) -> Result<()> {
         if start > end {
             bail!("bigger start than end: {} > {}", start, end);
@@ -61,7 +63,8 @@ impl SpScanner {
         let block_data_stream = self.backend.get_block_data_for_range(range, dust_limit);
 
         // process blocks using block data stream
-        self.process_blocks(block_data_stream).await?;
+        self.process_blocks(block_data_stream, keep_scanning)
+            .await?;
 
         // after processing, always send update
         self.updater.update_last_scan(end);
@@ -78,12 +81,17 @@ impl SpScanner {
     async fn process_blocks(
         &mut self,
         block_data_stream: impl Stream<Item = Result<BlockData>>,
+        keep_scanning: &AtomicBool,
     ) -> Result<()> {
         pin_mut!(block_data_stream);
 
         let mut update_time: Instant = Instant::now();
 
         while let Some(blockdata) = block_data_stream.next().await {
+            if !keep_scanning.load(std::sync::atomic::Ordering::Relaxed) {
+                return Ok(());
+            }
+
             let blockdata = blockdata?;
             let blkheight = blockdata.blkheight;
             let blkhash = blockdata.blkhash;