39 lines
1.1 KiB
Rust
39 lines
1.1 KiB
Rust
use beacon_chain::{block_production::Error as BlockProductionError, BeaconChain};
|
|
use block_producer::{DutiesReader, DutiesReaderError};
|
|
use db::ClientDB;
|
|
use slot_clock::SlotClock;
|
|
use std::sync::Arc;
|
|
use types::PublicKey;
|
|
|
|
pub struct DirectDuties<T: ClientDB, U: SlotClock> {
|
|
beacon_chain: Arc<BeaconChain<T, U>>,
|
|
pubkey: PublicKey,
|
|
}
|
|
|
|
impl<T: ClientDB, U: SlotClock> DirectDuties<T, U> {
|
|
pub fn new(pubkey: PublicKey, beacon_chain: Arc<BeaconChain<T, U>>) -> Self {
|
|
Self {
|
|
beacon_chain,
|
|
pubkey,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<T: ClientDB, U: SlotClock> DutiesReader for DirectDuties<T, U>
|
|
where
|
|
BlockProductionError: From<<U>::Error>,
|
|
{
|
|
fn is_block_production_slot(&self, slot: u64) -> Result<bool, DutiesReaderError> {
|
|
let validator_index = self
|
|
.beacon_chain
|
|
.validator_index(&self.pubkey)
|
|
.ok_or_else(|| DutiesReaderError::UnknownValidator)?;
|
|
|
|
match self.beacon_chain.block_proposer(slot) {
|
|
Some(proposer) if proposer == validator_index => Ok(true),
|
|
Some(_) => Ok(false),
|
|
None => Err(DutiesReaderError::UnknownEpoch),
|
|
}
|
|
}
|
|
}
|