interrupt prefetch after state root calc

Author: pythonberg1997

crates/blockchain-tree/src/chain.rs

@@ -236,10 +236,10 @@ impl AppendableChain {
 
         let initial_execution_outcome = ExecutionOutcome::from((state, block.number));
 
-        // stop the prefetch task.
-        if let Some(interrupt_tx) = interrupt_tx {
-            let _ = interrupt_tx.send(());
-        }
+        // // stop the prefetch task.
+        // if let Some(interrupt_tx) = interrupt_tx {
+        //     let _ = interrupt_tx.send(());
+        // }
 
         // check state root if the block extends the canonical chain __and__ if state root
         // validation was requested.
@@ -262,6 +262,12 @@ impl AppendableChain {
                 let state_root = provider.state_root(hashed_state)?;
                 (state_root, None)
             };
+
+            // stop the prefetch task.
+            if let Some(interrupt_tx) = interrupt_tx {
+                let _ = interrupt_tx.send(());
+            }
+
             if block.state_root != state_root {
                 tracing::debug!(
                     target: "blockchain_tree::chain",
@@ -286,6 +292,11 @@ impl AppendableChain {
 
             Ok((initial_execution_outcome, trie_updates))
         } else {
+            // stop the prefetch task.
+            if let Some(interrupt_tx) = interrupt_tx {
+                let _ = interrupt_tx.send(());
+            }
+
             Ok((initial_execution_outcome, None))
         }
     }