Set storage version (#1559)

Signed-off-by: Xavier Lau <xavier@inv.cafe>
darwinia-network · Aug 12, 2024 · 3b73ea2 · 3b73ea2
1 parent df83f28
commit 3b73ea2
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/runtime/crab/src/migration.rs b/runtime/crab/src/migration.rs
@@ -45,10 +45,13 @@ impl frame_support::traits::OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
 }
 
 fn migrate() -> frame_support::weights::Weight {
+	frame_support::traits::StorageVersion::new(2)
+		.put::<cumulus_pallet_dmp_queue::Pallet<Runtime>>();
+
 	const KTON_DAO_VAULT_ADDR: &str = "0x652182C6aBc0bBE41b5702b05a26d109A405EAcA";
 
 	#[cfg(feature = "try-runtime")]
-	assert!(array_bytes::hex_n_into::<_, _, 20>(KTON_DAO_VAULT_ADDR).is_ok());
+	assert!(array_bytes::hex_n_into::<_, AccountId, 20>(KTON_DAO_VAULT_ADDR).is_ok());
 
 	if let Ok(w) = array_bytes::hex_n_into::<_, _, 20>(KTON_DAO_VAULT_ADDR) {
 		<darwinia_staking::KtonRewardDistributionContract<Runtime>>::put(w);

diff --git a/runtime/darwinia/src/migration.rs b/runtime/darwinia/src/migration.rs
@@ -45,6 +45,9 @@ impl frame_support::traits::OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
 }
 
 fn migrate() -> frame_support::weights::Weight {
+	frame_support::traits::StorageVersion::new(2)
+		.put::<cumulus_pallet_dmp_queue::Pallet<Runtime>>();
+
 	let _ = migration::clear_storage_prefix(
 		b"BridgeKusamaGrandpa",
 		b"ImportedHeaders",
@@ -64,7 +67,7 @@ fn migrate() -> frame_support::weights::Weight {
 	const KTON_DAO_VAULT_ADDR: &str = "0x652182C6aBc0bBE41b5702b05a26d109A405EAcA";
 
 	#[cfg(feature = "try-runtime")]
-	assert!(array_bytes::hex_n_into::<_, _, 20>(KTON_DAO_VAULT_ADDR).is_ok());
+	assert!(array_bytes::hex_n_into::<_, AccountId, 20>(KTON_DAO_VAULT_ADDR).is_ok());
 
 	if let Ok(w) = array_bytes::hex_n_into::<_, _, 20>(KTON_DAO_VAULT_ADDR) {
 		<darwinia_staking::KtonRewardDistributionContract<Runtime>>::put(w);

diff --git a/runtime/koi/src/migration.rs b/runtime/koi/src/migration.rs
@@ -45,13 +45,9 @@ impl frame_support::traits::OnRuntimeUpgrade for CustomOnRuntimeUpgrade {
 }
 
 fn migrate() -> frame_support::weights::Weight {
-	let n = migration_helper::PalletCleaner {
-		name: b"EthereumXcm",
-		values: &[b"Nonce", b"EthereumXcmSuspended"],
-		maps: &[],
-	}
-	.remove_storage_values();
+	frame_support::traits::StorageVersion::new(2)
+		.put::<cumulus_pallet_dmp_queue::Pallet<Runtime>>();
 
 	// frame_support::weights::Weight::zero()
-	<Runtime as frame_system::Config>::DbWeight::get().reads_writes(0, n)
+	<Runtime as frame_system::Config>::DbWeight::get().reads_writes(0, 1)
 }