Core/grin
0
0
Fork 0
mirror of https://github.com/mimblewimble/grin.git synced 2025-01-21 03:21:08 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cleanup path handling with AsRef<Path> (#3284)

Browse source 
* AsRef<Path>

* bump
This commit is contained in:
Antioch Peverell 2020-03-30 11:35:21 +01:00 committed by GitHub
parent 7385e8ce7f
commit e49eecae5c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
chain/src/chain.rs
View file

@ -38,7 +38,7 @@ use grin_store::Error::NotFoundErr;
use std::collections::HashMap;
use std::fs::{self, File};
use std::io::Read;
use std::path::PathBuf;
use std::path::{Path, PathBuf};
use std::sync::atomic::{AtomicUsize, Ordering};
use std::sync::Arc;
use std::time::{Duration, Instant};
@ -175,17 +175,13 @@ impl Chain {
let mut txhashset = txhashset::TxHashSet::open(db_root.clone(), store.clone(), None)?;
let mut header_pmmr = PMMRHandle::new(
&db_root,
"header",
"header_head",
Path::new(&db_root).join("header").join("header_head"),
false,
ProtocolVersion(1),
None,
)?;
let mut sync_pmmr = PMMRHandle::new(
&db_root,
"header",
"sync_head",
Path::new(&db_root).join("header").join("sync_head"),
false,
ProtocolVersion(1),
None,

32
chain/src/txhashset/txhashset.rs
View file

@ -55,20 +55,14 @@ pub struct PMMRHandle<T: PMMRable> {
impl<T: PMMRable> PMMRHandle<T> {
/// Constructor to create a PMMR handle from an existing directory structure on disk.
/// Creates the backend files as necessary if they do not already exist.
pub fn new(
root_dir: &str,
sub_dir: &str,
file_name: &str,
pub fn new<P: AsRef<Path>>(
path: P,
prunable: bool,
version: ProtocolVersion,
header: Option<&BlockHeader>,
) -> Result<PMMRHandle<T>, Error> {
let path = Path::new(root_dir).join(sub_dir).join(file_name);
fs::create_dir_all(path.clone())?;
let path_str = path
.to_str()
.ok_or_else(|| ErrorKind::Other("invalid file path".to_owned()))?;
let backend = PMMRBackend::new(path_str.to_string(), prunable, version, header)?;
fs::create_dir_all(&path)?;
let backend = PMMRBackend::new(&path, prunable, version, header)?;
let last_pos = backend.unpruned_size();
Ok(PMMRHandle { backend, last_pos })
}
@ -130,18 +124,18 @@ impl TxHashSet {
header: Option<&BlockHeader>,
) -> Result<TxHashSet, Error> {
let output_pmmr_h = PMMRHandle::new(
&root_dir,
TXHASHSET_SUBDIR,
OUTPUT_SUBDIR,
Path::new(&root_dir)
.join(TXHASHSET_SUBDIR)
.join(OUTPUT_SUBDIR),
true,
ProtocolVersion(1),
header,
)?;
let rproof_pmmr_h = PMMRHandle::new(
&root_dir,
TXHASHSET_SUBDIR,
RANGE_PROOF_SUBDIR,
Path::new(&root_dir)
.join(TXHASHSET_SUBDIR)
.join(RANGE_PROOF_SUBDIR),
true,
ProtocolVersion(1),
header,
@ -154,9 +148,9 @@ impl TxHashSet {
let versions = vec![ProtocolVersion(2), ProtocolVersion(1)];
for version in versions {
let handle = PMMRHandle::new(
&root_dir,
TXHASHSET_SUBDIR,
KERNEL_SUBDIR,
Path::new(&root_dir)
.join(TXHASHSET_SUBDIR)
.join(KERNEL_SUBDIR),
false, // not prunable
version,
None,