Avoid looping over banks again, we know there is only root_bank

solana-labs · Dec 14, 2023 · 9cac54c · 9cac54c
1 parent 6a40a81
commit 9cac54c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/runtime/src/bank_forks.rs b/runtime/src/bank_forks.rs
@@ -110,12 +110,11 @@ impl BankForks {
         }
 
         let mut descendants = HashMap::<_, HashSet<_>>::new();
-        for (slot, bank) in &banks {
-            descendants.entry(*slot).or_default();
-            for parent in bank.proper_ancestors() {
-                descendants.entry(parent).or_default().insert(*slot);
-            }
+        descendants.entry(root_slot).or_default();
+        for parent in root_bank.proper_ancestors() {
+            descendants.entry(parent).or_default().insert(root_slot);
         }
+
         let bank_forks = Arc::new(RwLock::new(Self {
             root: Arc::new(AtomicSlot::new(root_slot)),
             banks,