feat: add hooks for utility batch

cumulus/parachains/runtimes/assets/asset-hub-rococo/src/lib.rs

@@ -473,6 +473,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/assets/asset-hub-westend/src/lib.rs

@@ -472,6 +472,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/bridge-hubs/bridge-hub-rococo/src/lib.rs

@@ -543,6 +543,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/bridge-hubs/bridge-hub-westend/src/lib.rs

@@ -519,6 +519,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/collectives/collectives-westend/src/lib.rs

@@ -264,6 +264,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/contracts/contracts-rococo/src/lib.rs

@@ -274,6 +274,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = pallet_utility::weights::SubstrateWeight<Runtime>;
 }
 

cumulus/parachains/runtimes/coretime/coretime-rococo/src/lib.rs

@@ -583,6 +583,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/coretime/coretime-westend/src/lib.rs

@@ -584,6 +584,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+    type BatchPreHook = ();
+    type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/people/people-rococo/src/lib.rs

@@ -526,6 +526,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

cumulus/parachains/runtimes/people/people-westend/src/lib.rs

@@ -525,6 +525,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

polkadot/runtime/rococo/src/lib.rs

@@ -748,6 +748,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

polkadot/runtime/westend/src/lib.rs

@@ -985,6 +985,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = weights::pallet_utility::WeightInfo<Runtime>;
 }
 

substrate/bin/node/runtime/src/lib.rs

@@ -374,6 +374,8 @@ impl pallet_utility::Config for Runtime {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = pallet_utility::weights::SubstrateWeight<Runtime>;
 }
 

substrate/frame/utility/README.md

@@ -22,6 +22,9 @@ This module contains two basic pieces of functionality:
 Since proxy filters are respected in all dispatches of this module, it should never need to be
 filtered by any proxy.
 
+Pre- and post-hooks can be configured via `BatchPostHook` and `BatchPostHook` associated types. 
+They are triggered for each batch call: `batch`, `batch_all`, and `force_batch`. Use the unit type `()` if no behavior is required.
+
 ## Interface
 
 ### Dispatchable Functions

substrate/frame/utility/src/lib.rs

@@ -68,9 +68,28 @@ use sp_core::TypeId;
 use sp_io::hashing::blake2_256;
 use sp_runtime::traits::{BadOrigin, Dispatchable, TrailingZeroInput};
 pub use weights::WeightInfo;
-
 pub use pallet::*;
 
+pub trait BatchPreHook {
+	fn on_batch_start() -> sp_runtime::DispatchResult;
+}
+
+pub trait BatchPostHook {
+	fn on_batch_end() -> sp_runtime::DispatchResult;
+}
+
+impl BatchPreHook for () {
+	fn on_batch_start() -> sp_runtime::DispatchResult {
+		Ok(())
+	}
+}
+
+impl BatchPostHook for () {
+	fn on_batch_end() -> sp_runtime::DispatchResult {
+		Ok(())
+	}
+}
+
 #[frame_support::pallet]
 pub mod pallet {
 	use super::*;
@@ -100,6 +119,12 @@ pub mod pallet {
 			Into<<Self as frame_system::Config>::RuntimeOrigin> +
 			IsType<<<Self as frame_system::Config>::RuntimeOrigin as frame_support::traits::OriginTrait>::PalletsOrigin>;
 
+		///Hook to be called before any batch operation
+		type BatchPreHook: BatchPreHook;
+
+		///Hook to be called after any batch operation
+		type BatchPostHook: BatchPostHook;
+
 		/// Weight information for extrinsics in this pallet.
 		type WeightInfo: WeightInfo;
 	}
@@ -196,6 +221,8 @@ pub mod pallet {
 				return Err(BadOrigin.into())
 			}
 
+			T::BatchPreHook::on_batch_start()?;
+
 			let is_root = ensure_root(origin.clone()).is_ok();
 			let calls_len = calls.len();
 			ensure!(calls_len <= Self::batched_calls_limit() as usize, Error::<T>::TooManyCalls);
@@ -225,6 +252,9 @@ pub mod pallet {
 				Self::deposit_event(Event::ItemCompleted);
 			}
 			Self::deposit_event(Event::BatchCompleted);
+
+			T::BatchPostHook::on_batch_end()?;
+
 			let base_weight = T::WeightInfo::batch(calls_len as u32);
 			Ok(Some(base_weight.saturating_add(weight)).into())
 		}
@@ -305,6 +335,8 @@ pub mod pallet {
 				return Err(BadOrigin.into())
 			}
 
+			T::BatchPreHook::on_batch_start()?;
+
 			let is_root = ensure_root(origin.clone()).is_ok();
 			let calls_len = calls.len();
 			ensure!(calls_len <= Self::batched_calls_limit() as usize, Error::<T>::TooManyCalls);
@@ -339,6 +371,9 @@ pub mod pallet {
 				Self::deposit_event(Event::ItemCompleted);
 			}
 			Self::deposit_event(Event::BatchCompleted);
+
+			T::BatchPostHook::on_batch_end()?;
+
 			let base_weight = T::WeightInfo::batch_all(calls_len as u32);
 			Ok(Some(base_weight.saturating_add(weight)).into())
 		}
@@ -401,6 +436,8 @@ pub mod pallet {
 				return Err(BadOrigin.into())
 			}
 
+			T::BatchPreHook::on_batch_start()?;
+
 			let is_root = ensure_root(origin.clone()).is_ok();
 			let calls_len = calls.len();
 			ensure!(calls_len <= Self::batched_calls_limit() as usize, Error::<T>::TooManyCalls);
@@ -431,6 +468,9 @@ pub mod pallet {
 			} else {
 				Self::deposit_event(Event::BatchCompleted);
 			}
+
+			T::BatchPostHook::on_batch_end()?;
+
 			let base_weight = T::WeightInfo::batch(calls_len as u32);
 			Ok(Some(base_weight.saturating_add(weight)).into())
 		}

substrate/frame/utility/src/tests.rs

@@ -222,6 +222,8 @@ impl Config for Test {
 	type RuntimeEvent = RuntimeEvent;
 	type RuntimeCall = RuntimeCall;
 	type PalletsOrigin = OriginCaller;
+	type BatchPreHook = ();
+	type BatchPostHook = ();
 	type WeightInfo = ();
 }