mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-12 17:01:09 +00:00
Liam Aharon
84ddbaf684
As suggested by @ggwpez (https://github.com/paritytech/polkadot-sdk/pull/2142#discussion_r1388145872), remove the `VersionChecked` prefix from version checked migrations (but leave `VersionUnchecked` prefixes) --------- Co-authored-by: command-bot <>
74 lines
2.7 KiB
Rust
74 lines
2.7 KiB
Rust
// Copyright (C) Parity Technologies (UK) Ltd.
|
|
// This file is part of Polkadot.
|
|
|
|
// Polkadot is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Polkadot is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Polkadot. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
use super::{Config, MaxPermanentSlots, MaxTemporarySlots, Pallet, LOG_TARGET};
|
|
use frame_support::traits::{Get, GetStorageVersion, OnRuntimeUpgrade};
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
use frame_support::ensure;
|
|
#[cfg(feature = "try-runtime")]
|
|
use sp_std::vec::Vec;
|
|
|
|
pub mod v1 {
|
|
|
|
use super::*;
|
|
pub struct VersionUncheckedMigrateToV1<T>(sp_std::marker::PhantomData<T>);
|
|
impl<T: Config> OnRuntimeUpgrade for VersionUncheckedMigrateToV1<T> {
|
|
#[cfg(feature = "try-runtime")]
|
|
fn pre_upgrade() -> Result<Vec<u8>, sp_runtime::TryRuntimeError> {
|
|
let onchain_version = Pallet::<T>::on_chain_storage_version();
|
|
ensure!(onchain_version < 1, "assigned_slots::MigrateToV1 migration can be deleted");
|
|
Ok(Default::default())
|
|
}
|
|
|
|
fn on_runtime_upgrade() -> frame_support::weights::Weight {
|
|
let onchain_version = Pallet::<T>::on_chain_storage_version();
|
|
if onchain_version < 1 {
|
|
const MAX_PERMANENT_SLOTS: u32 = 100;
|
|
const MAX_TEMPORARY_SLOTS: u32 = 100;
|
|
|
|
<MaxPermanentSlots<T>>::put(MAX_PERMANENT_SLOTS);
|
|
<MaxTemporarySlots<T>>::put(MAX_TEMPORARY_SLOTS);
|
|
// Return the weight consumed by the migration.
|
|
T::DbWeight::get().reads_writes(1, 3)
|
|
} else {
|
|
log::info!(target: LOG_TARGET, "MigrateToV1 should be removed");
|
|
T::DbWeight::get().reads(1)
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
fn post_upgrade(_state: Vec<u8>) -> Result<(), sp_runtime::TryRuntimeError> {
|
|
let onchain_version = Pallet::<T>::on_chain_storage_version();
|
|
ensure!(onchain_version == 1, "assigned_slots::MigrateToV1 needs to be run");
|
|
assert_eq!(<MaxPermanentSlots<T>>::get(), 100);
|
|
assert_eq!(<MaxTemporarySlots<T>>::get(), 100);
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/// [`VersionUncheckedMigrateToV1`] wrapped in a
|
|
/// [`VersionedMigration`](frame_support::migrations::VersionedMigration), ensuring the
|
|
/// migration is only performed when on-chain version is 0.
|
|
pub type MigrateToV1<T> = frame_support::migrations::VersionedMigration<
|
|
0,
|
|
1,
|
|
VersionUncheckedMigrateToV1<T>,
|
|
Pallet<T>,
|
|
<T as frame_system::Config>::DbWeight,
|
|
>;
|
|
}
|