feat: support single pallet check benchmark

centrifuge · Sep 20, 2023 · 3ed71fb · 3ed71fb
1 parent cfe3a84
commit 3ed71fb
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/scripts/check_benchmarks.sh b/scripts/check_benchmarks.sh
@@ -1,6 +1,7 @@
 set -eu
 
 runtime=$1
+pallet=$2
 
 run_benchmark() {
   pallet=$1
@@ -40,7 +41,16 @@ all_pallets=$(
   ./target/release/centrifuge-chain benchmark pallet --list --chain="${chain}" | tail -n+2 | cut -d',' -f1 | sort | uniq
 )
 
-for pallet in $all_pallets
-do
-    run_benchmark $pallet
-done
+if [ -n "$pallet" ];
+then
+  echo "Only benchmarking pallet: $pallet"
+  run_benchmark $pallet
+else
+  echo "Benchmarking all pallets"
+  for pallet in $all_pallets
+  do
+      run_benchmark $pallet
+  done
+fi
+
+