Track timing of consensus stages

Cargo.toml

@@ -281,6 +281,7 @@ serial = [ "snarkos-cli/serial" ]
 test_targets = [ "snarkos-cli/test_targets" ]
 test_consensus_heights = [ "snarkos-cli/test_consensus_heights" ]
 test_network = [ "snarkos-cli/test_network" ]
+test_consensus_tracking = [ "snarkos-node/test_consensus_tracking" ]
 tokio_console = [ "snarkos-cli/tokio_console" ]
 
 [dependencies.clap]

node/Cargo.toml

@@ -51,6 +51,7 @@ cuda = [
   "snarkos-node-sync/cuda"
 ]
 serial = [ "snarkos-node-bft/serial" ]
+test_consensus_tracking = [ "snarkos-node-bft/test_consensus_tracking", "snarkos-node-consensus/test_consensus_tracking" ] 
 test = []
 
 [dependencies.aleo-std]

node/bft/Cargo.toml

@@ -44,6 +44,7 @@ test = [
   "snarkos-node-bft-ledger-service/test",
   "snarkos-node-bft-storage-service/test"
 ]
+test_consensus_tracking = [ ]
 serial = [ "snarkos-node-bft-ledger-service/serial" ]
 
 [dependencies.aleo-std]
@@ -93,6 +94,16 @@ workspace = true
 [dependencies.rayon]
 workspace = true
 
+[dependencies.serde]
+workspace = true
+features = [ "derive" ]
+
+[dependencies.serde_json]
+workspace = true
+
+[dependencies.once_cell]
+version = "1.0"
+
 [dependencies.sha2]
 version = "0.10"
 default-features = false

node/bft/src/bft.rs

@@ -480,6 +480,9 @@ impl<N: Network> BFT<N> {
         // Insert the certificate into the DAG.
         self.dag.write().insert(certificate);
 
+        #[cfg(feature = "test_consensus_tracking")]
+        crate::helpers::record_event(certificate_round, crate::helpers::ConsensusStage::CertificateAdded);
+
         // Get the previous round number.
         let commit_round = certificate_round.saturating_sub(1);
 
@@ -562,6 +565,13 @@ impl<N: Network> BFT<N> {
     ) -> Result<()> {
         // Fetch the leader round.
         let latest_leader_round = leader_certificate.round();
+
+        #[cfg(feature = "test_consensus_tracking")]
+        crate::helpers::start_subdag_stage(
+            latest_leader_round.saturating_sub(2),
+            latest_leader_round,
+            crate::helpers::SubdagStage::SubdagProcessing,
+        );
         // Determine the list of all previous leader certificates since the last committed round.
         // The order of the leader certificates is from **newest** to **oldest**.
         let mut leader_certificates = vec![leader_certificate.clone()];

node/bft/src/helpers/mod.rs

@@ -51,6 +51,11 @@ pub use telemetry::*;
 pub mod timestamp;
 pub use timestamp::*;
 
+#[cfg(feature = "test_consensus_tracking")]
+pub mod timing;
+#[cfg(feature = "test_consensus_tracking")]
+pub use timing::*;
+
 /// Formats an ID into a truncated identifier (for logging purposes).
 pub fn fmt_id(id: impl ToString) -> String {
     let id = id.to_string();