From 3bca862db8d389191a283ea5945b066365a42f72 Mon Sep 17 00:00:00 2001
From: Wen <113942165+wen-coding@users.noreply.github.com>
Date: Wed, 6 Nov 2024 10:00:42 -0800
Subject: [PATCH] Add Prefix to log line so it's easier to search. (#3486)

* Add Prefix to log line so it's easier to search.

* Make linter happy.

* Also print the percentage in both log lines.
---
 wen-restart/src/heaviest_fork_aggregate.rs | 9 ++++++++-
 wen-restart/src/wen_restart.rs             | 8 +++++---
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/wen-restart/src/heaviest_fork_aggregate.rs b/wen-restart/src/heaviest_fork_aggregate.rs
index 710a6010292daf..0d62824a247963 100644
--- a/wen-restart/src/heaviest_fork_aggregate.rs
+++ b/wen-restart/src/heaviest_fork_aggregate.rs
@@ -162,8 +162,15 @@ impl HeaviestForkAggregate {
     }
 
     pub(crate) fn print_block_stake_map(&self) {
+        let total_stake = self.epoch_stakes.total_stake();
         for ((slot, hash), stake) in self.block_stake_map.iter() {
-            info!("Slot: {}, Hash: {}, Stake: {}", slot, hash, stake,);
+            info!(
+                "Heaviest Fork Aggregated Slot: {}, Hash: {}, Stake: {}, Percent: {:.2}%",
+                slot,
+                hash,
+                stake,
+                *stake as f64 / total_stake as f64 * 100.0,
+            );
         }
     }
 }
diff --git a/wen-restart/src/wen_restart.rs b/wen-restart/src/wen_restart.rs
index 703845e154c868..0c79f32eea8bbe 100644
--- a/wen-restart/src/wen_restart.rs
+++ b/wen-restart/src/wen_restart.rs
@@ -747,10 +747,12 @@ pub(crate) fn aggregate_restart_heaviest_fork(
                 .unwrap()
                 .total_active_stake = current_total_active_stake;
         }
+        let total_active_stake = heaviest_fork_aggregate.total_active_stake();
         info!(
-            "Total active stake: {} Total stake {}",
-            heaviest_fork_aggregate.total_active_stake(),
-            total_stake
+            "Total active stake: {} Total stake {} Active percent: {:.2}%",
+            total_active_stake,
+            total_stake,
+            total_active_stake as f64 / total_stake as f64 * 100.0,
         );
         write_wen_restart_records(wen_restart_path, progress)?;
         let elapsed = timestamp().saturating_sub(start);