Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 9 additions & 8 deletions examples/advanced/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -13,17 +13,18 @@ repository.workspace = true
workspace = true

[dev-dependencies]
foundry-fork-db = "0.13"
alloy.workspace = true
# foundry-fork-db = { git = "https://github.com/foundry-rs/foundry-fork-db.git", rev = "cdbbe64" }
alloy-evm = "0.7.2"

# reth
# revm = { version = "22.0.1", default-features = false }
# revm-primitives = { version = "18.0.0", default-features = false }
# reth-db = { git = "https://github.com/paradigmxyz/reth", package = "reth-db", rev = "7029951" }
# reth-provider = { git = "https://github.com/paradigmxyz/reth", package = "reth-provider", rev = "7029951" }
# reth-node-types = { git = "https://github.com/paradigmxyz/reth", package = "reth-node-types", rev = "7029951" }
# reth-chainspec = { git = "https://github.com/paradigmxyz/reth", package = "reth-chainspec", rev = "7029951" }
# reth-node-ethereum = { git = "https://github.com/paradigmxyz/reth", package = "reth-node-ethereum", rev = "7029951" }
revm = { version = "23.1.0", default-features = false }
revm-primitives = { version = "19.0.0", default-features = false }
reth-db = { git = "https://github.com/paradigmxyz/reth", package = "reth-db", rev = "55f4b0b" }
reth-provider = { git = "https://github.com/paradigmxyz/reth", package = "reth-provider", rev = "55f4b0b" }
reth-node-types = { git = "https://github.com/paradigmxyz/reth", package = "reth-node-types", rev = "55f4b0b" }
reth-chainspec = { git = "https://github.com/paradigmxyz/reth", package = "reth-chainspec", rev = "55f4b0b" }
reth-node-ethereum = { git = "https://github.com/paradigmxyz/reth", package = "reth-node-ethereum", rev = "55f4b0b" }
Comment on lines +23 to +27
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we could also move this entirely to reth and just reference it here


eyre.workspace = true
tokio = { workspace = true, features = ["macros", "rt-multi-thread"] }
Expand Down
304 changes: 153 additions & 151 deletions examples/advanced/examples/foundry_fork_db.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,154 +6,156 @@
//!
//! `foundry_fork_db` serves as the backend for Foundry's forking functionality in Anvil and Forge.

// use std::sync::Arc;

// use alloy::{
// consensus::BlockHeader,
// eips::BlockId,
// network::{AnyNetwork, AnyRpcBlock, TransactionBuilder},
// node_bindings::Anvil,
// primitives::U256,
// providers::{Provider, ProviderBuilder},
// rpc::types::TransactionRequest,
// };
// use eyre::Result;
// use foundry_fork_db::{cache::BlockchainDbMeta, BlockchainDb, SharedBackend};
// use revm::{db::CacheDB, DatabaseRef, Evm};
// use revm_primitives::{BlobExcessGasAndPrice, BlockEnv, TxEnv};

// #[tokio::main]
// async fn main() -> Result<()> {
// let anvil = Anvil::new().spawn();
// let provider =
// ProviderBuilder::new().network::<AnyNetwork>().connect_http(anvil.endpoint_url());

// let block = provider.get_block(BlockId::latest()).await?.unwrap();

// // The `BlockchainDbMeta` is used a identifier when the db is flushed to the disk.
// // This aids in cases where the disk contains data from multiple forks.
// let meta = BlockchainDbMeta::default()
// .with_chain_id(31337)
// .with_block(&block.inner)
// .with_url(&anvil.endpoint());

// let db = BlockchainDb::new(meta, None);

// // Spawn the backend with the db instance.
// // `SharedBackend` is used to send request to the `BackendHandler` which is responsible for
// // filling missing data in the db, and also deduplicate requests that are being sent to the
// // RPC provider.
// //
// // For example, if we send two requests to get_full_block(0) simultaneously, the
// // `BackendHandler` is smart enough to only send one request to the RPC provider, and queue
// the // other request until the response is received.
// // Once the response from RPC provider is received it relays the response to both the
// requests // over their respective channels.
// //
// // The `SharedBackend` and `BackendHandler` communicate over an unbounded channel.
// let shared = SharedBackend::spawn_backend(Arc::new(provider.clone()), db, None).await;

// let start_t = std::time::Instant::now();
// let block_rpc = shared.get_full_block(0).unwrap();
// let time_rpc = start_t.elapsed();

// // `SharedBackend` is cloneable and holds the channel to the same `BackendHandler`.
// #[allow(clippy::redundant_clone)]
// let cloned_backend = shared.clone();

// // Block gets cached in the db
// let start_t = std::time::Instant::now();
// let block_cache = cloned_backend.get_full_block(0).unwrap();
// let time_cache = start_t.elapsed();

