cerc-io/lighthouse
16
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Improve test_harness efficiency for attesatations

Browse Source 
Reduce the amount of attester.poll() calls by using a hashset
This commit is contained in:
Paul Hauner 2019-01-31 20:27:15 +11:00
parent becb81d842
commit b6b738e83a
No known key found for this signature in database
GPG Key ID: D362883A9218FCC6
1 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
beacon_node/beacon_chain/test_harness/src/harness.rs
View File

@ -8,8 +8,10 @@ use db::{
use log::debug; use log::debug;
use rayon::prelude::*; use rayon::prelude::*;
use slot_clock::TestingSlotClock; use slot_clock::TestingSlotClock;
use std::collections::HashSet;
use std::fs::File; use std::fs::File;
use std::io::prelude::*; use std::io::prelude::*;
use std::iter::FromIterator;
use std::sync::Arc; use std::sync::Arc;
use types::{BeaconBlock, ChainSpec, FreeAttestation, Keypair, Validator}; use types::{BeaconBlock, ChainSpec, FreeAttestation, Keypair, Validator};
@ -121,15 +123,34 @@ impl BeaconChainHarness {
pub fn gather_free_attesations(&mut self) -> Vec<FreeAttestation> { pub fn gather_free_attesations(&mut self) -> Vec<FreeAttestation> {
let present_slot = self.beacon_chain.present_slot().unwrap(); let present_slot = self.beacon_chain.present_slot().unwrap();
let attesting_validators = self
.beacon_chain
.state(present_slot)
.unwrap()
.get_crosslink_committees_at_slot(present_slot, &self.spec)
.unwrap()
.iter()
.fold(vec![], |mut acc, (committee, _slot)| {
acc.append(&mut committee.clone());
acc
});
let attesting_validators: HashSet<usize> =
HashSet::from_iter(attesting_validators.iter().cloned());
let free_attestations: Vec<FreeAttestation> = self let free_attestations: Vec<FreeAttestation> = self
.validators .validators
.par_iter_mut() .par_iter_mut()
.filter_map(|validator| { .enumerate()
// Advance the validator slot. .filter_map(|(i, validator)| {
validator.set_slot(present_slot); if attesting_validators.contains(&i) {
// Advance the validator slot.
validator.set_slot(present_slot);
// Prompt the validator to produce an attestation (if required). // Prompt the validator to produce an attestation (if required).
validator.produce_free_attestation().ok() validator.produce_free_attestation().ok()
} else {
None
}
}) })
.collect(); .collect();