2020-09-11 09:48:06 -07:00
|
|
|
use clap::{crate_description, crate_name, value_t, App, Arg};
|
2020-03-17 11:02:07 -07:00
|
|
|
use rayon::prelude::*;
|
|
|
|
use solana_measure::measure::Measure;
|
2020-04-26 19:07:03 -07:00
|
|
|
use solana_runtime::{
|
|
|
|
accounts::{create_test_accounts, update_accounts, Accounts},
|
|
|
|
accounts_index::Ancestors,
|
|
|
|
};
|
2020-09-08 07:55:09 -07:00
|
|
|
use solana_sdk::{genesis_config::ClusterType, pubkey::Pubkey};
|
2020-09-11 09:48:06 -07:00
|
|
|
use std::env;
|
2020-03-17 11:02:07 -07:00
|
|
|
use std::fs;
|
|
|
|
use std::path::PathBuf;
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
solana_logger::setup();
|
|
|
|
|
2020-09-11 09:48:06 -07:00
|
|
|
let matches = App::new(crate_name!())
|
|
|
|
.about(crate_description!())
|
|
|
|
.version(solana_version::version!())
|
2020-03-17 11:02:07 -07:00
|
|
|
.arg(
|
|
|
|
Arg::with_name("num_slots")
|
|
|
|
.long("num_slots")
|
|
|
|
.takes_value(true)
|
|
|
|
.value_name("SLOTS")
|
|
|
|
.help("Number of slots to store to."),
|
|
|
|
)
|
|
|
|
.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"),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("clean")
|
|
|
|
.long("clean")
|
|
|
|
.takes_value(false)
|
|
|
|
.help("Run clean"),
|
|
|
|
)
|
|
|
|
.get_matches();
|
|
|
|
|
|
|
|
let num_slots = value_t!(matches, "num_slots", usize).unwrap_or(4);
|
|
|
|
let num_accounts = value_t!(matches, "num_accounts", usize).unwrap_or(10_000);
|
|
|
|
let iterations = value_t!(matches, "iterations", usize).unwrap_or(20);
|
|
|
|
let clean = matches.is_present("clean");
|
|
|
|
println!("clean: {:?}", clean);
|
|
|
|
|
2020-09-11 09:48:06 -07:00
|
|
|
let path = PathBuf::from(env::var("FARF_DIR").unwrap_or_else(|_| "farf".to_owned()))
|
|
|
|
.join("accounts-bench");
|
2020-03-17 11:02:07 -07:00
|
|
|
if fs::remove_dir_all(path.clone()).is_err() {
|
|
|
|
println!("Warning: Couldn't remove {:?}", path);
|
|
|
|
}
|
2020-09-08 07:55:09 -07:00
|
|
|
let accounts = Accounts::new(vec![path], &ClusterType::Testnet);
|
2020-03-17 11:02:07 -07:00
|
|
|
println!("Creating {} accounts", num_accounts);
|
|
|
|
let mut create_time = Measure::start("create accounts");
|
|
|
|
let pubkeys: Vec<_> = (0..num_slots)
|
|
|
|
.into_par_iter()
|
|
|
|
.map(|slot| {
|
|
|
|
let mut pubkeys: Vec<Pubkey> = vec![];
|
|
|
|
create_test_accounts(
|
|
|
|
&accounts,
|
|
|
|
&mut pubkeys,
|
|
|
|
num_accounts / num_slots,
|
|
|
|
slot as u64,
|
|
|
|
);
|
|
|
|
pubkeys
|
|
|
|
})
|
|
|
|
.collect();
|
|
|
|
let pubkeys: Vec<_> = pubkeys.into_iter().flatten().collect();
|
|
|
|
create_time.stop();
|
|
|
|
println!(
|
|
|
|
"created {} accounts in {} slots {}",
|
|
|
|
(num_accounts / num_slots) * num_slots,
|
|
|
|
num_slots,
|
|
|
|
create_time
|
|
|
|
);
|
2020-04-26 19:07:03 -07:00
|
|
|
let mut ancestors: Ancestors = vec![(0, 0)].into_iter().collect();
|
2020-03-17 11:02:07 -07:00
|
|
|
for i in 1..num_slots {
|
|
|
|
ancestors.insert(i as u64, i - 1);
|
|
|
|
accounts.add_root(i as u64);
|
|
|
|
}
|
|
|
|
for x in 0..iterations {
|
|
|
|
if clean {
|
|
|
|
let mut time = Measure::start("clean");
|
2020-09-28 16:04:46 -07:00
|
|
|
accounts.accounts_db.clean_accounts(None);
|
2020-03-17 11:02:07 -07:00
|
|
|
time.stop();
|
|
|
|
println!("{}", time);
|
|
|
|
for slot in 0..num_slots {
|
|
|
|
update_accounts(&accounts, &pubkeys, ((x + 1) * num_slots + slot) as u64);
|
|
|
|
accounts.add_root((x * num_slots + slot) as u64);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
let mut pubkeys: Vec<Pubkey> = vec![];
|
|
|
|
let mut time = Measure::start("hash");
|
2020-09-11 09:48:06 -07:00
|
|
|
let hash = accounts.accounts_db.update_accounts_hash(0, &ancestors).0;
|
2020-03-17 11:02:07 -07:00
|
|
|
time.stop();
|
|
|
|
println!("hash: {} {}", hash, time);
|
|
|
|
create_test_accounts(&accounts, &mut pubkeys, 1, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|