Merge pull request #1921 from subspace/tx-range-migration

Migrate domain tx range on runtime upgrade
autonomys · Sep 4, 2023 · 4a685ef · 4a685ef
2 parents af445bf + 31fadec
commit 4a685ef
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/crates/pallet-domains/src/lib.rs b/crates/pallet-domains/src/lib.rs
@@ -1097,12 +1097,9 @@ mod pallet {
 
         // TODO: remove once the migration is done
         fn on_runtime_upgrade() -> Weight {
-            for (domain_id, stake_summary) in DomainStakingSummary::<T>::iter() {
-                if stake_summary.current_epoch_index.is_zero() {
-                    if let Err(err) = do_finalize_domain_current_epoch::<T>(domain_id, One::one()) {
-                        log::error!(target: "runtime::domains", "Failed to do epoch transition for {domain_id:?}: {err:?}");
-                    }
-                }
+            for (domain_id, mut tx_range_state) in DomainTxRangeState::<T>::iter() {
+                tx_range_state.tx_range = Self::initial_tx_range();
+                DomainTxRangeState::<T>::set(domain_id, Some(tx_range_state));
             }
             Weight::zero()
         }