mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-13 14:01:06 +00:00
Bastian Köcher
74428fa8ac
* generate_storage_alias: Rewrite as proc macro attribute
This rewrites the `generate_storage_alias!` declarative macro as proc-macro attribute. While doing
this the name is changed to `storage_alias`. The prefix can now also be the name of a pallet. This
makes storage aliases work in migrations for all kind of chains and not just for the ones that use
predefined prefixes.
* Fix compilation and FMT
* Moare fixes
* 🤦
* ......
* Rework the syntax and support instancing
* FMT
* Prefix variants with `Storage`
* Make it compile
* Fix where clause on rust stable
122 lines
3.9 KiB
Rust
122 lines
3.9 KiB
Rust
// This file is part of Substrate.
|
|
|
|
// Copyright (C) 2021-2022 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.
|
|
|
|
//! Migrate the reference counting state.
|
|
|
|
use crate::{Config, Pallet};
|
|
use codec::{Decode, Encode, FullCodec};
|
|
use frame_support::{
|
|
pallet_prelude::ValueQuery, traits::PalletInfoAccess, weights::Weight, Blake2_128Concat,
|
|
RuntimeDebug,
|
|
};
|
|
use sp_std::prelude::*;
|
|
|
|
/// Type used to encode the number of references an account has.
|
|
type RefCount = u32;
|
|
|
|
/// Information of an account.
|
|
#[derive(Clone, Eq, PartialEq, Default, RuntimeDebug, Encode, Decode)]
|
|
struct AccountInfo<Index, AccountData> {
|
|
nonce: Index,
|
|
consumers: RefCount,
|
|
providers: RefCount,
|
|
sufficients: RefCount,
|
|
data: AccountData,
|
|
}
|
|
|
|
/// Trait to implement to give information about types used for migration
|
|
pub trait V2ToV3 {
|
|
/// The system pallet.
|
|
type Pallet: 'static + PalletInfoAccess;
|
|
|
|
/// System config account id
|
|
type AccountId: 'static + FullCodec;
|
|
|
|
/// System config index
|
|
type Index: 'static + FullCodec + Copy;
|
|
|
|
/// System config account data
|
|
type AccountData: 'static + FullCodec;
|
|
}
|
|
|
|
#[frame_support::storage_alias]
|
|
type UpgradedToU32RefCount<T: Config> = StorageValue<Pallet<T>, bool, ValueQuery>;
|
|
|
|
#[frame_support::storage_alias]
|
|
type UpgradedToTripleRefCount<T: Config> = StorageValue<Pallet<T>, bool, ValueQuery>;
|
|
|
|
#[frame_support::storage_alias]
|
|
type Account<V, T: Config> = StorageMap<
|
|
Pallet<T>,
|
|
Blake2_128Concat,
|
|
<V as V2ToV3>::AccountId,
|
|
AccountInfo<<V as V2ToV3>::Index, <V as V2ToV3>::AccountData>,
|
|
>;
|
|
|
|
/// Migrate from unique `u8` reference counting to triple `u32` reference counting.
|
|
pub fn migrate_from_single_u8_to_triple_ref_count<V: V2ToV3, T: Config>() -> Weight {
|
|
let mut translated: usize = 0;
|
|
<Account<V, T>>::translate::<(V::Index, u8, V::AccountData), _>(|_key, (nonce, rc, data)| {
|
|
translated += 1;
|
|
Some(AccountInfo { nonce, consumers: rc as RefCount, providers: 1, sufficients: 0, data })
|
|
});
|
|
log::info!(
|
|
target: "runtime::system",
|
|
"Applied migration from single u8 to triple reference counting to {:?} elements.",
|
|
translated
|
|
);
|
|
<UpgradedToU32RefCount<T>>::put(true);
|
|
<UpgradedToTripleRefCount<T>>::put(true);
|
|
Weight::max_value()
|
|
}
|
|
|
|
/// Migrate from unique `u32` reference counting to triple `u32` reference counting.
|
|
pub fn migrate_from_single_to_triple_ref_count<V: V2ToV3, T: Config>() -> Weight {
|
|
let mut translated: usize = 0;
|
|
<Account<V, T>>::translate::<(V::Index, RefCount, V::AccountData), _>(
|
|
|_key, (nonce, consumers, data)| {
|
|
translated += 1;
|
|
Some(AccountInfo { nonce, consumers, providers: 1, sufficients: 0, data })
|
|
},
|
|
);
|
|
log::info!(
|
|
target: "runtime::system",
|
|
"Applied migration from single to triple reference counting to {:?} elements.",
|
|
translated
|
|
);
|
|
<UpgradedToTripleRefCount<T>>::put(true);
|
|
Weight::max_value()
|
|
}
|
|
|
|
/// Migrate from dual `u32` reference counting to triple `u32` reference counting.
|
|
pub fn migrate_from_dual_to_triple_ref_count<V: V2ToV3, T: Config>() -> Weight {
|
|
let mut translated: usize = 0;
|
|
<Account<V, T>>::translate::<(V::Index, RefCount, RefCount, V::AccountData), _>(
|
|
|_key, (nonce, consumers, providers, data)| {
|
|
translated += 1;
|
|
Some(AccountInfo { nonce, consumers, providers, sufficients: 0, data })
|
|
},
|
|
);
|
|
log::info!(
|
|
target: "runtime::system",
|
|
"Applied migration from dual to triple reference counting to {:?} elements.",
|
|
translated
|
|
);
|
|
<UpgradedToTripleRefCount<T>>::put(true);
|
|
Weight::max_value()
|
|
}
|