mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-21 04:01:01 +00:00
Bastian Köcher
41218ba184
* Vesting: Fix `post_migration` check
As the vesting migration could already have been done, people could already have started to merge schedules.
* 🤦
96 lines
2.7 KiB
Rust
96 lines
2.7 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) 2017-2021 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.
|
|
|
|
//! Storage migrations for the vesting pallet.
|
|
|
|
use super::*;
|
|
|
|
// Migration from single schedule to multiple schedules.
|
|
pub(crate) mod v1 {
|
|
use super::*;
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
pub(crate) fn pre_migrate<T: Config>() -> Result<(), &'static str> {
|
|
assert!(StorageVersion::<T>::get() == Releases::V0, "Storage version too high.");
|
|
|
|
log::debug!(
|
|
target: "runtime::vesting",
|
|
"migration: Vesting storage version v1 PRE migration checks succesful!"
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Migrate from single schedule to multi schedule storage.
|
|
/// WARNING: This migration will delete schedules if `MaxVestingSchedules < 1`.
|
|
pub(crate) fn migrate<T: Config>() -> Weight {
|
|
let mut reads_writes = 0;
|
|
|
|
Vesting::<T>::translate::<VestingInfo<BalanceOf<T>, T::BlockNumber>, _>(
|
|
|_key, vesting_info| {
|
|
reads_writes += 1;
|
|
let v: Option<
|
|
BoundedVec<
|
|
VestingInfo<BalanceOf<T>, T::BlockNumber>,
|
|
MaxVestingSchedulesGet<T>,
|
|
>,
|
|
> = vec![vesting_info].try_into().ok();
|
|
|
|
if v.is_none() {
|
|
log::warn!(
|
|
target: "runtime::vesting",
|
|
"migration: Failed to move a vesting schedule into a BoundedVec"
|
|
);
|
|
}
|
|
|
|
v
|
|
},
|
|
);
|
|
|
|
T::DbWeight::get().reads_writes(reads_writes, reads_writes)
|
|
}
|
|
|
|
#[cfg(feature = "try-runtime")]
|
|
pub(crate) fn post_migrate<T: Config>() -> Result<(), &'static str> {
|
|
assert_eq!(StorageVersion::<T>::get(), Releases::V1);
|
|
|
|
for (_key, schedules) in Vesting::<T>::iter() {
|
|
assert!(
|
|
schedules.len() >= 1,
|
|
"A bounded vec with incorrect count of items was created."
|
|
);
|
|
|
|
for s in schedules {
|
|
// It is ok if this does not pass, but ideally pre-existing schedules would pass
|
|
// this validation logic so we can be more confident about edge cases.
|
|
if !s.is_valid() {
|
|
log::warn!(
|
|
target: "runtime::vesting",
|
|
"migration: A schedule does not pass new validation logic.",
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
log::debug!(
|
|
target: "runtime::vesting",
|
|
"migration: Vesting storage version v1 POST migration checks successful!"
|
|
);
|
|
Ok(())
|
|
}
|
|
}
|