Adopt heaviest subtree fork choice rule (#10441)

* Add HeaviestSubtreeForkChoice * Make replay stage switch between two fork choice rules Co-authored-by: Carl <carl@solana.com>
2020-06-11 12:16:04 -07:00
parent 0510b6e336
commit 2e1d59ff85
10 changed files with 1700 additions and 635 deletions
--- a/runtime/src/epoch_stakes.rs
+++ b/runtime/src/epoch_stakes.rs
@@ -50,6 +50,13 @@ impl EpochStakes {
        &self.epoch_authorized_voters
    }

+    pub fn vote_account_stake(&self, vote_account: &Pubkey) -> u64 {
+        Stakes::vote_accounts(&self.stakes)
+            .get(vote_account)
+            .map(|(stake, _)| *stake)
+            .unwrap_or(0)
+    }
+
    fn parse_epoch_vote_accounts(
        epoch_vote_accounts: &HashMap<Pubkey, (u64, Account)>,
        leader_schedule_epoch: Epoch,