gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Reduce TempDir exposure

Browse Source
This commit is contained in:
Michael Vines
2021-01-04 12:48:34 -08:00
parent f2a7f561a0
commit 9f70f7dc3e
1 changed files with 3 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
runtime/src/snapshot_utils.rs
View File

@ -27,7 +27,6 @@ use std::{
str::FromStr, str::FromStr,
}; };
use tar::Archive; use tar::Archive;
use tempfile::TempDir;
use thiserror::Error; use thiserror::Error;
pub const SNAPSHOT_STATUS_CACHE_FILE_NAME: &str = "status_cache"; pub const SNAPSHOT_STATUS_CACHE_FILE_NAME: &str = "status_cache";
@ -236,7 +235,7 @@ pub fn archive_snapshot_package(snapshot_package: &AccountsPackage) -> Result<()
serialize_status_cache( serialize_status_cache(
snapshot_package.root, snapshot_package.root,
&snapshot_package.slot_deltas, &snapshot_package.slot_deltas,
&snapshot_package.snapshot_links, &snapshot_package.snapshot_links.path().join(SNAPSHOT_STATUS_CACHE_FILE_NAME),
)?; )?;
let mut timer = Measure::start("snapshot_package-package_snapshots"); let mut timer = Measure::start("snapshot_package-package_snapshots");
@ -546,14 +545,10 @@ pub fn add_snapshot<P: AsRef<Path>>(
fn serialize_status_cache( fn serialize_status_cache(
slot: Slot, slot: Slot,
slot_deltas: &[BankSlotDelta], slot_deltas: &[BankSlotDelta],
snapshot_links: &TempDir, status_cache_path: &Path,
) -> Result<()> { ) -> Result<()> {
// the status cache is stored as snapshot_path/status_cache
let snapshot_status_cache_file_path =
snapshot_links.path().join(SNAPSHOT_STATUS_CACHE_FILE_NAME);
let mut status_cache_serialize = Measure::start("status_cache_serialize-ms"); let mut status_cache_serialize = Measure::start("status_cache_serialize-ms");
let consumed_size = serialize_snapshot_data_file(&snapshot_status_cache_file_path, |stream| { let consumed_size = serialize_snapshot_data_file(status_cache_path, |stream| {
serialize_into(stream, slot_deltas)?; serialize_into(stream, slot_deltas)?;
Ok(()) Ok(())
})?; })?;