chore(rocks_db): move ShallowTempDir to benches crate

Author: rymnc

benches/benches/vm_set/blockchain.rs

@@ -20,10 +20,7 @@ use fuel_core::{
         GenesisDatabase,
     },
     service::Config,
-    state::{
-        historical_rocksdb::HistoricalRocksDB,
-        rocks_db::ShallowTempDir,
-    },
+    state::historical_rocksdb::HistoricalRocksDB,
 };
 use fuel_core_benches::*;
 use fuel_core_storage::{
@@ -67,12 +64,12 @@ use rand::{
 pub struct BenchDb {
     db: GenesisDatabase,
     /// Used for RAII cleanup. Contents of this directory are deleted on drop.
-    _tmp_dir: ShallowTempDir,
+    _tmp_dir: utils::ShallowTempDir,
 }
 
 impl BenchDb {
     fn new(contract_id: &ContractId) -> anyhow::Result<Self> {
-        let tmp_dir = ShallowTempDir::new();
+        let tmp_dir = utils::ShallowTempDir::new();
 
         let db = HistoricalRocksDB::<OnChain>::default_open(
             tmp_dir.path(),

benches/src/lib.rs

@@ -38,6 +38,7 @@ use std::{
 
 pub mod default_gas_costs;
 pub mod import;
+pub mod utils;
 
 pub use fuel_core_storage::vm_storage::VmStorage;
 pub use rand::Rng;

benches/src/utils.rs

@@ -0,0 +1,40 @@
+use rand::RngCore;
+use std::{
+    env,
+    path::PathBuf,
+};
+
+/// Reimplementation of `tempdir::TempDir` that allows creating a new
+/// instance without actually creating a new directory on the filesystem.
+/// This is needed since rocksdb requires empty directory for checkpoints.
+pub struct ShallowTempDir {
+    path: PathBuf,
+}
+
+impl Default for ShallowTempDir {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl ShallowTempDir {
+    /// Creates a random directory.
+    pub fn new() -> Self {
+        let mut rng = rand::thread_rng();
+        let mut path = env::temp_dir();
+        path.push(format!("fuel-core-shallow-{}", rng.next_u64()));
+        Self { path }
+    }
+
+    /// Returns the path of the directory.
+    pub fn path(&self) -> &PathBuf {
+        &self.path
+    }
+}
+
+impl Drop for ShallowTempDir {
+    fn drop(&mut self) {
+        // Ignore errors
+        let _ = std::fs::remove_dir_all(&self.path);
+    }
+}

crates/fuel-core/src/state/rocks_db.rs

@@ -35,7 +35,6 @@ use fuel_core_storage::{
     Result as StorageResult,
 };
 use itertools::Itertools;
-use rand::RngCore;
 use rocksdb::{
     BlockBasedOptions,
     BoundColumnFamily,
@@ -55,7 +54,6 @@ use rocksdb::{
 use std::{
     cmp,
     collections::BTreeMap,
-    env,
     fmt,
     fmt::Formatter,
     iter,
@@ -69,41 +67,6 @@ use tempfile::TempDir;
 
 type DB = DBWithThreadMode<MultiThreaded>;
 
-/// Reimplementation of `tempdir::TempDir` that allows creating a new
-/// instance without actually creating a new directory on the filesystem.
-/// This is needed since rocksdb requires empty directory for checkpoints.
-pub struct ShallowTempDir {
-    path: PathBuf,
-}
-
-impl Default for ShallowTempDir {
-    fn default() -> Self {
-        Self::new()
-    }
-}
-
-impl ShallowTempDir {
-    /// Creates a random directory.
-    pub fn new() -> Self {
-        let mut rng = rand::thread_rng();
-        let mut path = env::temp_dir();
-        path.push(format!("fuel-core-shallow-{}", rng.next_u64()));
-        Self { path }
-    }
-
-    /// Returns the path of the directory.
-    pub fn path(&self) -> &PathBuf {
-        &self.path
-    }
-}
-
-impl Drop for ShallowTempDir {
-    fn drop(&mut self) {
-        // Ignore errors
-        let _ = std::fs::remove_dir_all(&self.path);
-    }
-}
-
 type DropFn = Box<dyn FnOnce() + Send + Sync>;
 #[derive(Default)]
 struct DropResources {