initialize storage for ProviderBoost on runtime upgrade

pallets/capacity/src/migration/mod.rs

@@ -1,2 +1,4 @@
 /// Migration logic for 6 second block updates sideffects to capacity pallet
 pub mod v4;
+/// initial values for ProviderBoost storage
+pub mod provider_boost_init;

pallets/capacity/src/migration/provider_boost_init.rs

@@ -0,0 +1,57 @@
+use crate::{Config, CurrentEraInfo, RewardEraInfo, RewardPoolInfo, StakingRewardPool};
+use frame_support::{
+    pallet_prelude::Weight,
+    traits::OnRuntimeUpgrade
+};
+use frame_support::{
+    traits::Get,
+};
+use sp_runtime::TryRuntimeError;
+
+#[cfg(feature = "try-runtime")]
+use sp_std::vec::Vec;
+
+/// Initialization during runtime upgrade for Provider Boost storage 
+pub struct ProviderBoostInit<T: Config>(sp_std::marker::PhantomData<T>);
+
+impl<T: Config> OnRuntimeUpgrade for ProviderBoostInit<T> {
+    fn on_runtime_upgrade() -> Weight {
+        let current_era_info = CurrentEraInfo::<T>::get();
+        if current_era_info.eq(&RewardEraInfo::default()) { // 1r
+            let current_block = frame_system::Pallet::<T>::block_number(); // 1r
+            let era_index: T::RewardEra = 0u32.into();
+            CurrentEraInfo::<T>::set(RewardEraInfo { era_index, started_at: current_block, }); // 1w 
+            StakingRewardPool::<T>::insert(era_index, RewardPoolInfo::default());  // 1w
+            T::DbWeight::get().reads_writes(2, 2)
+        }   else {
+            T::DbWeight::get().reads(1)
+        } 
+    }
+
+    #[cfg(feature = "try-runtime")]
+    fn pre_upgrade() -> Result<Vec<u8>, TryRuntimeError> {
+        if CurrentEraInfo::<T>::exists() {
+            log::info!("CurrentEraInfo exists; initialization should be skipped.");
+        } else {
+            log::info!("CurrentEraInfo not found. Initialization should proceed.");
+        }
+        if StakingRewardPool::<T>::iter().count() == 0usize {
+            log::info!("StakingRewardPool will be updated with Era 0");
+        } else {
+            log::info!("StakingRewardPool has already been initialized.")
+        }
+        Ok(Vec::default())
+    }
+
+    #[cfg(feature = "try-runtime")]
+    fn post_upgrade(_state: Vec<u8>) -> Result<(), TryRuntimeError> {
+        assert!(CurrentEraInfo::<T>::exists());
+        let current_block = frame_system::Pallet::<T>::block_number();
+        let info = CurrentEraInfo::<T>::get() ;
+        assert_eq!(info.started_at, current_block);
+        log::info!("CurrentEraInfo.started_at is set to {:?}.", info.started_at);
+        assert_eq!(StakingRewardPool::<T>::iter().count(), 1);
+        Ok(())
+    }    
+}
+

runtime/frequency/src/lib.rs

@@ -335,7 +335,10 @@ pub type Executive = frame_executive::Executive<
 	frame_system::ChainContext<Runtime>,
 	Runtime,
 	AllPalletsWithSystem,
-	(MigratePalletsCurrentStorage<Runtime>, pallet_capacity::migration::v4::MigrationToV4<Runtime>),
+	(MigratePalletsCurrentStorage<Runtime>,
+     pallet_capacity::migration::v4::MigrationToV4<Runtime>,
+     pallet_capacity::migration::provider_boost_init::ProviderBoostInit<Runtime>,
+    ),
 >;
 
 pub struct MigratePalletsCurrentStorage<T>(sp_std::marker::PhantomData<T>);
@@ -562,9 +565,9 @@ impl pallet_capacity::Config for Runtime {
 	type EpochNumber = u32;
 	type CapacityPerToken = CapacityPerToken;
 	type RuntimeFreezeReason = RuntimeFreezeReason;
-	type RewardEra = RewardEra;
+	type RewardEra = u32;
 	type EraLength = ConstU32<{ 14 * DAYS }>;
-	type StakingRewardsPastErasMax = ConstU32<26u32>; // 1 year
+	type StakingRewardsPastErasMax = ConstU32<30u32>;
 	type RewardsProvider = Capacity;
 	type MaxRetargetsPerRewardEra = ConstU32<16>;
 }