From fbcb4d82169b3864e5e2be612d65224e77d65046 Mon Sep 17 00:00:00 2001
From: HaoranYi <haoran.yi@gmail.com>
Date: Fri, 24 Jun 2022 08:44:48 -0500
Subject: [PATCH] extract get_stake_reward fn from StakeReward (#26183)

---
 runtime/src/bank.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 9a5f0c79da..377b1517f4 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -1390,6 +1390,12 @@ struct StakeReward {
     stake_account: AccountSharedData,
 }
 
+impl StakeReward {
+    pub fn get_stake_reward(&self) -> i64 {
+        self.stake_reward_info.lamports
+    }
+}
+
 /// allow [StakeReward] to be passed to `StoreAccounts` directly without copies or vec construction
 impl<'a> StorableAccounts<'a, AccountSharedData> for (Slot, &'a [StakeReward]) {
     fn pubkey(&self, index: usize) -> &Pubkey {
@@ -3204,7 +3210,7 @@ impl Bank {
 
         let mut stake_rewards = stake_rewards
             .into_iter()
-            .filter(|x| x.stake_reward_info.lamports > 0)
+            .filter(|x| x.get_stake_reward() > 0)
             .map(|x| (x.stake_pubkey, x.stake_reward_info))
             .collect();