feat: forest-tool benchmark blockstore (#6331)

Author: hanabi1224

Cargo.lock

@@ -14,7 +14,7 @@ name = "forest"
 [workspace.dependencies]
 anyhow = "1"
 cid = { version = "0.11", default-features = false, features = ["std"] }
-flume = "0.11"
+flume = "0.12"
 futures = "0.3"
 get-size2 = { version = "0.7", features = ["derive", "hashbrown"] }
 hashlink = "0.11"
@@ -254,7 +254,7 @@ predicates = "3"
 proc-macro2 = { version = "1", default-features = false, features = ["span-locations"] }
 quickcheck = "1"
 quickcheck_macros = "1"
-ra_ap_syntax = "0.0.308"
+ra_ap_syntax = "0.0.309"
 regex-automata = "0.4"
 serial_test = "3"
 syn = { version = "2", default-features = false, features = ["full", "parsing", "visit", "printing", "extra-traits"] }

Cargo.toml

@@ -0,0 +1,73 @@
+// Copyright 2019-2025 ChainSafe Systems
+// SPDX-License-Identifier: Apache-2.0, MIT
+
+use super::*;
+
+pub enum Either<A: Blockstore, B: Blockstore> {
+    Left(A),
+    Right(B),
+}
+
+impl<A: Blockstore, B: Blockstore> Blockstore for Either<A, B> {
+    fn has(&self, k: &Cid) -> anyhow::Result<bool> {
+        match self {
+            Self::Left(v) => v.has(k),
+            Self::Right(v) => v.has(k),
+        }
+    }
+
+    #[allow(clippy::disallowed_types)]
+    fn put<D>(
+        &self,
+        mh_code: multihash_codetable::Code,
+        block: &fvm_ipld_blockstore::Block<D>,
+    ) -> anyhow::Result<Cid>
+    where
+        Self: Sized,
+        D: AsRef<[u8]>,
+    {
+        match self {
+            Self::Left(v) => v.put(mh_code, block),
+            Self::Right(v) => v.put(mh_code, block),
+        }
+    }
+
+    #[allow(clippy::disallowed_types)]
+    fn put_many<D, I>(&self, blocks: I) -> anyhow::Result<()>
+    where
+        Self: Sized,
+        D: AsRef<[u8]>,
+        I: IntoIterator<Item = (multihash_codetable::Code, fvm_ipld_blockstore::Block<D>)>,
+    {
+        match self {
+            Self::Left(v) => v.put_many(blocks),
+            Self::Right(v) => v.put_many(blocks),
+        }
+    }
+
+    fn put_many_keyed<D, I>(&self, blocks: I) -> anyhow::Result<()>
+    where
+        Self: Sized,
+        D: AsRef<[u8]>,
+        I: IntoIterator<Item = (Cid, D)>,
+    {
+        match self {
+            Self::Left(v) => v.put_many_keyed(blocks),
+            Self::Right(v) => v.put_many_keyed(blocks),
+        }
+    }
+
+    fn get(&self, k: &Cid) -> anyhow::Result<Option<Vec<u8>>> {
+        match self {
+            Self::Left(v) => v.get(k),
+            Self::Right(v) => v.get(k),
+        }
+    }
+
+    fn put_keyed(&self, k: &Cid, block: &[u8]) -> anyhow::Result<()> {
+        match self {
+            Self::Left(v) => v.put_keyed(k, block),
+            Self::Right(v) => v.put_keyed(k, block),
+        }
+    }
+}

src/db/either.rs

@@ -14,13 +14,15 @@ pub use blockstore_with_read_cache::*;
 pub use blockstore_with_write_buffer::BlockstoreWithWriteBuffer;
 pub use memory::MemoryDB;
 mod db_mode;
+mod either;
 pub mod migration;
+pub use either::Either;
 
 use crate::blocks::TipsetKey;
 use crate::rpc::eth::types::EthHash;
 use anyhow::{Context as _, bail};
 use cid::Cid;
-use fvm_ipld_blockstore::{Blockstore, MemoryBlockstore};
+pub use fvm_ipld_blockstore::{Blockstore, MemoryBlockstore};
 use serde::Serialize;
 use serde::de::DeserializeOwned;
 use std::sync::Arc;

src/db/mod.rs

@@ -679,7 +679,7 @@ async fn merge_f3_snapshot(filecoin: PathBuf, f3: PathBuf, output: PathBuf) -> a
     let f3_cid = crate::f3::snapshot::get_f3_snapshot_cid(&mut f3_data)?;
 
     let car_stream = CarStream::new_from_path(&filecoin).await?;
-    let chain_head = car_stream.header_v1.roots.clone();
+    let chain_head = car_stream.head_tipset_key().to_cids();
 
     println!("f3 snapshot cid: {f3_cid}");
     println!(