update weights calls

galacticcouncil · Jul 5, 2023 · 2ed11b8 · 2ed11b8
1 parent 3c038a0
commit 2ed11b8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 11 deletions.
diff --git a/pallets/stableswap/src/lib.rs b/pallets/stableswap/src/lib.rs
@@ -368,7 +368,7 @@ pub mod pallet {
 		///
 		/// Emits `FeesUpdated` event if successful.
 		#[pallet::call_index(1)]
-		#[pallet::weight(<T as Config>::WeightInfo::update_pool())]
+		#[pallet::weight(<T as Config>::WeightInfo::update_pool_fees())]
 		#[transactional]
 		pub fn update_pool_fees(
 			origin: OriginFor<T>,
@@ -407,7 +407,7 @@ pub mod pallet {
 		///
 		/// Emits `AmplificationUpdated` event if successful.
 		#[pallet::call_index(2)]
-		#[pallet::weight(<T as Config>::WeightInfo::update_pool())]
+		#[pallet::weight(<T as Config>::WeightInfo::update_amplification())]
 		#[transactional]
 		pub fn update_amplification(
 			origin: OriginFor<T>,

diff --git a/pallets/stableswap/src/weights.rs b/pallets/stableswap/src/weights.rs
@@ -46,12 +46,13 @@ use sp_std::marker::PhantomData;
 /// Weight functions needed for pallet_stableswap.
 pub trait WeightInfo {
 	fn create_pool() -> Weight;
-	fn update_pool() -> Weight;
 	fn add_liquidity() -> Weight;
 	fn remove_liquidity_one_asset() -> Weight;
 	fn sell() -> Weight;
 	fn buy() -> Weight;
 	fn set_asset_tradable_state() -> Weight;
+	fn update_pool_fees() -> Weight;
+	fn update_amplification() -> Weight;
 }
 
 pub struct BasiliskWeight<T>(PhantomData<T>);
@@ -63,10 +64,6 @@ impl<T: frame_system::Config> WeightInfo for BasiliskWeight<T> {
 			.saturating_add(T::DbWeight::get().writes(12 as u64))
 	}
 
-	fn update_pool() -> Weight {
-		Weight::from_ref_time(0u128 as u64)
-	}
-
 	fn add_liquidity() -> Weight {
 		Weight::from_ref_time(64_481_000 as u64)
 			.saturating_add(T::DbWeight::get().reads(10 as u64))
@@ -87,10 +84,15 @@ impl<T: frame_system::Config> WeightInfo for BasiliskWeight<T> {
 			.saturating_add(T::DbWeight::get().reads(9 as u64))
 			.saturating_add(T::DbWeight::get().writes(5 as u64))
 	}
-
 	fn set_asset_tradable_state() -> Weight {
 		Weight::from_ref_time(0)
 	}
+	fn update_pool_fees() -> Weight {
+		Weight::from_ref_time(0)
+	}
+	fn update_amplification() -> Weight {
+		Weight::from_ref_time(0)
+	}
 }
 
 // For backwards compatibility and tests
@@ -100,9 +102,6 @@ impl WeightInfo for () {
 			.saturating_add(RocksDbWeight::get().reads(13 as u64))
 			.saturating_add(RocksDbWeight::get().writes(12 as u64))
 	}
-	fn update_pool() -> Weight {
-		Weight::from_ref_time(0u128 as u64)
-	}
 	fn add_liquidity() -> Weight {
 		Weight::from_ref_time(64_481_000 as u64)
 			.saturating_add(RocksDbWeight::get().reads(10 as u64))
@@ -127,4 +126,10 @@ impl WeightInfo for () {
 	fn set_asset_tradable_state() -> Weight {
 		Weight::from_ref_time(0)
 	}
+	fn update_pool_fees() -> Weight {
+		Weight::from_ref_time(0)
+	}
+	fn update_amplification() -> Weight {
+		Weight::from_ref_time(0)
+	}
 }