Cache block time in Blockstore (#11955)

* Add blockstore column to cache block times

* Add method to cache block time

* Add service to cache block time

* Update rpc getBlockTime to use new method, and refactor blockstore slightly

* Return block_time with confirmed block, if available

* Add measure and warning to cache-block-time
This commit is contained in:
Tyera Eulberg 2020-09-09 09:33:14 -06:00 committed by GitHub
parent 28f2fa3fd5
commit 05db41fe9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
9 changed files with 316 additions and 106 deletions

View File

@ -0,0 +1,76 @@
use crossbeam_channel::{Receiver, RecvTimeoutError, Sender};
use solana_ledger::blockstore::Blockstore;
use solana_measure::measure::Measure;
use solana_runtime::bank::Bank;
use solana_sdk::timing::slot_duration_from_slots_per_year;
use std::{
collections::HashMap,
sync::{
atomic::{AtomicBool, Ordering},
Arc,
},
thread::{self, Builder, JoinHandle},
time::Duration,
};
pub type CacheBlockTimeReceiver = Receiver<Arc<Bank>>;
pub type CacheBlockTimeSender = Sender<Arc<Bank>>;
pub struct CacheBlockTimeService {
thread_hdl: JoinHandle<()>,
}
const CACHE_BLOCK_TIME_WARNING_MS: u64 = 150;
impl CacheBlockTimeService {
#[allow(clippy::new_ret_no_self)]
pub fn new(
cache_block_time_receiver: CacheBlockTimeReceiver,
blockstore: Arc<Blockstore>,
exit: &Arc<AtomicBool>,
) -> Self {
let exit = exit.clone();
let thread_hdl = Builder::new()
.name("solana-cache-block-time".to_string())
.spawn(move || loop {
if exit.load(Ordering::Relaxed) {
break;
}
let recv_result = cache_block_time_receiver.recv_timeout(Duration::from_secs(1));
match recv_result {
Err(RecvTimeoutError::Disconnected) => {
break;
}
Ok(bank) => {
let mut cache_block_time_timer = Measure::start("cache_block_time_timer");
Self::cache_block_time(bank, &blockstore);
cache_block_time_timer.stop();
if cache_block_time_timer.as_ms() > CACHE_BLOCK_TIME_WARNING_MS {
warn!(
"cache_block_time operation took: {}ms",
cache_block_time_timer.as_ms()
);
}
}
_ => {}
}
})
.unwrap();
Self { thread_hdl }
}
fn cache_block_time(bank: Arc<Bank>, blockstore: &Arc<Blockstore>) {
let slot_duration = slot_duration_from_slots_per_year(bank.slots_per_year());
let epoch = bank.epoch_schedule().get_epoch(bank.slot());
let stakes = HashMap::new();
let stakes = bank.epoch_vote_accounts(epoch).unwrap_or(&stakes);
if let Err(e) = blockstore.cache_block_time(bank.slot(), slot_duration, stakes) {
error!("cache_block_time failed: slot {:?} {:?}", bank.slot(), e);
}
}
pub fn join(self) -> thread::Result<()> {
self.thread_hdl.join()
}
}

View File

@ -11,6 +11,7 @@ pub mod accounts_hash_verifier;
pub mod banking_stage; pub mod banking_stage;
pub mod bigtable_upload_service; pub mod bigtable_upload_service;
pub mod broadcast_stage; pub mod broadcast_stage;
pub mod cache_block_time_service;
pub mod cluster_info_vote_listener; pub mod cluster_info_vote_listener;
pub mod commitment_service; pub mod commitment_service;
pub mod completed_data_sets_service; pub mod completed_data_sets_service;

View File

@ -3,6 +3,7 @@
use crate::{ use crate::{
bank_weight_fork_choice::BankWeightForkChoice, bank_weight_fork_choice::BankWeightForkChoice,
broadcast_stage::RetransmitSlotsSender, broadcast_stage::RetransmitSlotsSender,
cache_block_time_service::CacheBlockTimeSender,
cluster_info::ClusterInfo, cluster_info::ClusterInfo,
cluster_info_vote_listener::VoteTracker, cluster_info_vote_listener::VoteTracker,
cluster_slots::ClusterSlots, cluster_slots::ClusterSlots,
@ -106,6 +107,7 @@ pub struct ReplayStageConfig {
pub block_commitment_cache: Arc<RwLock<BlockCommitmentCache>>, pub block_commitment_cache: Arc<RwLock<BlockCommitmentCache>>,
pub transaction_status_sender: Option<TransactionStatusSender>, pub transaction_status_sender: Option<TransactionStatusSender>,
pub rewards_recorder_sender: Option<RewardsRecorderSender>, pub rewards_recorder_sender: Option<RewardsRecorderSender>,
pub cache_block_time_sender: Option<CacheBlockTimeSender>,
} }
#[derive(Default)] #[derive(Default)]
@ -235,6 +237,7 @@ impl ReplayStage {
block_commitment_cache, block_commitment_cache,
transaction_status_sender, transaction_status_sender,
rewards_recorder_sender, rewards_recorder_sender,
cache_block_time_sender,
} = config; } = config;
trace!("replay stage"); trace!("replay stage");
@ -494,6 +497,7 @@ impl ReplayStage {
&subscriptions, &subscriptions,
&block_commitment_cache, &block_commitment_cache,
&mut heaviest_subtree_fork_choice, &mut heaviest_subtree_fork_choice,
&cache_block_time_sender,
)?; )?;
}; };
voting_time.stop(); voting_time.stop();
@ -1004,6 +1008,7 @@ impl ReplayStage {
subscriptions: &Arc<RpcSubscriptions>, subscriptions: &Arc<RpcSubscriptions>,
block_commitment_cache: &Arc<RwLock<BlockCommitmentCache>>, block_commitment_cache: &Arc<RwLock<BlockCommitmentCache>>,
heaviest_subtree_fork_choice: &mut HeaviestSubtreeForkChoice, heaviest_subtree_fork_choice: &mut HeaviestSubtreeForkChoice,
cache_block_time_sender: &Option<CacheBlockTimeSender>,
) -> Result<()> { ) -> Result<()> {
if bank.is_empty() { if bank.is_empty() {
inc_new_counter_info!("replay_stage-voted_empty_bank", 1); inc_new_counter_info!("replay_stage-voted_empty_bank", 1);
@ -1029,6 +1034,12 @@ impl ReplayStage {
blockstore blockstore
.set_roots(&rooted_slots) .set_roots(&rooted_slots)
.expect("Ledger set roots failed"); .expect("Ledger set roots failed");
Self::cache_block_times(
blockstore,
bank_forks,
&rooted_slots,
cache_block_time_sender,
);
let highest_confirmed_root = Some( let highest_confirmed_root = Some(
block_commitment_cache block_commitment_cache
.read() .read()
@ -1855,6 +1866,36 @@ impl ReplayStage {
} }
} }
fn cache_block_times(
blockstore: &Arc<Blockstore>,
bank_forks: &Arc<RwLock<BankForks>>,
rooted_slots: &[Slot],
cache_block_time_sender: &Option<CacheBlockTimeSender>,
) {
if let Some(cache_block_time_sender) = cache_block_time_sender {
for slot in rooted_slots {
if blockstore
.get_block_time(*slot)
.unwrap_or_default()
.is_none()
{
if let Some(rooted_bank) = bank_forks.read().unwrap().get(*slot) {
cache_block_time_sender
.send(rooted_bank.clone())
.unwrap_or_else(|err| {
warn!("cache_block_time_sender failed: {:?}", err)
});
} else {
error!(
"rooted_bank {:?} not available in BankForks; block time not cached",
slot
);
}
}
}
}
}
pub fn get_unlock_switch_vote_slot(cluster_type: ClusterType) -> Slot { pub fn get_unlock_switch_vote_slot(cluster_type: ClusterType) -> Slot {
match cluster_type { match cluster_type {
ClusterType::Development => 0, ClusterType::Development => 0,

View File

@ -54,7 +54,6 @@ use solana_sdk::{
stake_history::StakeHistory, stake_history::StakeHistory,
system_instruction, system_instruction,
sysvar::{stake_history, Sysvar}, sysvar::{stake_history, Sysvar},
timing::slot_duration_from_slots_per_year,
transaction::{self, Transaction}, transaction::{self, Transaction},
}; };
use solana_stake_program::stake_state::StakeState; use solana_stake_program::stake_state::StakeState;
@ -686,18 +685,7 @@ impl JsonRpcRequestProcessor {
.unwrap() .unwrap()
.highest_confirmed_root() .highest_confirmed_root()
{ {
// This calculation currently assumes that bank.slots_per_year will remain unchanged after let result = self.blockstore.get_block_time(slot);
// genesis (ie. that this bank's slot_per_year will be applicable to any rooted slot being
// queried). If these values will be variable in the future, those timing parameters will
// need to be stored persistently, and the slot_duration calculation will likely need to be
// moved upstream into blockstore. Also, an explicit commitment level will need to be set.
let bank = self.bank(None);
let slot_duration = slot_duration_from_slots_per_year(bank.slots_per_year());
let epoch = bank.epoch_schedule().get_epoch(slot);
let stakes = HashMap::new();
let stakes = bank.epoch_vote_accounts(epoch).unwrap_or(&stakes);
let result = self.blockstore.get_block_time(slot, slot_duration, stakes);
self.check_slot_cleaned_up(&result, slot)?; self.check_slot_cleaned_up(&result, slot)?;
Ok(result.ok().unwrap_or(None)) Ok(result.ok().unwrap_or(None))
} else { } else {
@ -2544,6 +2532,7 @@ pub mod tests {
nonce, rpc_port, nonce, rpc_port,
signature::{Keypair, Signer}, signature::{Keypair, Signer},
system_program, system_transaction, system_program, system_transaction,
timing::slot_duration_from_slots_per_year,
transaction::{self, TransactionError}, transaction::{self, TransactionError},
}; };
use solana_transaction_status::{EncodedTransaction, TransactionWithStatusMeta, UiMessage}; use solana_transaction_status::{EncodedTransaction, TransactionWithStatusMeta, UiMessage};
@ -2555,7 +2544,7 @@ pub mod tests {
option::COption, solana_sdk::pubkey::Pubkey as SplTokenPubkey, option::COption, solana_sdk::pubkey::Pubkey as SplTokenPubkey,
state::AccountState as TokenAccountState, state::Mint, state::AccountState as TokenAccountState, state::Mint,
}; };
use std::collections::HashMap; use std::{collections::HashMap, time::Duration};
const TEST_MINT_LAMPORTS: u64 = 1_000_000; const TEST_MINT_LAMPORTS: u64 = 1_000_000;
const TEST_SLOTS_PER_EPOCH: u64 = DELINQUENT_VALIDATOR_SLOT_DISTANCE + 1; const TEST_SLOTS_PER_EPOCH: u64 = DELINQUENT_VALIDATOR_SLOT_DISTANCE + 1;
@ -2656,6 +2645,11 @@ pub mod tests {
for root in roots.iter() { for root in roots.iter() {
bank_forks.write().unwrap().set_root(*root, &None, Some(0)); bank_forks.write().unwrap().set_root(*root, &None, Some(0));
let mut stakes = HashMap::new();
stakes.insert(leader_vote_keypair.pubkey(), (1, Account::default()));
blockstore
.cache_block_time(*root, Duration::from_millis(400), &stakes)
.unwrap();
} }
} }

View File

@ -5,6 +5,7 @@ use crate::{
accounts_background_service::AccountsBackgroundService, accounts_background_service::AccountsBackgroundService,
accounts_hash_verifier::AccountsHashVerifier, accounts_hash_verifier::AccountsHashVerifier,
broadcast_stage::RetransmitSlotsSender, broadcast_stage::RetransmitSlotsSender,
cache_block_time_service::CacheBlockTimeSender,
cluster_info::ClusterInfo, cluster_info::ClusterInfo,
cluster_info_vote_listener::{VerifiedVoteReceiver, VoteTracker}, cluster_info_vote_listener::{VerifiedVoteReceiver, VoteTracker},
cluster_slots::ClusterSlots, cluster_slots::ClusterSlots,
@ -96,6 +97,7 @@ impl Tvu {
cfg: Option<Arc<AtomicBool>>, cfg: Option<Arc<AtomicBool>>,
transaction_status_sender: Option<TransactionStatusSender>, transaction_status_sender: Option<TransactionStatusSender>,
rewards_recorder_sender: Option<RewardsRecorderSender>, rewards_recorder_sender: Option<RewardsRecorderSender>,
cache_block_time_sender: Option<CacheBlockTimeSender>,
snapshot_package_sender: Option<AccountsPackageSender>, snapshot_package_sender: Option<AccountsPackageSender>,
vote_tracker: Arc<VoteTracker>, vote_tracker: Arc<VoteTracker>,
retransmit_slots_sender: RetransmitSlotsSender, retransmit_slots_sender: RetransmitSlotsSender,
@ -191,6 +193,7 @@ impl Tvu {
block_commitment_cache, block_commitment_cache,
transaction_status_sender, transaction_status_sender,
rewards_recorder_sender, rewards_recorder_sender,
cache_block_time_sender,
}; };
let replay_stage = ReplayStage::new( let replay_stage = ReplayStage::new(
@ -327,6 +330,7 @@ pub mod tests {
None, None,
None, None,
None, None,
None,
Arc::new(VoteTracker::new(&bank)), Arc::new(VoteTracker::new(&bank)),
retransmit_slots_sender, retransmit_slots_sender,
verified_vote_receiver, verified_vote_receiver,

View File

@ -2,6 +2,7 @@
use crate::{ use crate::{
broadcast_stage::BroadcastStageType, broadcast_stage::BroadcastStageType,
cache_block_time_service::{CacheBlockTimeSender, CacheBlockTimeService},
cluster_info::{ClusterInfo, Node}, cluster_info::{ClusterInfo, Node},
cluster_info_vote_listener::VoteTracker, cluster_info_vote_listener::VoteTracker,
completed_data_sets_service::CompletedDataSetsService, completed_data_sets_service::CompletedDataSetsService,
@ -149,6 +150,8 @@ struct TransactionHistoryServices {
transaction_status_service: Option<TransactionStatusService>, transaction_status_service: Option<TransactionStatusService>,
rewards_recorder_sender: Option<RewardsRecorderSender>, rewards_recorder_sender: Option<RewardsRecorderSender>,
rewards_recorder_service: Option<RewardsRecorderService>, rewards_recorder_service: Option<RewardsRecorderService>,
cache_block_time_sender: Option<CacheBlockTimeSender>,
cache_block_time_service: Option<CacheBlockTimeService>,
} }
pub struct Validator { pub struct Validator {
@ -157,6 +160,7 @@ pub struct Validator {
rpc_service: Option<(JsonRpcService, PubSubService, RpcBanksService)>, rpc_service: Option<(JsonRpcService, PubSubService, RpcBanksService)>,
transaction_status_service: Option<TransactionStatusService>, transaction_status_service: Option<TransactionStatusService>,
rewards_recorder_service: Option<RewardsRecorderService>, rewards_recorder_service: Option<RewardsRecorderService>,
cache_block_time_service: Option<CacheBlockTimeService>,
gossip_service: GossipService, gossip_service: GossipService,
serve_repair_service: ServeRepairService, serve_repair_service: ServeRepairService,
completed_data_sets_service: CompletedDataSetsService, completed_data_sets_service: CompletedDataSetsService,
@ -244,6 +248,8 @@ impl Validator {
transaction_status_service, transaction_status_service,
rewards_recorder_sender, rewards_recorder_sender,
rewards_recorder_service, rewards_recorder_service,
cache_block_time_sender,
cache_block_time_service,
}, },
) = new_banks_from_ledger(config, ledger_path, poh_verify, &exit); ) = new_banks_from_ledger(config, ledger_path, poh_verify, &exit);
@ -477,6 +483,7 @@ impl Validator {
config.enable_partition.clone(), config.enable_partition.clone(),
transaction_status_sender.clone(), transaction_status_sender.clone(),
rewards_recorder_sender, rewards_recorder_sender,
cache_block_time_sender,
snapshot_package_sender, snapshot_package_sender,
vote_tracker.clone(), vote_tracker.clone(),
retransmit_slots_sender, retransmit_slots_sender,
@ -523,6 +530,7 @@ impl Validator {
rpc_service, rpc_service,
transaction_status_service, transaction_status_service,
rewards_recorder_service, rewards_recorder_service,
cache_block_time_service,
snapshot_packager_service, snapshot_packager_service,
completed_data_sets_service, completed_data_sets_service,
tpu, tpu,
@ -587,6 +595,10 @@ impl Validator {
rewards_recorder_service.join()?; rewards_recorder_service.join()?;
} }
if let Some(cache_block_time_service) = self.cache_block_time_service {
cache_block_time_service.join()?;
}
if let Some(s) = self.snapshot_packager_service { if let Some(s) = self.snapshot_packager_service {
s.join()?; s.join()?;
} }
@ -772,6 +784,14 @@ fn initialize_rpc_transaction_history_services(
let rewards_recorder_sender = Some(rewards_recorder_sender); let rewards_recorder_sender = Some(rewards_recorder_sender);
let rewards_recorder_service = Some(RewardsRecorderService::new( let rewards_recorder_service = Some(RewardsRecorderService::new(
rewards_receiver, rewards_receiver,
blockstore.clone(),
exit,
));
let (cache_block_time_sender, cache_block_time_receiver) = unbounded();
let cache_block_time_sender = Some(cache_block_time_sender);
let cache_block_time_service = Some(CacheBlockTimeService::new(
cache_block_time_receiver,
blockstore, blockstore,
exit, exit,
)); ));
@ -780,6 +800,8 @@ fn initialize_rpc_transaction_history_services(
transaction_status_service, transaction_status_service,
rewards_recorder_sender, rewards_recorder_sender,
rewards_recorder_service, rewards_recorder_service,
cache_block_time_sender,
cache_block_time_service,
} }
} }

View File

@ -78,7 +78,6 @@ thread_local!(static PAR_THREAD_POOL_ALL_CPUS: RefCell<ThreadPool> = RefCell::ne
pub const MAX_COMPLETED_SLOTS_IN_CHANNEL: usize = 100_000; pub const MAX_COMPLETED_SLOTS_IN_CHANNEL: usize = 100_000;
pub const MAX_TURBINE_PROPAGATION_IN_MS: u64 = 100; pub const MAX_TURBINE_PROPAGATION_IN_MS: u64 = 100;
pub const MAX_TURBINE_DELAY_IN_TICKS: u64 = MAX_TURBINE_PROPAGATION_IN_MS / MS_PER_TICK; pub const MAX_TURBINE_DELAY_IN_TICKS: u64 = MAX_TURBINE_PROPAGATION_IN_MS / MS_PER_TICK;
const TIMESTAMP_SLOT_INTERVAL: u64 = 4500;
const TIMESTAMP_SLOT_RANGE: usize = 16; const TIMESTAMP_SLOT_RANGE: usize = 16;
// An upper bound on maximum number of data shreds we can handle in a slot // An upper bound on maximum number of data shreds we can handle in a slot
@ -137,6 +136,7 @@ pub struct Blockstore {
transaction_status_index_cf: LedgerColumn<cf::TransactionStatusIndex>, transaction_status_index_cf: LedgerColumn<cf::TransactionStatusIndex>,
active_transaction_status_index: RwLock<u64>, active_transaction_status_index: RwLock<u64>,
rewards_cf: LedgerColumn<cf::Rewards>, rewards_cf: LedgerColumn<cf::Rewards>,
blocktime_cf: LedgerColumn<cf::Blocktime>,
last_root: Arc<RwLock<Slot>>, last_root: Arc<RwLock<Slot>>,
insert_shreds_lock: Arc<Mutex<()>>, insert_shreds_lock: Arc<Mutex<()>>,
pub new_shreds_signals: Vec<SyncSender<bool>>, pub new_shreds_signals: Vec<SyncSender<bool>>,
@ -292,6 +292,7 @@ impl Blockstore {
let address_signatures_cf = db.column(); let address_signatures_cf = db.column();
let transaction_status_index_cf = db.column(); let transaction_status_index_cf = db.column();
let rewards_cf = db.column(); let rewards_cf = db.column();
let blocktime_cf = db.column();
let db = Arc::new(db); let db = Arc::new(db);
@ -336,6 +337,7 @@ impl Blockstore {
transaction_status_index_cf, transaction_status_index_cf,
active_transaction_status_index: RwLock::new(active_transaction_status_index), active_transaction_status_index: RwLock::new(active_transaction_status_index),
rewards_cf, rewards_cf,
blocktime_cf,
new_shreds_signals: vec![], new_shreds_signals: vec![],
completed_slots_senders: vec![], completed_slots_senders: vec![],
insert_shreds_lock: Arc::new(Mutex::new(())), insert_shreds_lock: Arc::new(Mutex::new(())),
@ -1557,12 +1559,7 @@ impl Blockstore {
} }
} }
pub fn get_block_time( pub fn get_block_time(&self, slot: Slot) -> Result<Option<UnixTimestamp>> {
&self,
slot: Slot,
slot_duration: Duration,
stakes: &HashMap<Pubkey, (u64, Account)>,
) -> Result<Option<UnixTimestamp>> {
datapoint_info!( datapoint_info!(
"blockstore-rpc-api", "blockstore-rpc-api",
("method", "get_block_time".to_string(), String) ("method", "get_block_time".to_string(), String)
@ -1573,18 +1570,56 @@ impl Blockstore {
if *lowest_cleanup_slot > 0 && *lowest_cleanup_slot >= slot { if *lowest_cleanup_slot > 0 && *lowest_cleanup_slot >= slot {
return Err(BlockstoreError::SlotCleanedUp); return Err(BlockstoreError::SlotCleanedUp);
} }
self.blocktime_cf.get(slot)
}
fn get_timestamp_slots(&self, slot: Slot, timestamp_sample_range: usize) -> Vec<Slot> {
let root_iterator = self
.db
.iter::<cf::Root>(IteratorMode::From(slot, IteratorDirection::Reverse));
if !self.is_root(slot) || root_iterator.is_err() {
return vec![];
}
let mut get_slots = Measure::start("get_slots");
let mut timestamp_slots: Vec<Slot> = root_iterator
.unwrap()
.map(|(iter_slot, _)| iter_slot)
.take(timestamp_sample_range)
.collect();
timestamp_slots.sort();
get_slots.stop();
datapoint_info!(
"blockstore-get-timestamp-slots",
("slot", slot as i64, i64),
("get_slots_us", get_slots.as_us() as i64, i64)
);
timestamp_slots
}
pub fn cache_block_time(
&self,
slot: Slot,
slot_duration: Duration,
stakes: &HashMap<Pubkey, (u64, Account)>,
) -> Result<()> {
if !self.is_root(slot) {
return Err(BlockstoreError::SlotNotRooted);
}
let mut get_unique_timestamps = Measure::start("get_unique_timestamps"); let mut get_unique_timestamps = Measure::start("get_unique_timestamps");
let unique_timestamps: HashMap<Pubkey, (Slot, UnixTimestamp)> = self let unique_timestamps: HashMap<Pubkey, (Slot, UnixTimestamp)> = self
.get_timestamp_slots(slot, TIMESTAMP_SLOT_INTERVAL, TIMESTAMP_SLOT_RANGE) .get_timestamp_slots(slot, TIMESTAMP_SLOT_RANGE)
.into_iter() .into_iter()
.flat_map(|query_slot| self.get_block_timestamps(query_slot).unwrap_or_default()) .flat_map(|query_slot| self.get_block_timestamps(query_slot).unwrap_or_default())
.collect(); .collect();
get_unique_timestamps.stop(); get_unique_timestamps.stop();
if unique_timestamps.is_empty() {
return Err(BlockstoreError::NoVoteTimestampsInRange);
}
let mut calculate_timestamp = Measure::start("calculate_timestamp"); let mut calculate_timestamp = Measure::start("calculate_timestamp");
let stake_weighted_timestamps = let stake_weighted_timestamp =
calculate_stake_weighted_timestamp(unique_timestamps, stakes, slot, slot_duration); calculate_stake_weighted_timestamp(unique_timestamps, stakes, slot, slot_duration)
.ok_or(BlockstoreError::EmptyEpochStakes)?;
calculate_timestamp.stop(); calculate_timestamp.stop();
datapoint_info!( datapoint_info!(
"blockstore-get-block-time", "blockstore-get-block-time",
@ -1600,52 +1635,7 @@ impl Blockstore {
i64 i64
) )
); );
self.blocktime_cf.put(slot, &stake_weighted_timestamp)
Ok(stake_weighted_timestamps)
}
fn get_timestamp_slots(
&self,
slot: Slot,
timestamp_interval: u64,
timestamp_sample_range: usize,
) -> Vec<Slot> {
let baseline_slot = slot - (slot % timestamp_interval);
let root_iterator = self.db.iter::<cf::Root>(IteratorMode::From(
baseline_slot,
IteratorDirection::Forward,
));
if !self.is_root(slot) || root_iterator.is_err() {
return vec![];
}
let mut get_slots = Measure::start("get_slots");
let mut slots: Vec<Slot> = root_iterator
.unwrap()
.map(|(iter_slot, _)| iter_slot)
.take(timestamp_sample_range)
.filter(|&iter_slot| iter_slot <= slot)
.collect();
if slots.len() < timestamp_sample_range && baseline_slot >= timestamp_interval {
let earlier_baseline = baseline_slot - timestamp_interval;
let earlier_root_iterator = self.db.iter::<cf::Root>(IteratorMode::From(
earlier_baseline,
IteratorDirection::Forward,
));
if let Ok(iterator) = earlier_root_iterator {
slots = iterator
.map(|(iter_slot, _)| iter_slot)
.take(timestamp_sample_range)
.collect();
}
}
get_slots.stop();
datapoint_info!(
"blockstore-get-timestamp-slots",
("slot", slot as i64, i64),
("get_slots_us", get_slots.as_us() as i64, i64)
);
slots
} }
pub fn get_first_available_block(&self) -> Result<Slot> { pub fn get_first_available_block(&self) -> Result<Slot> {
@ -1698,6 +1688,7 @@ impl Blockstore {
.unwrap_or_else(|| panic!("Rooted slot {:?} must have blockhash", slot)); .unwrap_or_else(|| panic!("Rooted slot {:?} must have blockhash", slot));
let rewards = self.rewards_cf.get(slot)?.unwrap_or_else(Vec::new); let rewards = self.rewards_cf.get(slot)?.unwrap_or_else(Vec::new);
let block_time = self.blocktime_cf.get(slot)?;
let block = ConfirmedBlock { let block = ConfirmedBlock {
previous_blockhash: previous_blockhash.to_string(), previous_blockhash: previous_blockhash.to_string(),
@ -1709,7 +1700,7 @@ impl Blockstore {
slot_transaction_iterator, slot_transaction_iterator,
), ),
rewards, rewards,
block_time: None, // See https://github.com/solana-labs/solana/issues/10089 block_time,
}; };
return Ok(block); return Ok(block);
} }
@ -5561,8 +5552,6 @@ pub mod tests {
fn test_get_timestamp_slots() { fn test_get_timestamp_slots() {
let timestamp_sample_range = 5; let timestamp_sample_range = 5;
let ticks_per_slot = 5; let ticks_per_slot = 5;
// Smaller interval than TIMESTAMP_SLOT_INTERVAL for convenience of building blockstore
let timestamp_interval = 7;
/* /*
Build a blockstore with < TIMESTAMP_SLOT_RANGE roots Build a blockstore with < TIMESTAMP_SLOT_RANGE roots
*/ */
@ -5589,11 +5578,11 @@ pub mod tests {
blockstore.set_roots(&[1, 2, 3]).unwrap(); blockstore.set_roots(&[1, 2, 3]).unwrap();
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(2, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(2, timestamp_sample_range),
vec![0, 1, 2] vec![0, 1, 2]
); );
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(3, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(3, timestamp_sample_range),
vec![0, 1, 2, 3] vec![0, 1, 2, 3]
); );
@ -5601,14 +5590,13 @@ pub mod tests {
Blockstore::destroy(&blockstore_path).expect("Expected successful database destruction"); Blockstore::destroy(&blockstore_path).expect("Expected successful database destruction");
/* /*
Build a blockstore in the ledger with the following rooted slots: Build a blockstore in the ledger with gaps in rooted slot sequence
[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 14, 15, 16, 17]
*/ */
let blockstore_path = get_tmp_ledger_path!(); let blockstore_path = get_tmp_ledger_path!();
let blockstore = Blockstore::open(&blockstore_path).unwrap(); let blockstore = Blockstore::open(&blockstore_path).unwrap();
blockstore.set_roots(&[0]).unwrap(); blockstore.set_roots(&[0]).unwrap();
let desired_roots = vec![1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 16, 17, 18, 19]; let desired_roots = vec![1, 2, 3, 5, 6, 8, 11];
let mut last_entry_hash = Hash::default(); let mut last_entry_hash = Hash::default();
for (i, slot) in desired_roots.iter().enumerate() { for (i, slot) in desired_roots.iter().enumerate() {
let parent = { let parent = {
@ -5629,28 +5617,20 @@ pub mod tests {
blockstore.set_roots(&desired_roots).unwrap(); blockstore.set_roots(&desired_roots).unwrap();
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(2, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(2, timestamp_sample_range),
vec![0, 1, 2] vec![0, 1, 2]
); );
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(6, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(6, timestamp_sample_range),
vec![0, 1, 2, 3, 4] vec![1, 2, 3, 5, 6]
); );
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(8, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(8, timestamp_sample_range),
vec![0, 1, 2, 3, 4] vec![2, 3, 5, 6, 8]
); );
assert_eq!( assert_eq!(
blockstore.get_timestamp_slots(13, timestamp_interval, timestamp_sample_range), blockstore.get_timestamp_slots(11, timestamp_sample_range),
vec![8, 9, 10, 11, 12] vec![3, 5, 6, 8, 11]
);
assert_eq!(
blockstore.get_timestamp_slots(18, timestamp_interval, timestamp_sample_range),
vec![8, 9, 10, 11, 12]
);
assert_eq!(
blockstore.get_timestamp_slots(19, timestamp_interval, timestamp_sample_range),
vec![14, 16, 17, 18, 19]
); );
} }
@ -5754,7 +5734,7 @@ pub mod tests {
let confirmed_block = ledger.get_confirmed_block(slot + 1, None).unwrap(); let confirmed_block = ledger.get_confirmed_block(slot + 1, None).unwrap();
assert_eq!(confirmed_block.transactions.len(), 100); assert_eq!(confirmed_block.transactions.len(), 100);
let expected_block = ConfirmedBlock { let mut expected_block = ConfirmedBlock {
transactions: expected_transactions transactions: expected_transactions
.iter() .iter()
.cloned() .cloned()
@ -5774,6 +5754,14 @@ pub mod tests {
let not_root = ledger.get_confirmed_block(slot + 2, None).unwrap_err(); let not_root = ledger.get_confirmed_block(slot + 2, None).unwrap_err();
assert_matches!(not_root, BlockstoreError::SlotNotRooted); assert_matches!(not_root, BlockstoreError::SlotNotRooted);
// Test block_time returns, if available
let timestamp = 1_576_183_541;
ledger.blocktime_cf.put(slot + 1, &timestamp).unwrap();
expected_block.block_time = Some(timestamp);
let confirmed_block = ledger.get_confirmed_block(slot + 1, None).unwrap();
assert_eq!(confirmed_block, expected_block);
drop(ledger); drop(ledger);
Blockstore::destroy(&ledger_path).expect("Expected successful database destruction"); Blockstore::destroy(&ledger_path).expect("Expected successful database destruction");
} }
@ -5821,14 +5809,25 @@ pub mod tests {
); );
assert_eq!(blockstore.get_block_timestamps(2).unwrap(), vec![]); assert_eq!(blockstore.get_block_timestamps(2).unwrap(), vec![]);
// Build epoch vote_accounts HashMap to test stake-weighted block time
blockstore.set_roots(&[3, 8]).unwrap(); blockstore.set_roots(&[3, 8]).unwrap();
let mut stakes = HashMap::new(); let mut stakes = HashMap::new();
let slot_duration = Duration::from_millis(400);
for slot in &[1, 2, 3, 8] {
assert!(blockstore
.cache_block_time(*slot, slot_duration, &stakes)
.is_err());
}
// Build epoch vote_accounts HashMap to test stake-weighted block time
for (i, keypair) in vote_keypairs.iter().enumerate() { for (i, keypair) in vote_keypairs.iter().enumerate() {
stakes.insert(keypair.pubkey(), (1 + i as u64, Account::default())); stakes.insert(keypair.pubkey(), (1 + i as u64, Account::default()));
} }
let slot_duration = Duration::from_millis(400); for slot in &[1, 2, 3, 8] {
let block_time_slot_3 = blockstore.get_block_time(3, slot_duration, &stakes); blockstore
.cache_block_time(*slot, slot_duration, &stakes)
.unwrap();
}
let block_time_slot_3 = blockstore.get_block_time(3);
let mut total_stake = 0; let mut total_stake = 0;
let mut expected_time: u64 = (0..6) let mut expected_time: u64 = (0..6)
@ -5844,14 +5843,53 @@ pub mod tests {
expected_time /= total_stake; expected_time /= total_stake;
assert_eq!(block_time_slot_3.unwrap().unwrap() as u64, expected_time); assert_eq!(block_time_slot_3.unwrap().unwrap() as u64, expected_time);
assert_eq!( assert_eq!(
blockstore blockstore.get_block_time(8).unwrap().unwrap() as u64,
.get_block_time(8, slot_duration, &stakes)
.unwrap()
.unwrap() as u64,
expected_time + 2 // At 400ms block duration, 5 slots == 2sec expected_time + 2 // At 400ms block duration, 5 slots == 2sec
); );
} }
#[test]
fn test_get_block_time_no_timestamps() {
let vote_keypairs: Vec<Keypair> = (0..6).map(|_| Keypair::new()).collect();
// Populate slot 1 with vote transactions, none of which have timestamps
let mut vote_entries: Vec<Entry> = Vec::new();
for (i, keypair) in vote_keypairs.iter().enumerate() {
let vote = Vote {
slots: vec![1],
hash: Hash::default(),
timestamp: None,
};
let vote_ix = vote_instruction::vote(&keypair.pubkey(), &keypair.pubkey(), vote);
let vote_msg = Message::new(&[vote_ix], Some(&keypair.pubkey()));
let vote_tx = Transaction::new(&[keypair], vote_msg, Hash::default());
vote_entries.push(next_entry_mut(&mut Hash::default(), 0, vec![vote_tx]));
let mut tick = create_ticks(1, 0, hash(&serialize(&i).unwrap()));
vote_entries.append(&mut tick);
}
let shreds = entries_to_test_shreds(vote_entries, 1, 0, true, 0);
let ledger_path = get_tmp_ledger_path!();
let blockstore = Blockstore::open(&ledger_path).unwrap();
blockstore.insert_shreds(shreds, None, false).unwrap();
// Populate slot 2 with ticks only
fill_blockstore_slot_with_ticks(&blockstore, 6, 2, 1, Hash::default());
blockstore.set_roots(&[0, 1, 2]).unwrap();
// Build epoch vote_accounts HashMap to test stake-weighted block time
let mut stakes = HashMap::new();
for (i, keypair) in vote_keypairs.iter().enumerate() {
stakes.insert(keypair.pubkey(), (1 + i as u64, Account::default()));
}
let slot_duration = Duration::from_millis(400);
for slot in &[1, 2, 3, 8] {
assert!(blockstore
.cache_block_time(*slot, slot_duration, &stakes)
.is_err());
assert_eq!(blockstore.get_block_time(*slot).unwrap(), None);
}
}
#[test] #[test]
fn test_calculate_stake_weighted_timestamp() { fn test_calculate_stake_weighted_timestamp() {
let recent_timestamp: UnixTimestamp = 1_578_909_061; let recent_timestamp: UnixTimestamp = 1_578_909_061;

View File

@ -133,6 +133,10 @@ impl Blockstore {
& self & self
.db .db
.delete_range_cf::<cf::Rewards>(&mut write_batch, from_slot, to_slot) .delete_range_cf::<cf::Rewards>(&mut write_batch, from_slot, to_slot)
.is_ok()
& self
.db
.delete_range_cf::<cf::Blocktime>(&mut write_batch, from_slot, to_slot)
.is_ok(); .is_ok();
let mut w_active_transaction_status_index = let mut w_active_transaction_status_index =
self.active_transaction_status_index.write().unwrap(); self.active_transaction_status_index.write().unwrap();
@ -223,6 +227,10 @@ impl Blockstore {
&& self && self
.rewards_cf .rewards_cf
.compact_range(from_slot, to_slot) .compact_range(from_slot, to_slot)
.unwrap_or(false)
&& self
.blocktime_cf
.compact_range(from_slot, to_slot)
.unwrap_or(false); .unwrap_or(false);
compact_timer.stop(); compact_timer.stop();
if !result { if !result {

View File

@ -10,7 +10,11 @@ use rocksdb::{
use serde::de::DeserializeOwned; use serde::de::DeserializeOwned;
use serde::Serialize; use serde::Serialize;
use solana_runtime::hardened_unpack::UnpackError; use solana_runtime::hardened_unpack::UnpackError;
use solana_sdk::{clock::Slot, pubkey::Pubkey, signature::Signature}; use solana_sdk::{
clock::{Slot, UnixTimestamp},
pubkey::Pubkey,
signature::Signature,
};
use solana_transaction_status::{Rewards, TransactionStatusMeta}; use solana_transaction_status::{Rewards, TransactionStatusMeta};
use std::{collections::HashMap, fs, marker::PhantomData, path::Path, sync::Arc}; use std::{collections::HashMap, fs, marker::PhantomData, path::Path, sync::Arc};
use thiserror::Error; use thiserror::Error;
@ -46,6 +50,8 @@ const ADDRESS_SIGNATURES_CF: &str = "address_signatures";
const TRANSACTION_STATUS_INDEX_CF: &str = "transaction_status_index"; const TRANSACTION_STATUS_INDEX_CF: &str = "transaction_status_index";
/// Column family for Rewards /// Column family for Rewards
const REWARDS_CF: &str = "rewards"; const REWARDS_CF: &str = "rewards";
/// Column family for Blocktime
const BLOCKTIME_CF: &str = "blocktime";
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum BlockstoreError { pub enum BlockstoreError {
@ -61,6 +67,8 @@ pub enum BlockstoreError {
UnpackError(#[from] UnpackError), UnpackError(#[from] UnpackError),
UnableToSetOpenFileDescriptorLimit, UnableToSetOpenFileDescriptorLimit,
TransactionStatusSlotMismatch, TransactionStatusSlotMismatch,
EmptyEpochStakes,
NoVoteTimestampsInRange,
} }
pub type Result<T> = std::result::Result<T, BlockstoreError>; pub type Result<T> = std::result::Result<T, BlockstoreError>;
@ -128,6 +136,10 @@ pub mod columns {
#[derive(Debug)] #[derive(Debug)]
/// The rewards column /// The rewards column
pub struct Rewards; pub struct Rewards;
#[derive(Debug)]
/// The blocktime column
pub struct Blocktime;
} }
pub enum AccessType { pub enum AccessType {
@ -187,8 +199,9 @@ impl Rocks {
recovery_mode: Option<BlockstoreRecoveryMode>, recovery_mode: Option<BlockstoreRecoveryMode>,
) -> Result<Rocks> { ) -> Result<Rocks> {
use columns::{ use columns::{
AddressSignatures, DeadSlots, DuplicateSlots, ErasureMeta, Index, Orphans, Rewards, AddressSignatures, Blocktime, DeadSlots, DuplicateSlots, ErasureMeta, Index, Orphans,
Root, ShredCode, ShredData, SlotMeta, TransactionStatus, TransactionStatusIndex, Rewards, Root, ShredCode, ShredData, SlotMeta, TransactionStatus,
TransactionStatusIndex,
}; };
fs::create_dir_all(&path)?; fs::create_dir_all(&path)?;
@ -221,6 +234,8 @@ impl Rocks {
let transaction_status_index_cf_descriptor = let transaction_status_index_cf_descriptor =
ColumnFamilyDescriptor::new(TransactionStatusIndex::NAME, get_cf_options()); ColumnFamilyDescriptor::new(TransactionStatusIndex::NAME, get_cf_options());
let rewards_cf_descriptor = ColumnFamilyDescriptor::new(Rewards::NAME, get_cf_options()); let rewards_cf_descriptor = ColumnFamilyDescriptor::new(Rewards::NAME, get_cf_options());
let blocktime_cf_descriptor =
ColumnFamilyDescriptor::new(Blocktime::NAME, get_cf_options());
let cfs = vec![ let cfs = vec![
(SlotMeta::NAME, meta_cf_descriptor), (SlotMeta::NAME, meta_cf_descriptor),
@ -239,6 +254,7 @@ impl Rocks {
transaction_status_index_cf_descriptor, transaction_status_index_cf_descriptor,
), ),
(Rewards::NAME, rewards_cf_descriptor), (Rewards::NAME, rewards_cf_descriptor),
(Blocktime::NAME, blocktime_cf_descriptor),
]; ];
// Open the database // Open the database
@ -276,8 +292,9 @@ impl Rocks {
fn columns(&self) -> Vec<&'static str> { fn columns(&self) -> Vec<&'static str> {
use columns::{ use columns::{
AddressSignatures, DeadSlots, DuplicateSlots, ErasureMeta, Index, Orphans, Rewards, AddressSignatures, Blocktime, DeadSlots, DuplicateSlots, ErasureMeta, Index, Orphans,
Root, ShredCode, ShredData, SlotMeta, TransactionStatus, TransactionStatusIndex, Rewards, Root, ShredCode, ShredData, SlotMeta, TransactionStatus,
TransactionStatusIndex,
}; };
vec![ vec![
@ -294,6 +311,7 @@ impl Rocks {
AddressSignatures::NAME, AddressSignatures::NAME,
TransactionStatusIndex::NAME, TransactionStatusIndex::NAME,
Rewards::NAME, Rewards::NAME,
Blocktime::NAME,
] ]
} }
@ -519,6 +537,14 @@ impl TypedColumn for columns::Rewards {
type Type = Rewards; type Type = Rewards;
} }
impl SlotColumn for columns::Blocktime {}
impl ColumnName for columns::Blocktime {
const NAME: &'static str = BLOCKTIME_CF;
}
impl TypedColumn for columns::Blocktime {
type Type = UnixTimestamp;
}
impl Column for columns::ShredCode { impl Column for columns::ShredCode {
type Index = (u64, u64); type Index = (u64, u64);