mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-11 04:31:51 +00:00
Change validator/duties endpoint
This commit is contained in:
@@ -93,31 +93,31 @@ fn return_validator_duties<T: BeaconChainTypes>(
|
||||
epoch: Epoch,
|
||||
validator_pubkeys: Vec<PublicKey>,
|
||||
) -> Result<Vec<ValidatorDuty>, ApiError> {
|
||||
let head_state = beacon_chain.head().beacon_state;
|
||||
let head_epoch = head_state.current_epoch();
|
||||
let slots_per_epoch = T::EthSpec::slots_per_epoch();
|
||||
let head_epoch = beacon_chain.head().beacon_state.current_epoch();
|
||||
|
||||
let relative_epoch = RelativeEpoch::from_epoch(head_epoch, epoch);
|
||||
let mut state = if relative_epoch.is_err() {
|
||||
head_state
|
||||
let mut state = if RelativeEpoch::from_epoch(head_epoch, epoch).is_ok() {
|
||||
beacon_chain.head().beacon_state
|
||||
} else {
|
||||
match beacon_chain.state_at_slot(epoch.start_slot(T::EthSpec::slots_per_epoch())) {
|
||||
Ok(state) => state,
|
||||
Err(e) => {
|
||||
return Err(ApiError::ServerError(format!(
|
||||
"Unable to load state for epoch {}: {:?}",
|
||||
epoch, e
|
||||
)))
|
||||
}
|
||||
}
|
||||
let slot = if epoch > head_epoch {
|
||||
// Move to the first slot of the epoch prior to the request.
|
||||
//
|
||||
// Taking advantage of saturating epoch subtraction.
|
||||
(epoch - 1).start_slot(slots_per_epoch)
|
||||
} else {
|
||||
// Move to the end of the epoch following the target.
|
||||
//
|
||||
// Taking advantage of saturating epoch subtraction.
|
||||
(epoch + 2).start_slot(slots_per_epoch) - 1
|
||||
};
|
||||
|
||||
beacon_chain.state_at_slot(slot).map_err(|e| {
|
||||
ApiError::ServerError(format!("Unable to load state for epoch {}: {:?}", epoch, e))
|
||||
})?
|
||||
};
|
||||
|
||||
let relative_epoch = relative_epoch.or_else(|_| {
|
||||
RelativeEpoch::from_epoch(state.current_epoch(), epoch).map_err(|_| {
|
||||
ApiError::BadRequest(String::from(
|
||||
"Epoch must be within one epoch of the current epoch",
|
||||
))
|
||||
})
|
||||
})?;
|
||||
let relative_epoch = RelativeEpoch::from_epoch(state.current_epoch(), epoch)
|
||||
.map_err(|_| ApiError::ServerError(String::from("Loaded state is in the wrong epoch")))?;
|
||||
|
||||
state
|
||||
.build_committee_cache(relative_epoch, &beacon_chain.spec)
|
||||
|
||||
@@ -205,7 +205,7 @@ fn validator_duties() {
|
||||
.beacon_chain()
|
||||
.expect("client should have beacon chain");
|
||||
|
||||
let epoch = Epoch::new(0);
|
||||
let mut epoch = Epoch::new(0);
|
||||
|
||||
let validators = beacon_chain
|
||||
.head()
|
||||
@@ -220,7 +220,26 @@ fn validator_duties() {
|
||||
.block_on(remote_node.http.validator().get_duties(epoch, &validators))
|
||||
.expect("should fetch duties from http api");
|
||||
|
||||
// 1. Check at the current epoch.
|
||||
check_duties(
|
||||
duties,
|
||||
epoch,
|
||||
validators.clone(),
|
||||
beacon_chain.clone(),
|
||||
spec,
|
||||
);
|
||||
|
||||
epoch += 4;
|
||||
let duties = env
|
||||
.runtime()
|
||||
.block_on(remote_node.http.validator().get_duties(epoch, &validators))
|
||||
.expect("should fetch duties from http api");
|
||||
|
||||
// 2. Check with a long skip forward.
|
||||
check_duties(duties, epoch, validators, beacon_chain, spec);
|
||||
|
||||
// TODO: test an epoch in the past. Blocked because the `LocalBeaconNode` cannot produce a
|
||||
// chain, yet.
|
||||
}
|
||||
|
||||
fn check_duties<T: BeaconChainTypes>(
|
||||
@@ -236,7 +255,9 @@ fn check_duties<T: BeaconChainTypes>(
|
||||
"there should be a duty for each validator"
|
||||
);
|
||||
|
||||
let state = beacon_chain.head().beacon_state.clone();
|
||||
let state = beacon_chain
|
||||
.state_at_slot(epoch.start_slot(T::EthSpec::slots_per_epoch()))
|
||||
.expect("should get state at slot");
|
||||
|
||||
validators
|
||||
.iter()
|
||||
|
||||
Reference in New Issue
Block a user