// assert_eq!(block_rpc, block_cache);

// println!("-------get_full_block--------");
// // The backend handle falls back to the RPC provider if the block is not in the cache.
// println!("1st request (via rpc): {:?}", time_rpc);
// // The block is cached due to the previous request and can be fetched from db.
// println!("2nd request (via fork db): {:?}\n", time_cache);

// let alice = anvil.addresses()[0];
// let bob = anvil.addresses()[1];

// let basefee = block.header.base_fee_per_gas.unwrap();

// let tx_req = TransactionRequest::default()
// .with_from(alice)
// .with_to(bob)
// .with_value(U256::from(100))
// .with_max_fee_per_gas(basefee as u128)
// .with_max_priority_fee_per_gas(basefee as u128 + 1)
// .with_gas_limit(21000)
// .with_nonce(0);

// let mut evm = configure_evm_env(block, shared.clone(), configure_tx_env(tx_req));

// // Fetches accounts from the RPC
// let start_t = std::time::Instant::now();
// let alice_bal = shared.basic_ref(alice)?.unwrap().balance;
// let bob_bal = shared.basic_ref(bob)?.unwrap().balance;
// let time_rpc = start_t.elapsed();

// let res = evm.transact().unwrap();

// let total_spent = U256::from(res.result.gas_used()) * U256::from(basefee) + U256::from(100);

// shared.data().do_commit(res.state);

// // Fetches accounts from the cache
// let start_t = std::time::Instant::now();
// let alice_bal_after = shared.basic_ref(alice)?.unwrap().balance;
// let bob_bal_after = shared.basic_ref(bob)?.unwrap().balance;
// let time_cache = start_t.elapsed();

// println!("-------get_account--------");
// println!("1st request (via rpc): {:?}", time_rpc);
// println!("2nd request (via fork db): {:?}\n", time_cache);

// assert_eq!(alice_bal_after, alice_bal - total_spent);
// assert_eq!(bob_bal_after, bob_bal + U256::from(100));

// Ok(())
// }

// fn configure_evm_env(
// block: AnyRpcBlock,
// shared: SharedBackend,
// tx_env: TxEnv,
// ) -> Evm<'static, (), CacheDB<SharedBackend>> {
// let basefee = block.header.base_fee_per_gas().map(U256::from).unwrap_or_default();
// let block_env = BlockEnv {
// number: U256::from(block.header.number()),
// coinbase: block.header.beneficiary(),
// timestamp: U256::from(block.header.timestamp()),
// gas_limit: U256::from(block.header.gas_limit()),
// basefee,
// prevrandao: block.header.mix_hash(),
// difficulty: block.header.difficulty(),
// blob_excess_gas_and_price: Some(BlobExcessGasAndPrice::new(
// block.header.excess_blob_gas().unwrap_or_default(),
// false,
// )),
// };

// let db = CacheDB::new(shared);

// let evm = Evm::builder().with_block_env(block_env).with_db(db).with_tx_env(tx_env).build();

// evm
// }

// fn configure_tx_env(tx_req: TransactionRequest) -> TxEnv {
// TxEnv {
// caller: tx_req.from.unwrap(),
// transact_to: tx_req.to.unwrap(),
// value: tx_req.value.unwrap(),
// gas_price: U256::from(tx_req.max_fee_per_gas.unwrap()),
// gas_limit: tx_req.gas.unwrap_or_default(),
// ..Default::default()
// }
// }

const fn main() {}
use std::sync::Arc;

use alloy::{
consensus::BlockHeader,
eips::BlockId,
network::{AnyNetwork, AnyRpcBlock, TransactionBuilder},
node_bindings::Anvil,
primitives::U256,
providers::{Provider, ProviderBuilder},
rpc::types::TransactionRequest,
};
use alloy_evm::{eth::EthEvmContext, EthEvm, Evm};
Copy link
Member Author

@zerosnacks zerosnacks May 12, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💯!

use eyre::Result;
use foundry_fork_db::{cache::BlockchainDbMeta, BlockchainDb, SharedBackend};
use revm::{
context::{BlockEnv, Evm as RevmEvm, TxEnv},
context_interface::block::BlobExcessGasAndPrice,
database::WrapDatabaseRef,
handler::{instructions::EthInstructions, EthPrecompiles},
inspector::NoOpInspector,
DatabaseRef,
};

