collect storage_read_elapsed for shrink and ancient (solana-labs#28833)

gnapoli23 · Dec 16, 2022 · 9624cc0 · 9624cc0
1 parent 714f31c
commit 9624cc0
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/runtime/src/accounts_db.rs b/runtime/src/accounts_db.rs
@@ -3805,11 +3805,17 @@ impl AccountsDb {
     where
         I: Iterator<Item = &'a Arc<AccountStorageEntry>>,
     {
-        let GetUniqueAccountsResult {
-            stored_accounts: stored_accounts_temp,
-            original_bytes,
-            store_ids,
-        } = self.get_unique_accounts_from_storages(stores);
+        let (
+            GetUniqueAccountsResult {
+                stored_accounts: stored_accounts_temp,
+                original_bytes,
+                store_ids,
+            },
+            storage_read_elapsed,
+        ) = measure!(self.get_unique_accounts_from_storages(stores));
+        stats
+            .storage_read_elapsed
+            .fetch_add(storage_read_elapsed.as_us(), Ordering::Relaxed);
         *stored_accounts = stored_accounts_temp;
 
         let mut index_read_elapsed = Measure::start("index_read_elapsed");