feat(tool): forest-tool api generate-test-snapshot

src/chain/store/index.rs

@@ -7,6 +7,7 @@ use crate::beacon::{BeaconEntry, IGNORE_DRAND_VAR};
 use crate::blocks::{Tipset, TipsetKey};
 use crate::metrics;
 use crate::shim::clock::ChainEpoch;
+use crate::utils::misc::env::is_env_truthy;
 use fvm_ipld_blockstore::Blockstore;
 use itertools::Itertools;
 use lru::LruCache;
@@ -47,11 +48,13 @@ impl<DB: Blockstore> ChainIndex<DB> {
     /// Loads a tipset from memory given the tipset keys and cache. Semantically
     /// identical to [`Tipset::load`] but the result is cached.
     pub fn load_tipset(&self, tsk: &TipsetKey) -> Result<Option<Arc<Tipset>>, Error> {
-        if let Some(ts) = self.ts_cache.lock().get(tsk) {
-            metrics::LRU_CACHE_HIT
-                .get_or_create(&metrics::values::TIPSET)
-                .inc();
-            return Ok(Some(ts.clone()));
+        if !is_env_truthy("FOREST_TIPSET_CACHE_DISABLED") {
+            if let Some(ts) = self.ts_cache.lock().get(tsk) {
+                metrics::LRU_CACHE_HIT
+                    .get_or_create(&metrics::values::TIPSET)
+                    .inc();
+                return Ok(Some(ts.clone()));
+            }
         }
 
         let ts_opt = Tipset::load(&self.db, tsk)?.map(Arc::new);

src/db/memory.rs

@@ -1,6 +1,7 @@
 // Copyright 2019-2024 ChainSafe Systems
 // SPDX-License-Identifier: Apache-2.0, MIT
 
+use super::{EthMappingsStore, SettingsStore};
 use crate::cid_collections::CidHashSet;
 use crate::db::{GarbageCollectable, PersistentStore};
 use crate::libp2p_bitswap::{BitswapStoreRead, BitswapStoreReadWrite};
@@ -11,8 +12,7 @@ use cid::Cid;
 use fvm_ipld_blockstore::Blockstore;
 use itertools::Itertools;
 use parking_lot::RwLock;
-
-use super::{EthMappingsStore, SettingsStore};
+use std::ops::Deref;
 
 #[derive(Debug, Default)]
 pub struct MemoryDB {
@@ -22,6 +22,33 @@ pub struct MemoryDB {
     eth_mappings_db: RwLock<HashMap<EthHash, Vec<u8>>>,
 }
 
+impl MemoryDB {
+    pub fn serialize(&self) -> anyhow::Result<Vec<u8>> {
+        let blockchain_db = self.blockchain_db.read();
+        let blockchain_persistent_db = self.blockchain_persistent_db.read();
+        let settings_db = self.settings_db.read();
+        let eth_mappings_db = self.eth_mappings_db.read();
+        let tuple = (
+            blockchain_db.deref(),
+            blockchain_persistent_db.deref(),
+            settings_db.deref(),
+            eth_mappings_db.deref(),
+        );
+        Ok(fvm_ipld_encoding::to_vec(&tuple)?)
+    }
+
+    pub fn deserialize_from(bytes: &[u8]) -> anyhow::Result<Self> {
+        let (blockchain_db, blockchain_persistent_db, settings_db, eth_mappings_db) =
+            fvm_ipld_encoding::from_slice(bytes)?;
+        Ok(Self {
+            blockchain_db: RwLock::new(blockchain_db),
+            blockchain_persistent_db: RwLock::new(blockchain_persistent_db),
+            settings_db: RwLock::new(settings_db),
+            eth_mappings_db: RwLock::new(eth_mappings_db),
+        })
+    }
+}
+
 impl GarbageCollectable<CidHashSet> for MemoryDB {
     fn get_keys(&self) -> anyhow::Result<CidHashSet> {
         let mut set = CidHashSet::new();