#[tokio::main]
async fn main() -> Result<()> {
let anvil = Anvil::new().spawn();
let provider =
ProviderBuilder::new().network::<AnyNetwork>().connect_http(anvil.endpoint_url());

let block = provider.get_block(BlockId::latest()).await?.unwrap();

// The `BlockchainDbMeta` is used a identifier when the db is flushed to the disk.
// This aids in cases where the disk contains data from multiple forks.
let meta = BlockchainDbMeta::default().with_block(&block.inner).with_url(&anvil.endpoint());

let db = BlockchainDb::new(meta, None);

// Spawn the backend with the db instance.
// `SharedBackend` is used to send request to the `BackendHandler` which is responsible for
// filling missing data in the db, and also deduplicate requests that are being sent to the
// RPC provider.
//
// For example, if we send two requests to get_full_block(0) simultaneously, the
// `BackendHandler` is smart enough to only send one request to the RPC provider, and queue the
// other request until the response is received. Once the response from RPC provider is
// received it relays the response to both the requests // over their respective channels.
//
// The `SharedBackend` and `BackendHandler` communicate over an unbounded channel.
let shared = SharedBackend::spawn_backend(Arc::new(provider.clone()), db, None).await;

let start_t = std::time::Instant::now();
let block_rpc = shared.get_full_block(0).unwrap();
let time_rpc = start_t.elapsed();

// `SharedBackend` is cloneable and holds the channel to the same `BackendHandler`.
#[allow(clippy::redundant_clone)]
let cloned_backend = shared.clone();

// Block gets cached in the db
let start_t = std::time::Instant::now();
let block_cache = cloned_backend.get_full_block(0).unwrap();
let time_cache = start_t.elapsed();

assert_eq!(block_rpc, block_cache);

println!("-------get_full_block--------");
// The backend handle falls back to the RPC provider if the block is not in the cache.
println!("1st request (via rpc): {:?}", time_rpc);
// The block is cached due to the previous request and can be fetched from db.
println!("2nd request (via fork db): {:?}\n", time_cache);

let alice = anvil.addresses()[0];
let bob = anvil.addresses()[1];

let basefee = block.header.base_fee_per_gas.unwrap();

let tx_req = TransactionRequest::default()
.with_from(alice)
.with_to(bob)
.with_value(U256::from(100))
.with_max_fee_per_gas(basefee as u128)
.with_max_priority_fee_per_gas(basefee as u128 + 1)
.with_gas_limit(21000)
.with_nonce(0);

let mut evm = configure_evm(block, shared.clone());

// Fetches accounts from the RPC
let start_t = std::time::Instant::now();
let alice_bal = shared.basic_ref(alice)?.unwrap().balance;
let bob_bal = shared.basic_ref(bob)?.unwrap().balance;
let time_rpc = start_t.elapsed();

let res = evm.transact(configure_tx_env(tx_req)).unwrap();

let total_spent = U256::from(res.result.gas_used()) * U256::from(basefee) + U256::from(100);

shared.data().do_commit(res.state);

// Fetches accounts from the cache
let start_t = std::time::Instant::now();
let alice_bal_after = shared.basic_ref(alice)?.unwrap().balance;
let bob_bal_after = shared.basic_ref(bob)?.unwrap().balance;
let time_cache = start_t.elapsed();

println!("-------get_account--------");
println!("1st request (via rpc): {:?}", time_rpc);
println!("2nd request (via fork db): {:?}\n", time_cache);

assert_eq!(alice_bal_after, alice_bal - total_spent);
assert_eq!(bob_bal_after, bob_bal + U256::from(100));

Ok(())
}

fn configure_evm(
block: AnyRpcBlock,
shared: SharedBackend,
) -> EthEvm<WrapDatabaseRef<SharedBackend>, NoOpInspector> {
let block_env = BlockEnv {
number: block.header.number(),
beneficiary: block.header.beneficiary(),
timestamp: block.header.timestamp(),
gas_limit: block.header.gas_limit(),
basefee: block.header.base_fee_per_gas().unwrap_or(0),
prevrandao: block.header.mix_hash(),
difficulty: block.header.difficulty(),
blob_excess_gas_and_price: Some(BlobExcessGasAndPrice::new(
block.header.excess_blob_gas().unwrap_or_default(),
false,
)),
};

let context =
EthEvmContext::new(WrapDatabaseRef(shared), revm_primitives::hardfork::SpecId::PRAGUE)
.with_block(block_env);

let evm = RevmEvm::new(context, EthInstructions::default(), EthPrecompiles::default())
.with_inspector(NoOpInspector);

EthEvm::new(evm, false)
}

fn configure_tx_env(tx_req: TransactionRequest) -> TxEnv {
TxEnv {
caller: tx_req.from.unwrap(),
kind: tx_req.to.unwrap(),
value: tx_req.value.unwrap(),
gas_price: tx_req.max_fee_per_gas.unwrap(),
gas_limit: tx_req.gas.unwrap_or_default(),
..Default::default()
}
}
Loading