Merge pull request #8 from moonbeam-foundation/tarekkma/pallet-parameters-account-weight

Author: librelois

Cargo.lock

@@ -18,6 +18,7 @@ frame-system.workspace = true
 sp-core.workspace = true
 sp-runtime.workspace = true
 frame-benchmarking = { optional = true, workspace = true }
+cumulus-primitives-storage-weight-reclaim = { workspace = true, default-features = false}
 
 [dev-dependencies]
 sp-core = { features = ["std"], default-features = true, path = "../../primitives/core" }
@@ -36,6 +37,7 @@ std = [
 	"serde",
 	"sp-core/std",
 	"sp-runtime/std",
+	"cumulus-primitives-storage-weight-reclaim/std",
 ]
 runtime-benchmarks = [
 	"frame-benchmarking/runtime-benchmarks",

substrate/frame/parameters/Cargo.toml

@@ -125,6 +125,7 @@ use frame_support::traits::{
 	dynamic_params::{AggregatedKeyValue, IntoKey, Key, RuntimeParameterStore, TryIntoKey},
 	EnsureOriginWithArg,
 };
+use cumulus_primitives_storage_weight_reclaim::get_proof_size;
 
 mod benchmarking;
 #[cfg(test)]
@@ -167,6 +168,22 @@ pub mod pallet {
 		type WeightInfo: WeightInfo;
 	}
 
+	#[pallet::hooks]
+	impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {
+		fn on_initialize(_: BlockNumberFor<T>) -> Weight {
+			let proof_size_before: u64 = get_proof_size().unwrap_or(0);
+
+			let items = Parameters::<T>::iter().count() as u64;
+
+			let proof_size_after: u64 = get_proof_size().unwrap_or(0);
+
+			let proof_size_diff = proof_size_after.saturating_sub(proof_size_before);
+
+			Weight::from_parts(0, proof_size_diff)
+				.saturating_add(T::DbWeight::get().reads(items))
+		}
+	}
+
 	#[pallet::event]
 	#[pallet::generate_deposit(pub(crate) fn deposit_event)]
 	pub enum Event<T: Config> {