2019-11-04 18:10:06 -08:00
|
|
|
use crate::{
|
|
|
|
bank_forks::{BankForks, SnapshotConfig},
|
2020-01-13 13:13:52 -08:00
|
|
|
blockstore::Blockstore,
|
|
|
|
blockstore_processor::{self, BankForksInfo, BlockstoreProcessorError, ProcessOptions},
|
2020-01-14 17:15:26 -08:00
|
|
|
entry::VerifyRecyclers,
|
2019-11-04 18:10:06 -08:00
|
|
|
leader_schedule_cache::LeaderScheduleCache,
|
|
|
|
snapshot_utils,
|
|
|
|
};
|
|
|
|
use log::*;
|
2019-11-08 20:56:57 -08:00
|
|
|
use solana_sdk::genesis_config::GenesisConfig;
|
2019-12-05 18:41:29 -08:00
|
|
|
use std::{fs, path::PathBuf, sync::Arc};
|
2019-11-04 18:10:06 -08:00
|
|
|
|
|
|
|
pub fn load(
|
2019-11-08 20:56:57 -08:00
|
|
|
genesis_config: &GenesisConfig,
|
2020-01-13 13:13:52 -08:00
|
|
|
blockstore: &Blockstore,
|
2019-12-05 18:41:29 -08:00
|
|
|
account_paths: Vec<PathBuf>,
|
2019-11-04 18:10:06 -08:00
|
|
|
snapshot_config: Option<&SnapshotConfig>,
|
|
|
|
process_options: ProcessOptions,
|
2020-01-13 13:13:52 -08:00
|
|
|
) -> Result<(BankForks, Vec<BankForksInfo>, LeaderScheduleCache), BlockstoreProcessorError> {
|
2019-11-04 18:10:06 -08:00
|
|
|
if let Some(snapshot_config) = snapshot_config.as_ref() {
|
|
|
|
info!(
|
|
|
|
"Initializing snapshot path: {:?}",
|
|
|
|
snapshot_config.snapshot_path
|
|
|
|
);
|
|
|
|
let _ = fs::remove_dir_all(&snapshot_config.snapshot_path);
|
|
|
|
fs::create_dir_all(&snapshot_config.snapshot_path)
|
|
|
|
.expect("Couldn't create snapshot directory");
|
|
|
|
|
|
|
|
let tar =
|
|
|
|
snapshot_utils::get_snapshot_tar_path(&snapshot_config.snapshot_package_output_path);
|
|
|
|
if tar.exists() {
|
|
|
|
info!("Loading snapshot package: {:?}", tar);
|
|
|
|
// Fail hard here if snapshot fails to load, don't silently continue
|
2019-12-05 18:41:29 -08:00
|
|
|
|
|
|
|
if account_paths.is_empty() {
|
|
|
|
panic!("Account paths not present when booting from snapshot")
|
|
|
|
}
|
|
|
|
|
2019-11-04 18:10:06 -08:00
|
|
|
let deserialized_bank = snapshot_utils::bank_from_archive(
|
2019-12-05 18:41:29 -08:00
|
|
|
&account_paths,
|
2019-11-04 18:10:06 -08:00
|
|
|
&snapshot_config.snapshot_path,
|
|
|
|
&tar,
|
|
|
|
)
|
|
|
|
.expect("Load from snapshot failed");
|
|
|
|
|
2020-01-13 13:13:52 -08:00
|
|
|
return blockstore_processor::process_blockstore_from_root(
|
2019-11-08 20:56:57 -08:00
|
|
|
genesis_config,
|
2020-01-13 13:13:52 -08:00
|
|
|
blockstore,
|
2019-11-04 18:10:06 -08:00
|
|
|
Arc::new(deserialized_bank),
|
|
|
|
&process_options,
|
2020-01-14 17:15:26 -08:00
|
|
|
&VerifyRecyclers::default(),
|
2019-11-04 18:10:06 -08:00
|
|
|
);
|
|
|
|
} else {
|
|
|
|
info!("Snapshot package does not exist: {:?}", tar);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
info!("Snapshots disabled");
|
|
|
|
}
|
|
|
|
|
|
|
|
info!("Processing ledger from genesis");
|
2020-01-13 13:13:52 -08:00
|
|
|
blockstore_processor::process_blockstore(
|
2019-11-08 20:56:57 -08:00
|
|
|
&genesis_config,
|
2020-01-13 13:13:52 -08:00
|
|
|
&blockstore,
|
2019-11-04 18:10:06 -08:00
|
|
|
account_paths,
|
|
|
|
process_options,
|
|
|
|
)
|
|
|
|
}
|