Use db reads/writes as weights

Moonsong-Labs · Mar 22, 2024 · 33fd351 · 33fd351
1 parent b2e77f4
commit 33fd351
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/pallets/emergency-para-xcm/Cargo.toml b/pallets/emergency-para-xcm/Cargo.toml
@@ -29,6 +29,9 @@ cumulus-pallet-parachain-system = { workspace = true }
 # Polkadot
 polkadot-parachain-primitives = { workspace = true }
 
+# Benchmarks
+frame-benchmarking = { workspace = true, optional = true }
+
 [dev-dependencies]
 sp-core = { workspace = true, features = [ "std" ] }
 sp-io = { workspace = true, features = [ "std" ] }
@@ -37,6 +40,7 @@ substrate-test-runtime-client = { workspace = true }
 [features]
 default = [ "std" ]
 std = [
+	"frame-benchmarking/std",
 	"cumulus-primitives-core/std",
 	"cumulus-pallet-parachain-system/std",
 	"frame-support/std",
@@ -48,4 +52,9 @@ std = [
 	"sp-runtime/std",
 	"sp-std/std",
 ]
+
+runtime-benchmarks = [
+	"frame-benchmarking",
+]
+
 try-runtime = [ "frame-support/try-runtime" ]
diff --git a/pallets/emergency-para-xcm/src/lib.rs b/pallets/emergency-para-xcm/src/lib.rs
@@ -115,7 +115,7 @@ pub mod pallet {
 	#[pallet::call]
 	impl<T: Config> Pallet<T> {
 		#[pallet::call_index(0)]
-		#[pallet::weight((0, DispatchClass::Operational))]
+		#[pallet::weight((T::DbWeight::get().reads_writes(1, 2), DispatchClass::Operational))]
 		pub fn paused_to_normal(origin: OriginFor<T>) -> DispatchResultWithPostInfo {
 			T::PausedToNormalOrigin::ensure_origin(origin)?;
 
@@ -131,7 +131,7 @@ pub mod pallet {
 		}
 
 		#[pallet::call_index(1)]
-		#[pallet::weight((1_000_000, DispatchClass::Operational))]
+		#[pallet::weight((T::DbWeight::get().read, DispatchClass::Operational))]
 		pub fn fast_authorize_upgrade(origin: OriginFor<T>, code_hash: T::Hash) -> DispatchResult {
 			T::FastAuthorizeUpgradeOrigin::ensure_origin(origin)?;
 			ensure!(