From 6d1bea7fb43afc4b8438bc03fa01b6adf94a018c Mon Sep 17 00:00:00 2001
From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com>
Date: Wed, 30 Sep 2020 01:08:04 +0000
Subject: [PATCH] Include active stake in 'epoch_rewards' datapoint (#12573)

(cherry picked from commit 82848d6c73ab3553f71ab72092286de9c1af60dd)

Co-authored-by: Michael Vines <mvines@gmail.com>
---
 runtime/src/bank.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index 339ae9fede..98833166fb 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -1157,6 +1157,14 @@ impl Bank {
         self.capitalization
             .fetch_add(validator_rewards_paid, Ordering::Relaxed);
 
+        let active_stake = if let Some(stake_history_entry) =
+            self.stakes.read().unwrap().history().get(&self.epoch)
+        {
+            stake_history_entry.effective
+        } else {
+            0
+        };
+
         datapoint_warn!(
             "epoch_rewards",
             ("slot", self.slot, i64),
@@ -1165,6 +1173,7 @@ impl Bank {
             ("foundation_rate", foundation_rate, f64),
             ("epoch_duration_in_years", epoch_duration_in_years, f64),
             ("validator_rewards", validator_rewards_paid, i64),
+            ("active_stake", active_stake, i64),
             ("pre_capitalization", capitalization, i64),
             ("post_capitalization", self.capitalization(), i64)
         );