Reduce DB creation

Author: kvinwang

crates/phactory/src/contracts/pink.rs

@@ -391,7 +391,7 @@ impl OCalls for RuntimeHandleMut<'_> {
         if !context::get().mode.is_transaction() {
             return;
         }
-        warn!("commit state root: {:?}", root);
+        info!("commit state root: {:?}", root);
         self.cluster.storage.commit(root, changes);
     }
 

crates/phala-trie-storage/src/kvdb/database.rs

@@ -128,6 +128,7 @@ pub(crate) fn with_cache_dir<T>(cache_dir: &str, f: impl FnOnce() -> T) -> T {
 
 pub(crate) fn create_db() -> (TransactionDB<MultiThreaded>, usize) {
     let test_path = test_cached_path::with(|path| path.clone());
+    // The PRUNTIME_TRIE_CACHE_PATH would be hardcoded in the manifest when running in gramine.
     let cache_path = &test_path
         .or_else(|| std::env::var("PRUNTIME_TRIE_CACHE_PATH").ok())
         .unwrap_or_else(|| "data/protected_files/caches".to_string());

crates/phala-trie-storage/src/lib.rs

@@ -173,7 +173,9 @@ where
 
     /// Apply storage changes calculated from `calc_root_if_changes`.
     pub fn apply_changes(&mut self, root: H::Out, transaction: MemoryDB<H>) {
-        let mut storage = core::mem::take(self).0.into_storage();
+        let mut storage = core::mem::replace(self, Self::default_memdb())
+            .0
+            .into_storage();
         storage.consolidate_mdb(transaction);
         let _ = core::mem::replace(&mut self.0, TrieBackendBuilder::new(storage, root).build());
     }
@@ -214,7 +216,9 @@ where
     }
 
     pub fn set_root(&mut self, root: H::Out) {
-        let storage = core::mem::take(self).0.into_storage();
+        let storage = core::mem::replace(self, Self::default_memdb())
+            .0
+            .into_storage();
         let _ = core::mem::replace(&mut self.0, TrieBackendBuilder::new(storage, root).build());
     }