mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-05-31 04:01:02 +00:00
Sergej Sakac
df87bae1a9
* Try-state: DispatchResult as return type * try_state for the rest of the pallets * pre_upgrade * post_upgrade * try_runtime_upgrade * fixes * bags-list fix * fix * update test * warning fix * ... * final fixes 🤞 * warning.. * frame-support * warnings * Update frame/staking/src/migrations.rs Co-authored-by: Liam Aharon <liam.aharon@hotmail.com> * fix * fix warning * nit fix * merge fixes * small fix * should be good now * missed these ones * introduce TryRuntimeError and TryRuntimeResult * fixes * fix * removed TryRuntimeResult & made some fixes * fix testsg * tests passing * unnecessary imports * Update frame/assets/src/migration.rs Co-authored-by: Keith Yeung <kungfukeith11@gmail.com> --------- Co-authored-by: Liam Aharon <liam.aharon@hotmail.com> Co-authored-by: Keith Yeung <kungfukeith11@gmail.com>
91 lines
2.6 KiB
Rust
91 lines
2.6 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) Parity Technologies (UK) Ltd.
|
|
// 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.
|
|
|
|
pub mod v1 {
|
|
use crate::{types::BalanceOf, *};
|
|
use frame_support::{
|
|
storage::unhashed,
|
|
traits::{Defensive, Get, GetStorageVersion, OnRuntimeUpgrade},
|
|
weights::Weight,
|
|
};
|
|
use sp_staking::EraIndex;
|
|
use sp_std::prelude::*;
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
use frame_support::ensure;
|
|
#[cfg(feature = "try-runtime")]
|
|
use sp_runtime::TryRuntimeError;
|
|
|
|
pub struct MigrateToV1<T>(sp_std::marker::PhantomData<T>);
|
|
impl<T: Config> OnRuntimeUpgrade for MigrateToV1<T> {
|
|
fn on_runtime_upgrade() -> Weight {
|
|
let current = Pallet::<T>::current_storage_version();
|
|
let onchain = Pallet::<T>::on_chain_storage_version();
|
|
|
|
log!(
|
|
info,
|
|
"Running migration with current storage version {:?} / onchain {:?}",
|
|
current,
|
|
onchain
|
|
);
|
|
|
|
if current == 1 && onchain == 0 {
|
|
// update the version nonetheless.
|
|
current.put::<Pallet<T>>();
|
|
|
|
// if a head exists, then we put them back into the queue.
|
|
if Head::<T>::exists() {
|
|
if let Some((stash, _, deposit)) =
|
|
unhashed::take::<(T::AccountId, Vec<EraIndex>, BalanceOf<T>)>(
|
|
&Head::<T>::hashed_key(),
|
|
)
|
|
.defensive()
|
|
{
|
|
Queue::<T>::insert(stash, deposit);
|
|
} else {
|
|
// not much we can do here -- head is already deleted.
|
|
}
|
|
T::DbWeight::get().reads_writes(2, 3)
|
|
} else {
|
|
T::DbWeight::get().reads(2)
|
|
}
|
|
} else {
|
|
log!(info, "Migration did not execute. This probably should be removed");
|
|
T::DbWeight::get().reads(1)
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
fn pre_upgrade() -> Result<Vec<u8>, TryRuntimeError> {
|
|
ensure!(
|
|
Pallet::<T>::on_chain_storage_version() == 0,
|
|
"The onchain storage version must be zero for the migration to execute."
|
|
);
|
|
Ok(Default::default())
|
|
}
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
fn post_upgrade(_: Vec<u8>) -> Result<(), TryRuntimeError> {
|
|
ensure!(
|
|
Pallet::<T>::on_chain_storage_version() == 1,
|
|
"The onchain version must be updated after the migration."
|
|
);
|
|
Ok(())
|
|
}
|
|
}
|
|
}
|