Files
pezkuwi-sdk/pezcumulus/pezpallets/aura-ext/src/migration.rs
T
pezkuwichain 3139ffa25e fix: Complete snowbridge pezpallet rebrand and critical bug fixes
- snowbridge-pezpallet-* → pezsnowbridge-pezpallet-* (201 refs)
- pallet/ directories → pezpallet/ (4 locations)
- Fixed pezpallet.rs self-include recursion bug
- Fixed sc-chain-spec hardcoded crate name in derive macro
- Reverted .pezpallet_by_name() to .pallet_by_name() (subxt API)
- Added BizinikiwiConfig type alias for zombienet tests
- Deleted obsolete session state files

Verified: pezsnowbridge-pezpallet-*, pezpallet-staking,
pezpallet-staking-async, pezframe-benchmarking-cli all pass cargo check
2025-12-16 09:57:23 +03:00

75 lines
2.3 KiB
Rust

// Copyright (C) Parity Technologies (UK) Ltd.
// This file is part of Pezcumulus.
// SPDX-License-Identifier: Apache-2.0
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
extern crate alloc;
use crate::{Config, Pezpallet};
#[cfg(feature = "try-runtime")]
use alloc::vec::Vec;
use pezframe_support::{migrations::VersionedMigration, pezpallet_prelude::StorageVersion};
/// The in-code storage version.
pub const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);
mod v0 {
use super::*;
use pezframe_support::{pezpallet_prelude::OptionQuery, storage_alias};
use pezsp_consensus_aura::Slot;
/// Current slot paired with a number of authored blocks.
///
/// Updated on each block initialization.
#[storage_alias]
pub(super) type SlotInfo<T: Config> = StorageValue<Pezpallet<T>, (Slot, u32), OptionQuery>;
}
mod v1 {
use super::*;
use pezframe_support::{pezpallet_prelude::*, traits::UncheckedOnRuntimeUpgrade};
pub struct UncheckedMigrationToV1<T: Config>(PhantomData<T>);
impl<T: Config> UncheckedOnRuntimeUpgrade for UncheckedMigrationToV1<T> {
fn on_runtime_upgrade() -> Weight {
let mut weight: Weight = Weight::zero();
weight += migrate::<T>();
weight
}
#[cfg(feature = "try-runtime")]
fn pre_upgrade() -> Result<Vec<u8>, pezsp_runtime::TryRuntimeError> {
Ok(Vec::new())
}
#[cfg(feature = "try-runtime")]
fn post_upgrade(_state: Vec<u8>) -> Result<(), pezsp_runtime::TryRuntimeError> {
ensure!(!v0::SlotInfo::<T>::exists(), "SlotInfo should not exist");
Ok(())
}
}
pub fn migrate<T: Config>() -> Weight {
v0::SlotInfo::<T>::kill();
T::DbWeight::get().writes(1)
}
}
/// Migrate `V0` to `V1`.
pub type MigrateV0ToV1<T> = VersionedMigration<
0,
1,
v1::UncheckedMigrationToV1<T>,
Pezpallet<T>,
<T as pezframe_system::Config>::DbWeight,
>;