diff --git a/beacon_node/http_api/src/validators.rs b/beacon_node/http_api/src/validators.rs index 20af7a680df..93e63953ef7 100644 --- a/beacon_node/http_api/src/validators.rs +++ b/beacon_node/http_api/src/validators.rs @@ -4,7 +4,7 @@ use eth2::types::{ self as api_types, ExecutionOptimisticFinalizedResponse, ValidatorBalanceData, ValidatorData, ValidatorId, ValidatorStatus, }; -use std::sync::Arc; +use std::{collections::HashSet, sync::Arc}; pub fn get_beacon_state_validators( state_id: StateId, @@ -18,6 +18,8 @@ pub fn get_beacon_state_validators( |state, execution_optimistic, finalized| { let epoch = state.current_epoch(); let far_future_epoch = chain.spec.far_future_epoch; + let ids_filter_set: Option> = + query_ids.as_ref().map(HashSet::from_iter); Ok(( state @@ -27,13 +29,9 @@ pub fn get_beacon_state_validators( .enumerate() // filter by validator id(s) if provided .filter(|(index, (validator, _))| { - query_ids.as_ref().map_or(true, |ids| { - ids.iter().any(|id| match id { - ValidatorId::PublicKey(pubkey) => &validator.pubkey == pubkey, - ValidatorId::Index(param_index) => { - *param_index == *index as u64 - } - }) + ids_filter_set.as_ref().map_or(true, |ids_set| { + ids_set.contains(&ValidatorId::PublicKey(validator.pubkey)) + || ids_set.contains(&ValidatorId::Index(*index as u64)) }) }) // filter by status(es) if provided and map the result @@ -83,6 +81,9 @@ pub fn get_beacon_state_validator_balances( .map_state_and_execution_optimistic_and_finalized( &chain, |state, execution_optimistic, finalized| { + let ids_filter_set: Option> = + optional_ids.map(|f| HashSet::from_iter(f.iter())); + Ok(( state .validators() @@ -91,13 +92,9 @@ pub fn get_beacon_state_validator_balances( .enumerate() // filter by validator id(s) if provided .filter(|(index, (validator, _))| { - optional_ids.map_or(true, |ids| { - ids.iter().any(|id| match id { - ValidatorId::PublicKey(pubkey) => &validator.pubkey == pubkey, - ValidatorId::Index(param_index) => { - *param_index == *index as u64 - } - }) + ids_filter_set.as_ref().map_or(true, |ids_set| { + ids_set.contains(&ValidatorId::PublicKey(validator.pubkey)) + || ids_set.contains(&ValidatorId::Index(*index as u64)) }) }) .map(|(index, (_, balance))| ValidatorBalanceData { diff --git a/common/eth2/src/types.rs b/common/eth2/src/types.rs index 5f85d777957..feff1d391a9 100644 --- a/common/eth2/src/types.rs +++ b/common/eth2/src/types.rs @@ -281,7 +281,7 @@ pub struct FinalityCheckpointsData { pub finalized: Checkpoint, } -#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)] +#[derive(Debug, Clone, Eq, PartialEq, Hash, Serialize, Deserialize)] #[serde(into = "String")] #[serde(try_from = "std::borrow::Cow")] pub enum ValidatorId {