Improve allocation in get_attestation_participants

2026-03-03 00:31:50 +00:00 · 2019-03-10 21:07:09 +11:00
parent 53456a6c79
commit a44d80006a
1 changed files with 5 additions and 3 deletions
--- a/eth2/types/src/beacon_state.rs
+++ b/eth2/types/src/beacon_state.rs
@@ -531,12 +531,14 @@ impl BeaconState {
            return Err(Error::InvalidBitfield);
        }

-        let mut participants = vec![];
+        let mut participants = Vec::with_capacity(committee.len());
        for (i, validator_index) in committee.iter().enumerate() {
-            if bitfield.get(i).unwrap() {
-                participants.push(*validator_index);
+            match bitfield.get(i) {
+                Ok(bit) if bit == true => participants.push(*validator_index),
+                _ => {}
            }
        }
+        participants.shrink_to_fit();

        Ok(participants)
    }