55 lines
1.7 KiB
Rust
55 lines
1.7 KiB
Rust
extern crate log;
|
|
use {
|
|
clap::{crate_description, crate_name, value_t, App, Arg},
|
|
solana_accounts_db::accounts_hash::AccountsHasher,
|
|
solana_measure::measure::Measure,
|
|
solana_sdk::{hash::Hash, pubkey::Pubkey},
|
|
};
|
|
|
|
fn main() {
|
|
solana_logger::setup();
|
|
|
|
let matches = App::new(crate_name!())
|
|
.about(crate_description!())
|
|
.version(solana_version::version!())
|
|
.arg(
|
|
Arg::with_name("num_accounts")
|
|
.long("num_accounts")
|
|
.takes_value(true)
|
|
.value_name("NUM_ACCOUNTS")
|
|
.help("Total number of accounts"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("iterations")
|
|
.long("iterations")
|
|
.takes_value(true)
|
|
.value_name("ITERATIONS")
|
|
.help("Number of bench iterations"),
|
|
)
|
|
.get_matches();
|
|
|
|
let num_accounts = value_t!(matches, "num_accounts", usize).unwrap_or(10_000);
|
|
let iterations = value_t!(matches, "iterations", usize).unwrap_or(20);
|
|
let hashes: Vec<_> = (0..num_accounts)
|
|
.map(|_| (Pubkey::new_unique(), Hash::new_unique()))
|
|
.collect();
|
|
let elapsed: Vec<_> = (0..iterations)
|
|
.map(|_| {
|
|
let hashes = hashes.clone(); // done outside timing
|
|
let mut time = Measure::start("compute_merkle_root");
|
|
let fanout = 16;
|
|
AccountsHasher::compute_merkle_root(hashes, fanout);
|
|
time.stop();
|
|
time.as_us()
|
|
})
|
|
.collect();
|
|
|
|
for result in &elapsed {
|
|
println!("compute_merkle_root(us),{result}");
|
|
}
|
|
println!(
|
|
"compute_merkle_root(us) avg: {}",
|
|
elapsed.into_iter().sum::<u64>() as f64 / iterations as f64
|
|
);
|
|
}
|