revme/cmd/bench/
transfer.rs1use context::{ContextTr, TxEnv};
2use criterion::Criterion;
3use database::{BenchmarkDB, BENCH_CALLER, BENCH_TARGET, BENCH_TARGET_BALANCE};
4use revm::{
5 bytecode::Bytecode,
6 context_interface::JournalTr,
7 primitives::{TxKind, U256},
8 Context, ExecuteEvm, MainBuilder, MainContext,
9};
10
11pub fn run(criterion: &mut Criterion) {
12 let mut evm = Context::mainnet()
13 .with_db(BenchmarkDB::new_bytecode(Bytecode::new()))
14 .modify_cfg_chained(|cfg| cfg.disable_nonce_check = true)
15 .build_mainnet();
16
17 let tx = TxEnv::builder()
18 .caller(BENCH_CALLER)
19 .kind(TxKind::Call(BENCH_TARGET))
20 .value(U256::from(1))
21 .gas_price(1)
22 .gas_priority_fee(None)
23 .build()
24 .unwrap();
25
26 evm.ctx.tx = tx.clone();
27
28 let mut i = 0;
29 criterion.bench_function("transfer", |b| {
30 b.iter_batched(
31 || tx.clone(),
32 |input| {
33 i += 1;
34 evm.transact_one(input).unwrap()
35 },
36 criterion::BatchSize::SmallInput,
37 );
38 });
39
40 let balance = evm
41 .journal_mut()
42 .load_account(BENCH_TARGET)
43 .unwrap()
44 .data
45 .info
46 .balance;
47
48 if balance != BENCH_TARGET_BALANCE + U256::from(i) {
49 panic!("balance of transfers is not correct");
50 }
51
52 let _ = evm.finalize();
54
55 evm.modify_cfg(|cfg| cfg.disable_nonce_check = false);
56
57 criterion.bench_function("transfer_finalize", |b| b.iter(|| evm.replay().unwrap()));
58}