Skip to content

Commit

Permalink
bench/cli: extend --extrinsic name filtering (paritytech#10677)
Browse files Browse the repository at this point in the history
* implement benchmarking multiple extrinsics

* ran cargo +nightly fmt

* remove contains check and made it an equals check
  • Loading branch information
dharjeezy authored and grishasobol committed Mar 28, 2022
1 parent 7778501 commit 100af48
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions utils/frame/benchmarking-cli/src/command.rs
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,8 @@ impl BenchmarkCmd {
let pallet = self.pallet.clone().unwrap_or_else(|| String::new());
let pallet = pallet.as_bytes();
let extrinsic = self.extrinsic.clone().unwrap_or_else(|| String::new());
let extrinsic = extrinsic.as_bytes();
let extrinsic_split: Vec<&str> = extrinsic.split(',').collect();
let extrinsics: Vec<_> = extrinsic_split.iter().map(|x| x.trim().as_bytes()).collect();

let genesis_storage = spec.build_storage()?;
let mut changes = Default::default();
Expand Down Expand Up @@ -176,7 +177,10 @@ impl BenchmarkCmd {
.filter(|item| pallet.is_empty() || pallet == &b"*"[..] || pallet == &item.pallet[..])
.for_each(|item| {
for benchmark in &item.benchmarks {
if extrinsic.is_empty() || extrinsic == &b"*"[..] || extrinsic == benchmark.name
let benchmark_name = &benchmark.name;
if extrinsic.is_empty() ||
extrinsic.as_bytes() == &b"*"[..] ||
extrinsics.contains(&&benchmark_name[..])
{
benchmarks_to_run.push((
item.pallet.clone(),
Expand Down

0 comments on commit 100af48

Please sign in to comment.