Add set_root timing metrics (#21119) (#21135)

(cherry picked from commit 1e5212e60d)

Co-authored-by: sakridge <sakridge@gmail.com>

runtime/src/bank.rs

@@ -71,7 +71,7 @@ use rayon::{
     ThreadPool, ThreadPoolBuilder,
 };
 use solana_measure::measure::Measure;
-use solana_metrics::{datapoint_debug, inc_new_counter_debug, inc_new_counter_info};
+use solana_metrics::{inc_new_counter_debug, inc_new_counter_info};
 use solana_sdk::{
     account::{
         create_account_shared_data_with_fields as create_account, from_account, Account,
@@ -212,6 +212,11 @@ type RentCollectionCycleParams = (
     PartitionsPerCycle,
 );
 
+pub struct SquashTiming {
+    pub squash_accounts_ms: u64,
+    pub squash_cache_ms: u64,
+}
+
 type EpochCount = u64;
 
 #[derive(Clone)]
@@ -2786,7 +2791,7 @@ impl Bank {
 
     /// squash the parent's state up into this Bank,
     ///   this Bank becomes a root
-    pub fn squash(&self) {
+    pub fn squash(&self) -> SquashTiming {
         self.freeze();
 
         //this bank and all its parents are now on the rooted path
@@ -2808,11 +2813,10 @@ impl Bank {
             .for_each(|slot| self.src.status_cache.write().unwrap().add_root(*slot));
         squash_cache_time.stop();
 
-        datapoint_debug!(
-            "tower-observed",
-            ("squash_accounts_ms", squash_accounts_time.as_ms(), i64),
-            ("squash_cache_ms", squash_cache_time.as_ms(), i64)
-        );
+        SquashTiming {
+            squash_accounts_ms: squash_accounts_time.as_ms(),
+            squash_cache_ms: squash_cache_time.as_ms(),
+        }
     }
 
     /// Return the more recent checkpoint of this bank instance.

runtime/src/bank_forks.rs

@@ -5,7 +5,7 @@ use crate::{
     bank::Bank,
 };
 use log::*;
-use solana_metrics::inc_new_counter_info;
+use solana_measure::measure::Measure;
 use solana_sdk::{clock::Slot, hash::Hash, timing};
 use std::{
     collections::{hash_map::Entry, HashMap, HashSet},
@@ -238,15 +238,22 @@ impl BankForks {
         let mut banks = vec![root_bank];
         let parents = root_bank.parents();
         banks.extend(parents.iter());
+        let total_banks = banks.len();
+        let mut total_squash_accounts_ms = 0;
+        let mut total_squash_cache_ms = 0;
+        let mut total_snapshot_ms = 0;
         for bank in banks.iter() {
             let bank_slot = bank.slot();
             if bank.block_height() % self.accounts_hash_interval_slots == 0
                 && bank_slot > self.last_accounts_hash_slot
             {
                 self.last_accounts_hash_slot = bank_slot;
-                bank.squash();
+                let squash_timing = bank.squash();
+                total_squash_accounts_ms += squash_timing.squash_accounts_ms;
+                total_squash_cache_ms += squash_timing.squash_cache_ms;
                 is_root_bank_squashed = bank_slot == root;
 
+                let mut snapshot_time = Measure::start("squash::snapshot_time");
                 if self.snapshot_config.is_some()
                     && accounts_background_request_sender.is_snapshot_creation_enabled()
                 {
@@ -267,22 +274,35 @@ impl BankForks {
                         );
                     }
                 }
+                snapshot_time.stop();
+                total_snapshot_ms += snapshot_time.as_ms();
                 break;
             }
         }
         if !is_root_bank_squashed {
-            root_bank.squash();
+            let squash_timing = root_bank.squash();
+            total_squash_accounts_ms += squash_timing.squash_accounts_ms;
+            total_squash_cache_ms += squash_timing.squash_cache_ms;
         }
         let new_tx_count = root_bank.transaction_count();
+        let mut prune_time = Measure::start("set_root::prune");
         self.prune_non_rooted(root, highest_confirmed_root);
+        prune_time.stop();
 
-        inc_new_counter_info!(
-            "bank-forks_set_root_ms",
-            timing::duration_as_ms(&set_root_start.elapsed()) as usize
-        );
-        inc_new_counter_info!(
-            "bank-forks_set_root_tx_count",
-            (new_tx_count - root_tx_count) as usize
+        datapoint_info!(
+            "bank-forks_set_root",
+            (
+                "elapsed_ms",
+                timing::duration_as_ms(&set_root_start.elapsed()) as usize,
+                i64
+            ),
+            ("slot", root, i64),
+            ("total_banks", total_banks, i64),
+            ("total_squash_cache_ms", total_squash_cache_ms, i64),
+            ("total_squash_accounts_ms", total_squash_accounts_ms, i64),
+            ("total_snapshot_ms", total_snapshot_ms, i64),
+            ("tx_count", (new_tx_count - root_tx_count) as usize, i64),
+            ("prune_non_rooted_ms", prune_time.as_ms(), i64),
         );
     }