revme/cmd/bench/
transfer.rs

1use 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    // drop the journal
53    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}