From 9e8639f7aecd20a078ffdae87649f3dab524be28 Mon Sep 17 00:00:00 2001
From: HaoranYi <haoran.yi@gmail.com>
Date: Fri, 21 Jul 2023 14:42:52 -0500
Subject: [PATCH] Add distributed rewards for the block in partitioned reward
 metrics (#32580)

report distributed rewards for the block in metrics

Co-authored-by: HaoranYi <haoran.yi@solana.com>
---
 runtime/src/bank.rs         | 1 +
 runtime/src/bank/metrics.rs | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index e354ae575e..e4e7668fb6 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -3714,6 +3714,7 @@ impl Bank {
             partition_index,
             store_stake_accounts_us,
             store_stake_accounts_count: this_partition_stake_rewards.len(),
+            distributed_rewards: total_rewards_in_lamports,
         };
 
         report_partitioned_reward_metrics(self, metrics);
diff --git a/runtime/src/bank/metrics.rs b/runtime/src/bank/metrics.rs
index 5766091f32..1cde4251d7 100644
--- a/runtime/src/bank/metrics.rs
+++ b/runtime/src/bank/metrics.rs
@@ -172,13 +172,13 @@ pub(crate) fn report_new_bank_metrics(
 }
 
 /// Metrics for partitioned epoch reward store
-#[allow(dead_code)]
 #[derive(Debug, Default)]
 pub(crate) struct RewardsStoreMetrics {
     pub(crate) partition_index: u64,
     pub(crate) store_stake_accounts_us: u64,
     pub(crate) store_stake_accounts_count: usize,
     pub(crate) total_stake_accounts_count: usize,
+    pub(crate) distributed_rewards: u64,
     pub(crate) pre_capitalization: u64,
     pub(crate) post_capitalization: u64,
 }
@@ -207,6 +207,7 @@ pub(crate) fn report_partitioned_reward_metrics(bank: &Bank, timings: RewardsSto
             timings.total_stake_accounts_count,
             i64
         ),
+        ("distributed_rewards", timings.distributed_rewards, i64),
         ("pre_capitalization", timings.pre_capitalization, i64),
         ("post_capitalization", timings.post_capitalization, i64),
     );