blockworks-foundation
/
solana-with-rpc-optimizations
mirror of https://github.com/blockworks-foundation/solana-with-rpc-optimizations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove all snapshot not matching the desired hash

This commit is contained in:
Michael Vines 2020-03-17 18:10:48 -07:00
parent e6e43d236f
commit 3a501ad69e
No known key found for this signature in database
GPG Key ID: 33F4FDEC4E0E88BD
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
download-utils/src/lib.rs
View File

@ -173,20 +173,35 @@ pub fn download_genesis(
pub fn download_snapshot(
rpc_addr: &SocketAddr,
ledger_path: &Path,
snapshot_hash: (Slot, Hash),
desired_snapshot_hash: (Slot, Hash),
) -> Result<(), String> {
let snapshot_package =
solana_ledger::snapshot_utils::get_snapshot_archive_path(ledger_path, &snapshot_hash);
if snapshot_package.exists() {
// Remove all snapshot not matching the desired hash
let snapshot_packages = solana_ledger::snapshot_utils::get_snapshot_archives(ledger_path);
for (snapshot_package, snapshot_hash) in snapshot_packages.iter() {
if snapshot_hash != desired_snapshot_hash {
fs::remove_file(snapshot_package)
.unwrap_or_else(|err| info!("Failed to remove old snapshot: {:}", err));
}
}
let desired_snapshot_package = solana_ledger::snapshot_utils::get_snapshot_archive_path(
ledger_path,
&desired_snapshot_hash,
);
if desired_snapshot_package.exists() {
Ok(())
} else {
download_file(
&format!(
"http://{}/{}",
rpc_addr,
snapshot_package.file_name().unwrap().to_str().unwrap()
desired_snapshot_package
.file_name()
.unwrap()
.to_str()
.unwrap()
),
&snapshot_package,
&desired_snapshot_package,
)
}
}