From 2f141a41bd6517bbb28d3aaebd33df4aa0c842c1 Mon Sep 17 00:00:00 2001
From: Fynn <me@enriquefynn.com>
Date: Thu, 14 Jul 2022 10:54:03 -0300
Subject: [PATCH] Add active stake per monitored validator

---
 prometheus/src/cluster_metrics.rs | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/prometheus/src/cluster_metrics.rs b/prometheus/src/cluster_metrics.rs
index 1478bb4a3812f8..59e4462d196e82 100644
--- a/prometheus/src/cluster_metrics.rs
+++ b/prometheus/src/cluster_metrics.rs
@@ -15,12 +15,13 @@ struct ValidatorVoteInfo {
     last_vote: Slot,
     vote_credits: u64,
     identity: Pubkey,
+    activated_stake: Lamports,
 }
 
 fn get_vote_state(bank: &Bank, vote_pubkey: &Pubkey) -> Option<ValidatorVoteInfo> {
     let default_vote_state = VoteState::default();
     let vote_accounts = bank.vote_accounts();
-    let (_activated_stake, vote_account) = vote_accounts.get(vote_pubkey)?;
+    let (activated_stake, vote_account) = vote_accounts.get(vote_pubkey)?;
     let vote_state = vote_account.vote_state();
     let vote_state = vote_state.as_ref().unwrap_or(&default_vote_state);
 
@@ -32,6 +33,7 @@ fn get_vote_state(bank: &Bank, vote_pubkey: &Pubkey) -> Option<ValidatorVoteInfo
         last_vote,
         vote_credits,
         identity: vote_state.node_pubkey,
+        activated_stake: Lamports(*activated_stake),
     })
 }
 
@@ -136,6 +138,23 @@ pub fn write_cluster_metrics<W: io::Write>(
                 }),
             },
         )?;
+
+        write_metric(
+            out,
+            &MetricFamily {
+                name: "solana_validator_activated_stake_sol",
+                help: "The total amount of Sol actively staked to this validator",
+                type_: "gauge",
+                metrics: banks_with_commitments.for_each_commitment(|bank| {
+                    let vote_info = get_vote_state(bank, vote_account)?;
+                    Some(
+                        Metric::new_sol(vote_info.activated_stake)
+                            .with_label("identity_account", vote_info.identity.to_string())
+                            .with_label("vote_account", vote_account.to_string()),
+                    )
+                }),
+            },
+        )?;
     }
 
     Ok(())