Skip to content

Commit

Permalink
Add storage root recalculation time to benchmarks (paritytech#5035)
Browse files Browse the repository at this point in the history
  • Loading branch information
shawntabrizi authored Feb 24, 2020
1 parent ddfa359 commit 41bb219
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 8 deletions.
17 changes: 12 additions & 5 deletions frame/benchmarking/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,8 @@

mod utils;
pub use utils::*;
#[doc(hidden)]
pub use sp_io::storage::root as storage_root;

/// Construct pallet benchmarks for weighing dispatchables.
///
Expand Down Expand Up @@ -177,12 +179,17 @@ macro_rules! impl_benchmark {
// Commit the externalities to the database, flushing the DB cache.
// This will enable worst case scenario for reading from the database.
$crate::benchmarking::commit_db();
// Run the benchmark.
let start = $crate::benchmarking::current_time();
// Time the extrinsic logic.
let start_extrinsic = $crate::benchmarking::current_time();
call.dispatch(caller.into())?;
let finish = $crate::benchmarking::current_time();
let elapsed = finish - start;
results.push((c.clone(), elapsed));
let finish_extrinsic = $crate::benchmarking::current_time();
let elapsed_extrinsic = finish_extrinsic - start_extrinsic;
// Time the storage root recalculation.
let start_storage_root = $crate::benchmarking::current_time();
$crate::storage_root();
let finish_storage_root = $crate::benchmarking::current_time();
let elapsed_storage_root = finish_storage_root - start_storage_root;
results.push((c.clone(), elapsed_extrinsic, elapsed_storage_root));
// Wipe the DB back to the genesis state.
$crate::benchmarking::wipe_db();
}
Expand Down
2 changes: 1 addition & 1 deletion frame/benchmarking/src/utils.rs
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ pub enum BenchmarkParameter {
/// Results from running benchmarks on a FRAME pallet.
/// Contains duration of the function call in nanoseconds along with the benchmark parameters
/// used for that benchmark result.
pub type BenchmarkResults = (Vec<(BenchmarkParameter, u32)>, u128);
pub type BenchmarkResults = (Vec<(BenchmarkParameter, u32)>, u128, u128);

sp_api::decl_runtime_apis! {
/// Runtime api for benchmarking a FRAME runtime.
Expand Down
5 changes: 3 additions & 2 deletions utils/frame/benchmarking-cli/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -124,12 +124,13 @@ impl BenchmarkCmd {
// Print the table header
results[0].0.iter().for_each(|param| print!("{:?},", param.0));

print!("time\n");
print!("extrinsic_time,storage_root_time\n");
// Print the values
results.iter().for_each(|result| {
let parameters = &result.0;
parameters.iter().for_each(|param| print!("{:?},", param.1));
print!("{:?}\n", result.1);
// Print extrinsic time and storage root time
print!("{:?},{:?}\n", result.1, result.2);
});

eprintln!("Done.");
Expand Down

0 comments on commit 41bb219

Please sign in to comment.