fix(rln): use config.open instead of sled::open

vacp2p · Aug 24, 2023 · a49f20a · a49f20a
1 parent c6c1bfd
commit a49f20a
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 23 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/rln/src/pm_tree_adapter.rs b/rln/src/pm_tree_adapter.rs
@@ -3,6 +3,8 @@ use std::path::PathBuf;
 use std::str::FromStr;
 
 use color_eyre::{Report, Result};
+use rand::distributions::Alphanumeric;
+use rand::{thread_rng, Rng};
 use serde_json::Value;
 
 use utils::pmtree::{Database, Hasher};
@@ -49,7 +51,13 @@ impl Hasher for PoseidonHash {
 }
 
 fn get_tmp_path() -> PathBuf {
-    std::env::temp_dir().join(format!("pmtree-{}", rand::random::<u64>()))
+    let rand_dir: String = thread_rng()
+        .sample_iter(&Alphanumeric)
+        .take(40)
+        .map(char::from)
+        .collect();
+
+    std::env::temp_dir().join(format!("pmtree-{}", rand_dir))
 }
 
 fn get_tmp() -> bool {
@@ -103,7 +111,7 @@ impl Default for PmtreeConfig {
         let tmp_path = get_tmp_path();
         PmtreeConfig(
             Config::new()
-                .temporary(true)
+                .temporary(false)
                 .path(tmp_path)
                 .cache_capacity(150_000)
                 .mode(Mode::HighThroughput)

diff --git a/utils/src/pm_tree/sled_adapter.rs b/utils/src/pm_tree/sled_adapter.rs
@@ -24,7 +24,7 @@ impl Database for SledDB {
     }
 
     fn load(config: Self::Config) -> PmtreeResult<Self> {
-        let db: Sled = match sled::open(&config.path) {
+        let db: Sled = match config.open() {
             Ok(db) => db,
             Err(e) => {
                 return Err(PmtreeErrorKind::DatabaseError(