Migrate PalletVersion to StorageVersion

(paritytech/substrate#9165)
chainx-org · Feb 8, 2022 · 99f71bb · 99f71bb
1 parent 59b36b5
commit 99f71bb
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/runtime/dev/src/lib.rs b/runtime/dev/src/lib.rs
@@ -1192,7 +1192,11 @@ pub type Executive = frame_executive::Executive<
     frame_system::ChainContext<Runtime>,
     Runtime,
     AllPallets,
-    (RemoveCollectiveFlip, PhragmenElectionDepositRuntimeUpgrade),
+    (
+        RemoveCollectiveFlip,
+        MigratePalletVersionToStorageVersion,
+        PhragmenElectionDepositRuntimeUpgrade,
+    ),
 >;
 
 impl_runtime_apis! {

diff --git a/runtime/dev/src/migrations.rs b/runtime/dev/src/migrations.rs
@@ -10,6 +10,16 @@ impl frame_support::traits::OnRuntimeUpgrade for RemoveCollectiveFlip {
     }
 }
 
+/// Migrate from `PalletVersion` to the new `StorageVersion`
+pub struct MigratePalletVersionToStorageVersion;
+impl frame_support::traits::OnRuntimeUpgrade for MigratePalletVersionToStorageVersion {
+    fn on_runtime_upgrade() -> frame_support::weights::Weight {
+        frame_support::migrations::migrate_from_pallet_version_to_storage_version::<
+            AllPalletsWithSystem,
+        >(&RocksDbWeight::get())
+    }
+}
+
 // 10 PCX
 const OLD_CANDIDACY_BOND: Balance = 1000 * DOLLARS;
 // 10 mPCX