Adds AccountsDb::get_bank_hash_info() (solana-labs#29908)

kruelmanta · Jan 25, 2023 · dde4b5d · dde4b5d
1 parent fa39bfe
commit dde4b5d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/runtime/src/accounts.rs b/runtime/src/accounts.rs
@@ -2443,15 +2443,15 @@ mod tests {
     }
 
     #[test]
-    #[should_panic]
     fn test_accounts_empty_bank_hash() {
         let accounts = Accounts::new_with_config_for_tests(
             Vec::new(),
             &ClusterType::Development,
             AccountSecondaryIndexes::default(),
             AccountShrinkThreshold::default(),
         );
-        accounts.bank_hash_info_at(1);
+        assert!(accounts.accounts_db.get_bank_hash_info(0).is_some());
+        assert!(accounts.accounts_db.get_bank_hash_info(1).is_none());
     }
 
     #[test]

diff --git a/runtime/src/accounts_db.rs b/runtime/src/accounts_db.rs
@@ -7547,6 +7547,11 @@ impl AccountsDb {
         accounts_delta_hash
     }
 
+    /// Get the bank hash info for `slot`
+    pub fn get_bank_hash_info(&self, slot: Slot) -> Option<BankHashInfo> {
+        self.bank_hashes.read().unwrap().get(&slot).cloned()
+    }
+
     fn update_index<'a, T: ReadableAccount + Sync>(
         &self,
         infos: Vec<AccountInfo>,