mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-06-11 18:51:12 +00:00
Introduce on_runtime_upgrade (#5058)
* Initial idea of `on_runtime_upgrade`
* Runtime storage for module version
* Gui shawntabrizi runtime upgrade (#5118)
* adding unleash to ci (#5020)
* adding unleash to ci
* fixing formatting
* with a dot please
* alpha.3 now
* do not publish testing helpers
* remove old test-helpers cruft
* fix cargo.lock
* with alpha 4
* do not publish runtime-interface-test either
* disable more test crates from publishing
* switch to alpha.5
* replace tempdir with tempfile
* update lru
* switch to bytes 0.5
* release script fixes
* switch on and to latest alpha
* BUT THE SPACES
* Fix: CI failing for some CLI tests (#5043)
* Initial commit
Forked at: ad90ab7ec9
Parent branch: origin/master
* Increase killing grace period of CLI tests and display more info
* Use --dev everywhere possible
* Put pruning mode to its own params struct
* Add pruning params to export-blocks command
* Added missing file
* Removed not-dev mode in tests
* Add pruning mode to the revert command
* Decrease killing grace period again
* Move back unsafe_pruning to import_params
* Applied proposed changes
* aura: remove unused tx pool (#5046)
* aura: remove unused transaction pool parameter
* node-template: remove transaction pool from aura
* aura: fix tests
* Extend rust nightly detection in `wasm-builder` (#5021)
Instead of just testing `cargo` and `rustup run nightly`, we now test
the `CARGO` env variable and also scan non default nightlies. The user
is also now able to select the toolchain with `WASM_BUILD_TOOLCHAIN`.
* Add steps setting to benchmarking CLI (#5033)
* Add steps setting to CLI, use max value to hit worst case.
* Bump impl_version.
* Apply review suggestion.
* Remove indices from node-template (#5025)
* Remove indices from node-template
* Use identity lookup instead
* Bump impl
* clean cargo.toml
* Fix documentation for "BlockBuilder::push_trusted" (#5051)
* fix doc
* rephrase
* do not check unleash on every PR, only master and tags (#5054)
* do not check unleash on every PR, only master and tags
* move scripts folder
* add signed-tag check to CI
* remove publish-to-crates-io dependencies
Co-authored-by: s3krit <pugh@s3kr.it>
* prepare version to alpha.1 (#5055)
bump version to -alpha.1
* Sync: validate block responses for required data (#5052)
* Less verbose state-db logging
* Validate block responses for block bodies
* Update client/network/src/protocol.rs
Co-Authored-By: Bastian Köcher <bkchr@users.noreply.github.com>
* Added validation test
* Disconnect on missing header as well
* Typo
Co-Authored-By: André Silva <andre.beat@gmail.com>
Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
Co-authored-by: André Silva <andre.beat@gmail.com>
* Make these chainspecs fields private (#5031)
* Fix dockerfile (#5059)
* Adds documentation for `wipe` and `commit` (#5053)
* Adds documentation for `wipe` and `commit`
This adds documentation to `wipe` and `commit` of `Externalities`.
Besides that it removes the default implementation that would just panic
and requires that all implementers of the trait implement the functions.
* Update primitives/externalities/src/lib.rs
Co-Authored-By: joe petrowski <25483142+joepetrowski@users.noreply.github.com>
Co-authored-by: joe petrowski <25483142+joepetrowski@users.noreply.github.com>
* Fix the issue with `trybuild`'s `ui` tests (#4992)
* repro ui bug
* fix the tests
* test with the new image
* test without CARGO_HOME
* test without fixes
* test again
* fix trybuild old versions
* bump CArgo.lock
* fix trybuild newest versions
* bump Cargo.lock
* trying on the latest image
* bump Cargo.lock
* run with the old image
* ci will be green on the image from 2020-02-19 [skip ci]
* bump Cargo.lock
* Activate publishing of draft releases... (#5062)
* Activate publishing of draft releases...
... And fix the message sending (missing parameter).
* publish_draft_release.sh now checks latest...
... release on github rather than just a tag
* Fix/div by zero (#5041)
* Handle gas_price being zero separately
* Bump spec_version
* Add a unit & integration tests for gas price = 0
* set missing metadata fields, prepping alpha.2 (#5067)
* setting first batch of descriptions
* fix what I just broke
* next batch
* and pallets, too
* last batch
* set cargo.lock
* keep'em dev-deps
* bump version to alpha.2
* Fix revalidation not revalidating multiple times (#5065)
* removes use of sc_client::Client from sc_finality_grandpa (#5030)
* removes use of sc_client::Client from sc_finality_grandpa
* code formatting
* code formatting
* removes use of sc_client::Client from sc_finality_grandpa
* Remove deprecated host functions (#5038)
Sadly we need to keep one function `ext_blake2_256`. This function is
manually defined in `sp-core`.
* removes use of sc_client::Client from sc_basic_authorship (#5050)
* removes use of sc-client from sc-basic-authorship
* refactor use of ProposerFactory
* correct dep path
* pallet-transaction-payment clean up (#5070)
* Formatting clean up
* Introduce separate setters for the fees.
* *: Rename prometheus-exporter crate to substrate-prometheus-end… (#5076)
This patch renames the crate for the following two reasons:
1. The prometheus-exporter crate introduces native in-process Prometheus
style instrumentation to the Substrate project. Within the Prometheus
ecosystem the term "exporter" is used for external processes exposing
metrics for e.g. the Linux Kernel. In-process exposition would be
described via the term "endpoint".
2. "prometheus-exporter" is generic and ignores the fact that it is only
usable within the context of Substrate. In addition the name
"prometheus-exporter" is already taken on crates.io.
* rename `browser-utils` to `substrate-browser-utils` (#5079)
* prepping for Alpha.3 (#5080)
* Bump to alpha.3
* update gitlab-ci
* Propagate DispatchError for benchmarks. (#5075)
* Propagate DispatchError for benchmarks.
* Apply review suggestions.
* Use RuntimeString.
* fix expect
Co-Authored-By: Bastian Köcher <bkchr@users.noreply.github.com>
Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
* Add options to overwrite range bounds in benchmark command. (#5072)
* Add --mins --maxs to benchmark command.
* Apply review suggestions.
* Update yamux to version 0.4.4. (#5086)
* Remove more instances of futures01 (#4633)
* Start removing last few instances of futures01
* Use to_poll on wasm
* Revert "Use to_poll on wasm"
This reverts commit 1c61728f10d520df5f9b28c415a0db68e478b9c7.
* Fix fg test
* Upgrade network test futures
* Update offchain hyper version
* Update service test
* bump tokio to 0.2.10
* Removed some unneeded tokios
* fixes
* fix run_until_all_full
* Make service test debuggable
* Update client/offchain/src/api/http.rs
Co-Authored-By: Demi Obenour <48690212+DemiMarie-parity@users.noreply.github.com>
* Add service_test to test-int output
* nitpicking
* Finally fix test
* Give up and revert client/serviec/test
* Revert gitlab ci too
Co-authored-by: Demi Obenour <demi@parity.io>
* Make export blocks default to json on stdout (#5090)
* Make export blocks default to json on stdout
* Multiline instead of single line to stay under 100 cols
* Change --json flag to --binary, defaulting to json
* Offence reporting returns a result (#5082)
* Offence reporting returns a result
* Bump spec_version
* Use unwrap instead of assertions
* Fix more review grumbles
* Update to libp2p 0.16.2 (#5088)
* Remove request ID from the new protocol (#5049)
* Make sure we remove a peer on disconnect in gossip (#5104)
* Make sure we remove peers on disconnect in gossip state machine
* Clear up the code
* Add a comment
* Expose `state-db` memory info (#5110)
This exposes memory statistics from the state-db.
* Change extrinsic_count to extrinsic_index in pallet-utility (#5044)
Co-authored-by: Benjamin Kampmann <ben.kampmann@googlemail.com>
* Add more metrics to prometheus (#5034)
* Add a few things
* Add finality_grandpa_round
* fix fg tests
* Nitpicks
* Nitpicks
* Fix name of prometheus crate
* Update to SCALE 1.2.0 (#5113)
This updates `parity-scale-codec` to `1.2.0`, which includes multiple
performance improvements and a fix that bounds the capacity of a vector
at decoding.
* Lazy payouts (#4474)
* TODOs
* Remove superfluous:
* partial implementation
* full implementation
* fix preferences
* update comments
* upgrade test WIP
* fix more tests
* fix cutoff
* fix saturation
* comment
* upgrade mock
* upgrade test
* WIP migration
* WIP migration
* remove slot stake stuff
* fix merge
* migration of ledger
* remove equalize from test
* add test
* fix
* update doc
* fix compilation
* improve test readibility
* improve doc
* fix most todo
* fix migration and test
* remove println
* WIP
* add test and spec
* weight
* update doc
* safer end_era
* fix exposure of conversion
* Revert "safer end_era"
This reverts commit 72ff737d27be67d87308514b13e2574bc5f09fce.
* fix useless put
* exposure clipped
* doc
* fix payout with clipped
* fix node runtime
* add doc
* pluggable and generalized staking module
* remove print
* update doc
* refactor
* improve documentation and implementation
* fix test
* Fix test
* fix test
* fix test
* fix remove lowest stake from exposure, not biggest.
* nomination index arguments in nominator_payout
* add test
* try to fix offence
* apply slashed and bond eras until active era
* doc
* update spec version
* add test upgrade from previous test environment
* Apply suggestions from code review
Co-Authored-By: Shawn Tabrizi <shawntabrizi@gmail.com>
* nominators upgrade has been cleaned
* dynamic history depth implementation
* make current_era - history_depth included
* Change equality check to start era to less than or equal
* Use era specific validator prefs
* Add print statement and comment about start era if <
* fix next_reward overflow
* make more check for bad era claim for zero cost
* small refactor
* code refactor + fix use of deprecated storage
* fix wasm build
* add comment
* Fix tests
* remove outdated comment
* Apply suggestions from code review
Co-Authored-By: Shawn Tabrizi <shawntabrizi@gmail.com>
* gather active era information into one storage
Co-authored-by: thiolliere <gui.thiolliere@gmail.com>
Co-authored-by: Shawn Tabrizi <shawntabrizi@gmail.com>
* impl on_runtime_upgrade
Co-authored-by: Benjamin Kampmann <ben.kampmann@googlemail.com>
Co-authored-by: Cecile Tonglet <cecile.tonglet@cecton.com>
Co-authored-by: André Silva <andre.beat@gmail.com>
Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
Co-authored-by: Marcio Diaz <marcio@parity.io>
Co-authored-by: Nikolay Volf <nikvolf@gmail.com>
Co-authored-by: s3krit <pugh@s3kr.it>
Co-authored-by: Arkadiy Paronyan <arkady.paronyan@gmail.com>
Co-authored-by: Pierre Krieger <pierre.krieger1708@gmail.com>
Co-authored-by: Chevdor <chevdor@users.noreply.github.com>
Co-authored-by: joe petrowski <25483142+joepetrowski@users.noreply.github.com>
Co-authored-by: Denis Pisarev <denis.pisarev@parity.io>
Co-authored-by: Eric <e.lim0322@gmail.com>
Co-authored-by: Seun Lanlege <seunlanlege@gmail.com>
Co-authored-by: Sergei Pepyakin <s.pepyakin@gmail.com>
Co-authored-by: Max Inden <mail@max-inden.de>
Co-authored-by: Ashley <ashley.ruglys@gmail.com>
Co-authored-by: Toralf Wittner <tw@dtex.org>
Co-authored-by: Demi Obenour <demi@parity.io>
Co-authored-by: pscott <30843220+pscott@users.noreply.github.com>
Co-authored-by: Fedor Sakharov <fedor.sakharov@gmail.com>
Co-authored-by: Gavin Wood <github@gavwood.com>
Co-authored-by: thiolliere <gui.thiolliere@gmail.com>
* make compile
* Add some tests
* docs
* Remove "useless" code
* Fix merge and use n + 1 block number
* Fix tests
* unfix ui tests
* Update on_initialize.stderr
* fix test
* Fix test
* Bump spec
* Remove `on_finalise` and `on_initialise`
* Use bool for tracking runtime upgraded
* typo
* Support runtime upgrade with `set_storage`
* Refactor migration code location
* add trailing newlines
* Remove old `IsUpgraded` flag
* Update state root
* Exhaustive match statement
* Apply suggestions from code review
Co-Authored-By: Kian Paimani <5588131+kianenigma@users.noreply.github.com>
Co-authored-by: Benjamin Kampmann <ben.kampmann@googlemail.com>
Co-authored-by: Cecile Tonglet <cecile.tonglet@cecton.com>
Co-authored-by: André Silva <andre.beat@gmail.com>
Co-authored-by: Bastian Köcher <bkchr@users.noreply.github.com>
Co-authored-by: Marcio Diaz <marcio@parity.io>
Co-authored-by: Nikolay Volf <nikvolf@gmail.com>
Co-authored-by: s3krit <pugh@s3kr.it>
Co-authored-by: Arkadiy Paronyan <arkady.paronyan@gmail.com>
Co-authored-by: Pierre Krieger <pierre.krieger1708@gmail.com>
Co-authored-by: Chevdor <chevdor@users.noreply.github.com>
Co-authored-by: joe petrowski <25483142+joepetrowski@users.noreply.github.com>
Co-authored-by: Denis Pisarev <denis.pisarev@parity.io>
Co-authored-by: Eric <e.lim0322@gmail.com>
Co-authored-by: Seun Lanlege <seunlanlege@gmail.com>
Co-authored-by: Sergei Pepyakin <s.pepyakin@gmail.com>
Co-authored-by: Max Inden <mail@max-inden.de>
Co-authored-by: Ashley <ashley.ruglys@gmail.com>
Co-authored-by: Toralf Wittner <tw@dtex.org>
Co-authored-by: Demi Obenour <demi@parity.io>
Co-authored-by: pscott <30843220+pscott@users.noreply.github.com>
Co-authored-by: Fedor Sakharov <fedor.sakharov@gmail.com>
Co-authored-by: Gavin Wood <github@gavwood.com>
Co-authored-by: thiolliere <gui.thiolliere@gmail.com>
Co-authored-by: Kian Paimani <5588131+kianenigma@users.noreply.github.com>
This commit is contained in:
@@ -157,6 +157,7 @@ mod tests_composite;
|
||||
mod tests;
|
||||
#[cfg(feature = "runtime-benchmarks")]
|
||||
mod benchmarking;
|
||||
mod migration;
|
||||
|
||||
use sp_std::prelude::*;
|
||||
use sp_std::{cmp, result, mem, fmt::Debug, ops::BitOr, convert::Infallible};
|
||||
@@ -348,6 +349,21 @@ impl<Balance: Saturating + Copy + Ord> AccountData<Balance> {
|
||||
}
|
||||
}
|
||||
|
||||
// A value placed in storage that represents the current version of the Balances storage.
|
||||
// This value is used by the `on_runtime_upgrade` logic to determine whether we run
|
||||
// storage migration logic. This should match directly with the semantic versions of the Rust crate.
|
||||
#[derive(Encode, Decode, Clone, Copy, PartialEq, Eq, RuntimeDebug)]
|
||||
enum Releases {
|
||||
V1_0_0,
|
||||
V2_0_0,
|
||||
}
|
||||
|
||||
impl Default for Releases {
|
||||
fn default() -> Self {
|
||||
Releases::V1_0_0
|
||||
}
|
||||
}
|
||||
|
||||
decl_storage! {
|
||||
trait Store for Module<T: Trait<I>, I: Instance=DefaultInstance> as Balances {
|
||||
/// The total units issued in the system.
|
||||
@@ -367,10 +383,10 @@ decl_storage! {
|
||||
/// NOTE: Should only be accessed when setting, changing and freeing a lock.
|
||||
pub Locks get(fn locks): map hasher(blake2_256) T::AccountId => Vec<BalanceLock<T::Balance>>;
|
||||
|
||||
/// True if network has been upgraded to this version.
|
||||
/// Storage version of the pallet.
|
||||
///
|
||||
/// True for new networks.
|
||||
IsUpgraded build(|_: &GenesisConfig<T, I>| true): bool;
|
||||
/// This is set to v2.0.0 for new networks.
|
||||
StorageVersion build(|_: &GenesisConfig<T, I>| Releases::V2_0_0): Releases;
|
||||
}
|
||||
add_extra_genesis {
|
||||
config(balances): Vec<(T::AccountId, T::Balance)>;
|
||||
@@ -518,11 +534,8 @@ decl_module! {
|
||||
<Self as Currency<_>>::transfer(&transactor, &dest, value, KeepAlive)?;
|
||||
}
|
||||
|
||||
fn on_initialize() {
|
||||
if !IsUpgraded::<I>::get() {
|
||||
IsUpgraded::<I>::put(true);
|
||||
Self::do_upgrade();
|
||||
}
|
||||
fn on_runtime_upgrade() {
|
||||
migration::on_runtime_upgrade::<T, I>();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -548,82 +561,6 @@ impl<Balance, BlockNumber> OldBalanceLock<Balance, BlockNumber> {
|
||||
impl<T: Trait<I>, I: Instance> Module<T, I> {
|
||||
// PRIVATE MUTABLES
|
||||
|
||||
// Upgrade from the pre-#4649 balances/vesting into the new balances.
|
||||
pub fn do_upgrade() {
|
||||
sp_runtime::print("Upgrading account balances...");
|
||||
// First, migrate from old FreeBalance to new Account.
|
||||
// We also move all locks across since only accounts with FreeBalance values have locks.
|
||||
// FreeBalance: map T::AccountId => T::Balance
|
||||
for (hash, free) in StorageIterator::<T::Balance>::new(b"Balances", b"FreeBalance").drain() {
|
||||
let mut account = AccountData { free, ..Default::default() };
|
||||
// Locks: map T::AccountId => Vec<BalanceLock>
|
||||
let old_locks = get_storage_value::<Vec<OldBalanceLock<T::Balance, T::BlockNumber>>>(b"Balances", b"Locks", &hash);
|
||||
if let Some(locks) = old_locks {
|
||||
let locks = locks.into_iter()
|
||||
.map(|i| {
|
||||
let (result, expiry) = i.upgraded();
|
||||
if expiry != T::BlockNumber::max_value() {
|
||||
// Any `until`s that are not T::BlockNumber::max_value come from
|
||||
// democracy and need to be migrated over there.
|
||||
// Democracy: Locks get(locks): map T::AccountId => Option<T::BlockNumber>;
|
||||
put_storage_value(b"Democracy", b"Locks", &hash, expiry);
|
||||
}
|
||||
result
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
for l in locks.iter() {
|
||||
if l.reasons == Reasons::All || l.reasons == Reasons::Misc {
|
||||
account.misc_frozen = account.misc_frozen.max(l.amount);
|
||||
}
|
||||
if l.reasons == Reasons::All || l.reasons == Reasons::Fee {
|
||||
account.fee_frozen = account.fee_frozen.max(l.amount);
|
||||
}
|
||||
}
|
||||
put_storage_value(b"Balances", b"Locks", &hash, locks);
|
||||
}
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
// Second, migrate old ReservedBalance into new Account.
|
||||
// ReservedBalance: map T::AccountId => T::Balance
|
||||
for (hash, reserved) in StorageIterator::<T::Balance>::new(b"Balances", b"ReservedBalance").drain() {
|
||||
let mut account = get_storage_value::<AccountData<T::Balance>>(b"Balances", b"Account", &hash).unwrap_or_default();
|
||||
account.reserved = reserved;
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
|
||||
// Finally, migrate vesting and ensure locks are in place. We will be lazy and just lock
|
||||
// for the maximum amount (i.e. at genesis). Users will need to call "vest" to reduce the
|
||||
// lock to something sensible.
|
||||
// pub Vesting: map T::AccountId => Option<VestingSchedule>;
|
||||
for (hash, vesting) in StorageIterator::<(T::Balance, T::Balance, T::BlockNumber)>::new(b"Balances", b"Vesting").drain() {
|
||||
let mut account = get_storage_value::<AccountData<T::Balance>>(b"Balances", b"Account", &hash).unwrap_or_default();
|
||||
let mut locks = get_storage_value::<Vec<BalanceLock<T::Balance>>>(b"Balances", b"Locks", &hash).unwrap_or_default();
|
||||
locks.push(BalanceLock {
|
||||
id: *b"vesting ",
|
||||
amount: vesting.0.clone(),
|
||||
reasons: Reasons::Misc,
|
||||
});
|
||||
account.misc_frozen = account.misc_frozen.max(vesting.0.clone());
|
||||
put_storage_value(b"Vesting", b"Vesting", &hash, vesting);
|
||||
put_storage_value(b"Balances", b"Locks", &hash, locks);
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
|
||||
for (hash, balances) in StorageIterator::<AccountData<T::Balance>>::new(b"Balances", b"Account").drain() {
|
||||
let nonce = take_storage_value::<T::Index>(b"System", b"AccountNonce", &hash).unwrap_or_default();
|
||||
let mut refs: system::RefCount = 0;
|
||||
// The items in Kusama that would result in a ref count being incremented.
|
||||
if have_storage_value(b"Democracy", b"Proxy", &hash) { refs += 1 }
|
||||
// We skip Recovered since it's being replaced anyway.
|
||||
let mut prefixed_hash = twox_64(&b":session:keys"[..]).to_vec();
|
||||
prefixed_hash.extend(&b":session:keys"[..]);
|
||||
prefixed_hash.extend(&hash[..]);
|
||||
if have_storage_value(b"Session", b"NextKeys", &prefixed_hash) { refs += 1 }
|
||||
if have_storage_value(b"Staking", b"Bonded", &hash) { refs += 1 }
|
||||
put_storage_value(b"System", b"Account", &hash, (nonce, refs, &balances));
|
||||
}
|
||||
}
|
||||
|
||||
/// Get the free balance of an account.
|
||||
pub fn free_balance(who: impl sp_std::borrow::Borrow<T::AccountId>) -> T::Balance {
|
||||
Self::account(who.borrow()).free
|
||||
|
||||
@@ -0,0 +1,88 @@
|
||||
use super::*;
|
||||
|
||||
pub fn on_runtime_upgrade<T: Trait<I>, I: Instance>() {
|
||||
match StorageVersion::<I>::get() {
|
||||
Releases::V2_0_0 => return,
|
||||
Releases::V1_0_0 => upgrade_v1_to_v2::<T, I>(),
|
||||
}
|
||||
}
|
||||
|
||||
// Upgrade from the pre-#4649 balances/vesting into the new balances.
|
||||
fn upgrade_v1_to_v2<T: Trait<I>, I: Instance>() {
|
||||
sp_runtime::print("Upgrading account balances...");
|
||||
// First, migrate from old FreeBalance to new Account.
|
||||
// We also move all locks across since only accounts with FreeBalance values have locks.
|
||||
// FreeBalance: map T::AccountId => T::Balance
|
||||
for (hash, free) in StorageIterator::<T::Balance>::new(b"Balances", b"FreeBalance").drain() {
|
||||
let mut account = AccountData { free, ..Default::default() };
|
||||
// Locks: map T::AccountId => Vec<BalanceLock>
|
||||
let old_locks = get_storage_value::<Vec<OldBalanceLock<T::Balance, T::BlockNumber>>>(b"Balances", b"Locks", &hash);
|
||||
if let Some(locks) = old_locks {
|
||||
let locks = locks.into_iter()
|
||||
.map(|i| {
|
||||
let (result, expiry) = i.upgraded();
|
||||
if expiry != T::BlockNumber::max_value() {
|
||||
// Any `until`s that are not T::BlockNumber::max_value come from
|
||||
// democracy and need to be migrated over there.
|
||||
// Democracy: Locks get(locks): map T::AccountId => Option<T::BlockNumber>;
|
||||
put_storage_value(b"Democracy", b"Locks", &hash, expiry);
|
||||
}
|
||||
result
|
||||
})
|
||||
.collect::<Vec<_>>();
|
||||
for l in locks.iter() {
|
||||
if l.reasons == Reasons::All || l.reasons == Reasons::Misc {
|
||||
account.misc_frozen = account.misc_frozen.max(l.amount);
|
||||
}
|
||||
if l.reasons == Reasons::All || l.reasons == Reasons::Fee {
|
||||
account.fee_frozen = account.fee_frozen.max(l.amount);
|
||||
}
|
||||
}
|
||||
put_storage_value(b"Balances", b"Locks", &hash, locks);
|
||||
}
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
// Second, migrate old ReservedBalance into new Account.
|
||||
// ReservedBalance: map T::AccountId => T::Balance
|
||||
for (hash, reserved) in StorageIterator::<T::Balance>::new(b"Balances", b"ReservedBalance").drain() {
|
||||
let mut account = get_storage_value::<AccountData<T::Balance>>(b"Balances", b"Account", &hash).unwrap_or_default();
|
||||
account.reserved = reserved;
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
|
||||
// Finally, migrate vesting and ensure locks are in place. We will be lazy and just lock
|
||||
// for the maximum amount (i.e. at genesis). Users will need to call "vest" to reduce the
|
||||
// lock to something sensible.
|
||||
// pub Vesting: map T::AccountId => Option<VestingSchedule>;
|
||||
for (hash, vesting) in StorageIterator::<(T::Balance, T::Balance, T::BlockNumber)>::new(b"Balances", b"Vesting").drain() {
|
||||
let mut account = get_storage_value::<AccountData<T::Balance>>(b"Balances", b"Account", &hash).unwrap_or_default();
|
||||
let mut locks = get_storage_value::<Vec<BalanceLock<T::Balance>>>(b"Balances", b"Locks", &hash).unwrap_or_default();
|
||||
locks.push(BalanceLock {
|
||||
id: *b"vesting ",
|
||||
amount: vesting.0.clone(),
|
||||
reasons: Reasons::Misc,
|
||||
});
|
||||
account.misc_frozen = account.misc_frozen.max(vesting.0.clone());
|
||||
put_storage_value(b"Vesting", b"Vesting", &hash, vesting);
|
||||
put_storage_value(b"Balances", b"Locks", &hash, locks);
|
||||
put_storage_value(b"Balances", b"Account", &hash, account);
|
||||
}
|
||||
|
||||
for (hash, balances) in StorageIterator::<AccountData<T::Balance>>::new(b"Balances", b"Account").drain() {
|
||||
let nonce = take_storage_value::<T::Index>(b"System", b"AccountNonce", &hash).unwrap_or_default();
|
||||
let mut refs: system::RefCount = 0;
|
||||
// The items in Kusama that would result in a ref count being incremented.
|
||||
if have_storage_value(b"Democracy", b"Proxy", &hash) { refs += 1 }
|
||||
// We skip Recovered since it's being replaced anyway.
|
||||
let mut prefixed_hash = twox_64(&b":session:keys"[..]).to_vec();
|
||||
prefixed_hash.extend(&b":session:keys"[..]);
|
||||
prefixed_hash.extend(&hash[..]);
|
||||
if have_storage_value(b"Session", b"NextKeys", &prefixed_hash) { refs += 1 }
|
||||
if have_storage_value(b"Staking", b"Bonded", &hash) { refs += 1 }
|
||||
put_storage_value(b"System", b"Account", &hash, (nonce, refs, &balances));
|
||||
}
|
||||
|
||||
take_storage_value::<T::Index>(b"Balances", b"IsUpgraded", &[]);
|
||||
|
||||
StorageVersion::<I>::put(Releases::V2_0_0);
|
||||
}
|
||||
@@ -1555,7 +1555,7 @@ mod tests {
|
||||
};
|
||||
use sp_core::H256;
|
||||
use sp_runtime::{
|
||||
traits::{BlakeTwo256, IdentityLookup, Bounded, BadOrigin, OnInitialize},
|
||||
traits::{BlakeTwo256, IdentityLookup, Bounded, BadOrigin, OnRuntimeUpgrade},
|
||||
testing::Header, Perbill,
|
||||
};
|
||||
use pallet_balances::{BalanceLock, Error as BalancesError};
|
||||
@@ -1713,7 +1713,7 @@ mod tests {
|
||||
];
|
||||
s.top = data.into_iter().collect();
|
||||
sp_io::TestExternalities::new(s).execute_with(|| {
|
||||
Balances::on_initialize(1);
|
||||
Balances::on_runtime_upgrade();
|
||||
assert_eq!(Balances::free_balance(1), 5);
|
||||
assert_eq!(Balances::reserved_balance(1), 5);
|
||||
assert_eq!(Balances::usable_balance(&1), 2);
|
||||
|
||||
@@ -75,13 +75,16 @@
|
||||
#![cfg_attr(not(feature = "std"), no_std)]
|
||||
|
||||
use sp_std::{prelude::*, marker::PhantomData};
|
||||
use frame_support::weights::{GetDispatchInfo, WeighBlock, DispatchInfo};
|
||||
use frame_support::{
|
||||
storage::StorageValue,
|
||||
weights::{GetDispatchInfo, WeighBlock, DispatchInfo}
|
||||
};
|
||||
use sp_runtime::{
|
||||
generic::Digest,
|
||||
ApplyExtrinsicResult,
|
||||
traits::{
|
||||
self, Header, Zero, One, Checkable, Applyable, CheckEqual, OnFinalize, OnInitialize,
|
||||
NumberFor, Block as BlockT, OffchainWorker, Dispatchable, Saturating,
|
||||
NumberFor, Block as BlockT, OffchainWorker, Dispatchable, Saturating, OnRuntimeUpgrade,
|
||||
},
|
||||
transaction_validity::TransactionValidity,
|
||||
};
|
||||
@@ -110,6 +113,7 @@ impl<
|
||||
Context: Default,
|
||||
UnsignedValidator,
|
||||
AllModules:
|
||||
OnRuntimeUpgrade +
|
||||
OnInitialize<System::BlockNumber> +
|
||||
OnFinalize<System::BlockNumber> +
|
||||
OffchainWorker<System::BlockNumber> +
|
||||
@@ -135,6 +139,7 @@ impl<
|
||||
Context: Default,
|
||||
UnsignedValidator,
|
||||
AllModules:
|
||||
OnRuntimeUpgrade +
|
||||
OnInitialize<System::BlockNumber> +
|
||||
OnFinalize<System::BlockNumber> +
|
||||
OffchainWorker<System::BlockNumber> +
|
||||
@@ -176,6 +181,12 @@ where
|
||||
extrinsics_root: &System::Hash,
|
||||
digest: &Digest<System::Hash>,
|
||||
) {
|
||||
if frame_system::RuntimeUpgraded::take() {
|
||||
<AllModules as OnRuntimeUpgrade>::on_runtime_upgrade();
|
||||
<frame_system::Module<System>>::register_extra_weight_unchecked(
|
||||
<AllModules as WeighBlock<System::BlockNumber>>::on_runtime_upgrade()
|
||||
);
|
||||
}
|
||||
<frame_system::Module<System>>::initialize(
|
||||
block_number,
|
||||
parent_hash,
|
||||
@@ -573,7 +584,7 @@ mod tests {
|
||||
header: Header {
|
||||
parent_hash: [69u8; 32].into(),
|
||||
number: 1,
|
||||
state_root: hex!("17caebd966d10cc6dc9659edf7fa3196511593f6c39f80f9b97cdbc3b0855cf3").into(),
|
||||
state_root: hex!("8a22606e925c39bb0c8e8f6f5871c0aceab88a2fcff6b2d92660af8f6daff0b1").into(),
|
||||
extrinsics_root: hex!("03170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314").into(),
|
||||
digest: Digest { logs: vec![], },
|
||||
},
|
||||
|
||||
@@ -254,6 +254,7 @@ mod mock;
|
||||
#[cfg(test)]
|
||||
mod tests;
|
||||
mod slashing;
|
||||
mod migration;
|
||||
|
||||
pub mod inflation;
|
||||
|
||||
@@ -679,6 +680,21 @@ impl Default for Forcing {
|
||||
fn default() -> Self { Forcing::NotForcing }
|
||||
}
|
||||
|
||||
// A value placed in storage that represents the current version of the Staking storage.
|
||||
// This value is used by the `on_runtime_upgrade` logic to determine whether we run
|
||||
// storage migration logic. This should match directly with the semantic versions of the Rust crate.
|
||||
#[derive(Encode, Decode, Clone, Copy, PartialEq, Eq, RuntimeDebug)]
|
||||
enum Releases {
|
||||
V1_0_0,
|
||||
V2_0_0,
|
||||
}
|
||||
|
||||
impl Default for Releases {
|
||||
fn default() -> Self {
|
||||
Releases::V1_0_0
|
||||
}
|
||||
}
|
||||
|
||||
decl_storage! {
|
||||
trait Store for Module<T: Trait> as Staking {
|
||||
/// Number of era to keep in history.
|
||||
@@ -832,10 +848,10 @@ decl_storage! {
|
||||
/// The earliest era for which we have a pending, unapplied slash.
|
||||
EarliestUnappliedSlash: Option<EraIndex>;
|
||||
|
||||
/// True if network has been upgraded to this version.
|
||||
/// Storage version of the pallet.
|
||||
///
|
||||
/// True for new networks.
|
||||
IsUpgraded build(|_| true): bool;
|
||||
/// This is set to v2.0.0 for new networks.
|
||||
StorageVersion build(|_: &GenesisConfig<T>| Releases::V2_0_0): Releases;
|
||||
}
|
||||
add_extra_genesis {
|
||||
config(stakers):
|
||||
@@ -927,8 +943,8 @@ decl_module! {
|
||||
|
||||
fn deposit_event() = default;
|
||||
|
||||
fn on_initialize() {
|
||||
Self::ensure_storage_upgraded();
|
||||
fn on_runtime_upgrade() {
|
||||
migration::on_runtime_upgrade::<T>();
|
||||
}
|
||||
|
||||
fn on_finalize() {
|
||||
@@ -1132,8 +1148,6 @@ decl_module! {
|
||||
/// # </weight>
|
||||
#[weight = SimpleDispatchInfo::FixedNormal(750_000)]
|
||||
fn validate(origin, prefs: ValidatorPrefs) {
|
||||
Self::ensure_storage_upgraded();
|
||||
|
||||
let controller = ensure_signed(origin)?;
|
||||
let ledger = Self::ledger(&controller).ok_or(Error::<T>::NotController)?;
|
||||
let stash = &ledger.stash;
|
||||
@@ -1154,8 +1168,6 @@ decl_module! {
|
||||
/// # </weight>
|
||||
#[weight = SimpleDispatchInfo::FixedNormal(750_000)]
|
||||
fn nominate(origin, targets: Vec<<T::Lookup as StaticLookup>::Source>) {
|
||||
Self::ensure_storage_upgraded();
|
||||
|
||||
let controller = ensure_signed(origin)?;
|
||||
let ledger = Self::ledger(&controller).ok_or(Error::<T>::NotController)?;
|
||||
let stash = &ledger.stash;
|
||||
@@ -1433,7 +1445,6 @@ decl_module! {
|
||||
///
|
||||
/// - `stash`: The stash account to reap. Its balance must be zero.
|
||||
fn reap_stash(_origin, stash: T::AccountId) {
|
||||
Self::ensure_storage_upgraded();
|
||||
ensure!(T::Currency::total_balance(&stash).is_zero(), Error::<T>::FundedTarget);
|
||||
Self::kill_stash(&stash)?;
|
||||
T::Currency::remove_lock(STAKING_ID, &stash);
|
||||
@@ -1577,14 +1588,6 @@ impl<T: Trait> Module<T> {
|
||||
<Nominators<T>>::remove(stash);
|
||||
}
|
||||
|
||||
/// Ensures storage is upgraded to most recent necessary state.
|
||||
fn ensure_storage_upgraded() {
|
||||
if !IsUpgraded::get() {
|
||||
IsUpgraded::put(true);
|
||||
Self::do_upgrade();
|
||||
}
|
||||
}
|
||||
|
||||
/// Actually make a payment to a staker. This uses the currency's reward function
|
||||
/// to pay the right payee for the given staker account.
|
||||
fn make_payout(stash: &T::AccountId, amount: BalanceOf<T>) -> Option<PositiveImbalanceOf<T>> {
|
||||
@@ -1958,147 +1961,6 @@ impl<T: Trait> Module<T> {
|
||||
_ => ForceEra::put(Forcing::ForceNew),
|
||||
}
|
||||
}
|
||||
|
||||
/// Update storages to current version
|
||||
///
|
||||
/// In old version the staking module has several issue about handling session delay, the
|
||||
/// current era was always considered the active one.
|
||||
///
|
||||
/// After the migration the current era will still be considered the active one for the era of
|
||||
/// the upgrade. And the delay issue will be fixed when planning the next era.
|
||||
// * create:
|
||||
// * ActiveEraStart
|
||||
// * ErasRewardPoints
|
||||
// * ActiveEra
|
||||
// * ErasStakers
|
||||
// * ErasStakersClipped
|
||||
// * ErasValidatorPrefs
|
||||
// * ErasTotalStake
|
||||
// * ErasStartSessionIndex
|
||||
// * translate StakingLedger
|
||||
// * removal of:
|
||||
// * Stakers
|
||||
// * SlotStake
|
||||
// * CurrentElected
|
||||
// * CurrentEraStart
|
||||
// * CurrentEraStartSessionIndex
|
||||
// * CurrentEraPointsEarned
|
||||
fn do_upgrade() {
|
||||
/// Deprecated storages used for migration only.
|
||||
mod deprecated {
|
||||
use crate::{Trait, BalanceOf, MomentOf, SessionIndex, Exposure};
|
||||
use codec::{Encode, Decode};
|
||||
use frame_support::{decl_module, decl_storage};
|
||||
use sp_std::prelude::*;
|
||||
|
||||
/// Reward points of an era. Used to split era total payout between validators.
|
||||
#[derive(Encode, Decode, Default)]
|
||||
pub struct EraPoints {
|
||||
/// Total number of points. Equals the sum of reward points for each validator.
|
||||
pub total: u32,
|
||||
/// The reward points earned by a given validator. The index of this vec corresponds to the
|
||||
/// index into the current validator set.
|
||||
pub individual: Vec<u32>,
|
||||
}
|
||||
|
||||
decl_module! {
|
||||
pub struct Module<T: Trait> for enum Call where origin: T::Origin { }
|
||||
}
|
||||
|
||||
decl_storage! {
|
||||
pub trait Store for Module<T: Trait> as Staking {
|
||||
pub SlotStake: BalanceOf<T>;
|
||||
|
||||
/// The currently elected validator set keyed by stash account ID.
|
||||
pub CurrentElected: Vec<T::AccountId>;
|
||||
|
||||
/// The start of the current era.
|
||||
pub CurrentEraStart: MomentOf<T>;
|
||||
|
||||
/// The session index at which the current era started.
|
||||
pub CurrentEraStartSessionIndex: SessionIndex;
|
||||
|
||||
/// Rewards for the current era. Using indices of current elected set.
|
||||
pub CurrentEraPointsEarned: EraPoints;
|
||||
|
||||
/// Nominators for a particular account that is in action right now. You can't iterate
|
||||
/// through validators here, but you can find them in the Session module.
|
||||
///
|
||||
/// This is keyed by the stash account.
|
||||
pub Stakers: map hasher(blake2_256) T::AccountId => Exposure<T::AccountId, BalanceOf<T>>;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Encode, Decode)]
|
||||
struct OldStakingLedger<AccountId, Balance: HasCompact> {
|
||||
stash: AccountId,
|
||||
#[codec(compact)]
|
||||
total: Balance,
|
||||
#[codec(compact)]
|
||||
active: Balance,
|
||||
unlocking: Vec<UnlockChunk<Balance>>,
|
||||
}
|
||||
|
||||
let current_era_start_index = deprecated::CurrentEraStartSessionIndex::get();
|
||||
let current_era = <Module<T> as Store>::CurrentEra::get().unwrap_or(0);
|
||||
let current_era_start = deprecated::CurrentEraStart::<T>::get();
|
||||
<Module<T> as Store>::ErasStartSessionIndex::insert(current_era, current_era_start_index);
|
||||
<Module<T> as Store>::ActiveEra::put(ActiveEraInfo {
|
||||
index: current_era,
|
||||
start: Some(current_era_start),
|
||||
});
|
||||
|
||||
let current_elected = deprecated::CurrentElected::<T>::get();
|
||||
let mut current_total_stake = <BalanceOf<T>>::zero();
|
||||
for validator in ¤t_elected {
|
||||
let exposure = deprecated::Stakers::<T>::get(validator);
|
||||
current_total_stake += exposure.total;
|
||||
<Module<T> as Store>::ErasStakers::insert(current_era, validator, &exposure);
|
||||
|
||||
let mut exposure_clipped = exposure;
|
||||
let clipped_max_len = T::MaxNominatorRewardedPerValidator::get() as usize;
|
||||
if exposure_clipped.others.len() > clipped_max_len {
|
||||
exposure_clipped.others.sort_unstable_by(|a, b| a.value.cmp(&b.value).reverse());
|
||||
exposure_clipped.others.truncate(clipped_max_len);
|
||||
}
|
||||
<Module<T> as Store>::ErasStakersClipped::insert(current_era, validator, exposure_clipped);
|
||||
|
||||
let pref = <Module<T> as Store>::Validators::get(validator);
|
||||
<Module<T> as Store>::ErasValidatorPrefs::insert(current_era, validator, pref);
|
||||
}
|
||||
<Module<T> as Store>::ErasTotalStake::insert(current_era, current_total_stake);
|
||||
|
||||
let points = deprecated::CurrentEraPointsEarned::get();
|
||||
<Module<T> as Store>::ErasRewardPoints::insert(current_era, EraRewardPoints {
|
||||
total: points.total,
|
||||
individual: current_elected.iter().cloned().zip(points.individual.iter().cloned()).collect(),
|
||||
});
|
||||
|
||||
let res = <Module<T> as Store>::Ledger::translate_values(
|
||||
|old: OldStakingLedger<T::AccountId, BalanceOf<T>>| StakingLedger {
|
||||
stash: old.stash,
|
||||
total: old.total,
|
||||
active: old.active,
|
||||
unlocking: old.unlocking,
|
||||
last_reward: None,
|
||||
}
|
||||
);
|
||||
if let Err(e) = res {
|
||||
frame_support::print("Encountered error in migration of Staking::Ledger map.");
|
||||
frame_support::print("The number of removed key/value is:");
|
||||
frame_support::print(e);
|
||||
}
|
||||
|
||||
|
||||
// Kill old storages
|
||||
deprecated::Stakers::<T>::remove_all();
|
||||
deprecated::SlotStake::<T>::kill();
|
||||
deprecated::CurrentElected::<T>::kill();
|
||||
deprecated::CurrentEraStart::<T>::kill();
|
||||
deprecated::CurrentEraStartSessionIndex::kill();
|
||||
deprecated::CurrentEraPointsEarned::kill();
|
||||
}
|
||||
}
|
||||
|
||||
/// In this implementation `new_session(session)` must be called before `end_session(session-1)`
|
||||
@@ -2108,7 +1970,6 @@ impl<T: Trait> Module<T> {
|
||||
/// some session can lag in between the newest session planned and the latest session started.
|
||||
impl<T: Trait> pallet_session::SessionManager<T::AccountId> for Module<T> {
|
||||
fn new_session(new_index: SessionIndex) -> Option<Vec<T::AccountId>> {
|
||||
Self::ensure_storage_upgraded();
|
||||
Self::new_session(new_index)
|
||||
}
|
||||
fn start_session(start_index: SessionIndex) {
|
||||
@@ -2208,8 +2069,6 @@ impl <T: Trait> OnOffenceHandler<T::AccountId, pallet_session::historical::Ident
|
||||
slash_fraction: &[Perbill],
|
||||
slash_session: SessionIndex,
|
||||
) {
|
||||
<Module<T>>::ensure_storage_upgraded();
|
||||
|
||||
let reward_proportion = SlashRewardFraction::get();
|
||||
|
||||
let active_era = {
|
||||
@@ -2297,8 +2156,6 @@ impl<T, Reporter, Offender, R, O> ReportOffence<Reporter, Offender, O>
|
||||
O: Offence<Offender>,
|
||||
{
|
||||
fn report_offence(reporters: Vec<Reporter>, offence: O) -> Result<(), OffenceError> {
|
||||
<Module<T>>::ensure_storage_upgraded();
|
||||
|
||||
// disallow any slashing from before the current bonding period.
|
||||
let offence_session = offence.session_index();
|
||||
let bonded_eras = BondedEras::get();
|
||||
|
||||
@@ -0,0 +1,73 @@
|
||||
// Copyright 2020 Parity Technologies (UK) Ltd.
|
||||
// This file is part of Substrate.
|
||||
|
||||
// Substrate 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.
|
||||
|
||||
// Substrate 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 Substrate. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
/// Deprecated storages used for migration from v1.0.0 to v2.0.0 only.
|
||||
|
||||
use crate::{Trait, BalanceOf, MomentOf, SessionIndex, Exposure, UnlockChunk};
|
||||
use codec::{Encode, Decode, HasCompact};
|
||||
use frame_support::{decl_module, decl_storage};
|
||||
use sp_std::prelude::*;
|
||||
|
||||
/// Reward points of an era. Used to split era total payout between validators.
|
||||
#[derive(Encode, Decode, Default)]
|
||||
pub struct EraPoints {
|
||||
/// Total number of points. Equals the sum of reward points for each validator.
|
||||
pub total: u32,
|
||||
/// The reward points earned by a given validator. The index of this vec corresponds to the
|
||||
/// index into the current validator set.
|
||||
pub individual: Vec<u32>,
|
||||
}
|
||||
|
||||
#[derive(Encode, Decode)]
|
||||
pub struct OldStakingLedger<AccountId, Balance: HasCompact> {
|
||||
pub stash: AccountId,
|
||||
#[codec(compact)]
|
||||
pub total: Balance,
|
||||
#[codec(compact)]
|
||||
pub active: Balance,
|
||||
pub unlocking: Vec<UnlockChunk<Balance>>,
|
||||
}
|
||||
|
||||
decl_module! {
|
||||
pub struct Module<T: Trait> for enum Call where origin: T::Origin { }
|
||||
}
|
||||
|
||||
decl_storage! {
|
||||
pub trait Store for Module<T: Trait> as Staking {
|
||||
pub SlotStake: BalanceOf<T>;
|
||||
|
||||
/// The currently elected validator set keyed by stash account ID.
|
||||
pub CurrentElected: Vec<T::AccountId>;
|
||||
|
||||
/// The start of the current era.
|
||||
pub CurrentEraStart: MomentOf<T>;
|
||||
|
||||
/// The session index at which the current era started.
|
||||
pub CurrentEraStartSessionIndex: SessionIndex;
|
||||
|
||||
/// Rewards for the current era. Using indices of current elected set.
|
||||
pub CurrentEraPointsEarned: EraPoints;
|
||||
|
||||
/// Nominators for a particular account that is in action right now. You can't iterate
|
||||
/// through validators here, but you can find them in the Session module.
|
||||
///
|
||||
/// This is keyed by the stash account.
|
||||
pub Stakers: map hasher(blake2_256) T::AccountId => Exposure<T::AccountId, BalanceOf<T>>;
|
||||
|
||||
/// Old upgrade flag.
|
||||
pub IsUpgraded: bool;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,119 @@
|
||||
// Copyright 2020 Parity Technologies (UK) Ltd.
|
||||
// This file is part of Substrate.
|
||||
|
||||
// Substrate 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.
|
||||
|
||||
// Substrate 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 Substrate. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
//! Update storage from v1.0.0 to v2.0.0
|
||||
//!
|
||||
//! In old version the staking module has several issue about handling session delay, the
|
||||
//! current era was always considered the active one.
|
||||
//!
|
||||
//! After the migration the current era will still be considered the active one for the era of
|
||||
//! the upgrade. And the delay issue will be fixed when planning the next era.
|
||||
// * create:
|
||||
// * ActiveEraStart
|
||||
// * ErasRewardPoints
|
||||
// * ActiveEra
|
||||
// * ErasStakers
|
||||
// * ErasStakersClipped
|
||||
// * ErasValidatorPrefs
|
||||
// * ErasTotalStake
|
||||
// * ErasStartSessionIndex
|
||||
// * translate StakingLedger
|
||||
// * removal of:
|
||||
// * Stakers
|
||||
// * SlotStake
|
||||
// * CurrentElected
|
||||
// * CurrentEraStart
|
||||
// * CurrentEraStartSessionIndex
|
||||
// * CurrentEraPointsEarned
|
||||
|
||||
use super::*;
|
||||
mod deprecated;
|
||||
#[cfg(test)]
|
||||
mod tests;
|
||||
#[cfg(test)]
|
||||
mod test_upgrade_from_master_dataset;
|
||||
|
||||
pub fn on_runtime_upgrade<T: Trait>() {
|
||||
match StorageVersion::get() {
|
||||
Releases::V2_0_0 => return,
|
||||
Releases::V1_0_0 => upgrade_v1_to_v2::<T>(),
|
||||
}
|
||||
}
|
||||
|
||||
fn upgrade_v1_to_v2<T: Trait>() {
|
||||
deprecated::IsUpgraded::kill();
|
||||
|
||||
let current_era_start_index = deprecated::CurrentEraStartSessionIndex::get();
|
||||
let current_era = <Module<T> as Store>::CurrentEra::get().unwrap_or(0);
|
||||
let current_era_start = deprecated::CurrentEraStart::<T>::get();
|
||||
<Module<T> as Store>::ErasStartSessionIndex::insert(current_era, current_era_start_index);
|
||||
<Module<T> as Store>::ActiveEra::put(ActiveEraInfo {
|
||||
index: current_era,
|
||||
start: Some(current_era_start),
|
||||
});
|
||||
|
||||
let current_elected = deprecated::CurrentElected::<T>::get();
|
||||
let mut current_total_stake = <BalanceOf<T>>::zero();
|
||||
for validator in ¤t_elected {
|
||||
let exposure = deprecated::Stakers::<T>::get(validator);
|
||||
current_total_stake += exposure.total;
|
||||
<Module<T> as Store>::ErasStakers::insert(current_era, validator, &exposure);
|
||||
|
||||
let mut exposure_clipped = exposure;
|
||||
let clipped_max_len = T::MaxNominatorRewardedPerValidator::get() as usize;
|
||||
if exposure_clipped.others.len() > clipped_max_len {
|
||||
exposure_clipped.others.sort_unstable_by(|a, b| a.value.cmp(&b.value).reverse());
|
||||
exposure_clipped.others.truncate(clipped_max_len);
|
||||
}
|
||||
<Module<T> as Store>::ErasStakersClipped::insert(current_era, validator, exposure_clipped);
|
||||
|
||||
let pref = <Module<T> as Store>::Validators::get(validator);
|
||||
<Module<T> as Store>::ErasValidatorPrefs::insert(current_era, validator, pref);
|
||||
}
|
||||
<Module<T> as Store>::ErasTotalStake::insert(current_era, current_total_stake);
|
||||
|
||||
let points = deprecated::CurrentEraPointsEarned::get();
|
||||
<Module<T> as Store>::ErasRewardPoints::insert(current_era, EraRewardPoints {
|
||||
total: points.total,
|
||||
individual: current_elected.iter().cloned().zip(points.individual.iter().cloned()).collect(),
|
||||
});
|
||||
|
||||
let res = <Module<T> as Store>::Ledger::translate_values(
|
||||
|old: deprecated::OldStakingLedger<T::AccountId, BalanceOf<T>>| StakingLedger {
|
||||
stash: old.stash,
|
||||
total: old.total,
|
||||
active: old.active,
|
||||
unlocking: old.unlocking,
|
||||
last_reward: None,
|
||||
}
|
||||
);
|
||||
if let Err(e) = res {
|
||||
frame_support::print("Encountered error in migration of Staking::Ledger map.");
|
||||
frame_support::print("The number of removed key/value is:");
|
||||
frame_support::print(e);
|
||||
}
|
||||
|
||||
|
||||
// Kill old storages
|
||||
deprecated::Stakers::<T>::remove_all();
|
||||
deprecated::SlotStake::<T>::kill();
|
||||
deprecated::CurrentElected::<T>::kill();
|
||||
deprecated::CurrentEraStart::<T>::kill();
|
||||
deprecated::CurrentEraStartSessionIndex::kill();
|
||||
deprecated::CurrentEraPointsEarned::kill();
|
||||
|
||||
StorageVersion::put(Releases::V2_0_0);
|
||||
}
|
||||
@@ -0,0 +1,220 @@
|
||||
use crate::*;
|
||||
use crate::mock::*;
|
||||
use frame_support::storage::migration::*;
|
||||
use sp_core::hashing::blake2_256;
|
||||
use super::test_upgrade_from_master_dataset;
|
||||
use sp_runtime::traits::OnRuntimeUpgrade;
|
||||
|
||||
#[test]
|
||||
fn upgrade_works() {
|
||||
ExtBuilder::default().build().execute_with(|| {
|
||||
start_era(3);
|
||||
|
||||
assert_eq!(Session::validators(), vec![21, 11]);
|
||||
|
||||
// Insert fake data to check the migration
|
||||
put_storage_value::<Vec<AccountId>>(b"Staking", b"CurrentElected", b"", vec![21, 31]);
|
||||
put_storage_value::<SessionIndex>(b"Staking", b"CurrentEraStartSessionIndex", b"", 5);
|
||||
put_storage_value::<MomentOf<Test>>(b"Staking", b"CurrentEraStart", b"", 777);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&11u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 10,
|
||||
own: 10,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&21u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&31u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value::<(u32, Vec<u32>)>(b"Staking", b"CurrentEraPointsEarned", b"", (12, vec![2, 10]));
|
||||
<Staking as Store>::ErasStakers::remove_all();
|
||||
<Staking as Store>::ErasStakersClipped::remove_all();
|
||||
|
||||
<Staking as Store>::StorageVersion::put(Releases::V1_0_0);
|
||||
|
||||
// Perform upgrade
|
||||
Staking::on_runtime_upgrade();
|
||||
|
||||
assert_eq!(<Staking as Store>::StorageVersion::get(), Releases::V2_0_0);
|
||||
|
||||
// Check migration
|
||||
assert_eq!(<Staking as Store>::ErasStartSessionIndex::get(3).unwrap(), 5);
|
||||
assert_eq!(<Staking as Store>::ErasRewardPoints::get(3), EraRewardPoints {
|
||||
total: 12,
|
||||
individual: vec![(21, 2), (31, 10)].into_iter().collect(),
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ActiveEra::get().unwrap().index, 3);
|
||||
assert_eq!(<Staking as Store>::ActiveEra::get().unwrap().start, Some(777));
|
||||
assert_eq!(<Staking as Store>::CurrentEra::get().unwrap(), 3);
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 11), Exposure {
|
||||
total: 0,
|
||||
own: 0,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 21), Exposure {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 31), Exposure {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 11), Exposure {
|
||||
total: 0,
|
||||
own: 0,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 21), Exposure {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 31), Exposure {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::get(3, 21), Staking::validators(21));
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::get(3, 31), Staking::validators(31));
|
||||
assert_eq!(<Staking as Store>::ErasTotalStake::get(3), 50);
|
||||
})
|
||||
}
|
||||
|
||||
// Test that an upgrade from previous test environment works.
|
||||
#[test]
|
||||
fn test_upgrade_from_master_works() {
|
||||
let data_sets = &[
|
||||
test_upgrade_from_master_dataset::_0,
|
||||
test_upgrade_from_master_dataset::_1,
|
||||
test_upgrade_from_master_dataset::_2,
|
||||
test_upgrade_from_master_dataset::_3,
|
||||
test_upgrade_from_master_dataset::_4,
|
||||
test_upgrade_from_master_dataset::_5,
|
||||
test_upgrade_from_master_dataset::_6,
|
||||
test_upgrade_from_master_dataset::_7,
|
||||
test_upgrade_from_master_dataset::_8,
|
||||
];
|
||||
for data_set in data_sets.iter() {
|
||||
let mut storage = sp_runtime::Storage::default();
|
||||
for (key, value) in data_set.iter() {
|
||||
storage.top.insert(key.to_vec(), value.to_vec());
|
||||
}
|
||||
let mut ext = sp_io::TestExternalities::from(storage);
|
||||
ext.execute_with(|| {
|
||||
let old_stakers =
|
||||
get_storage_value::<Vec<AccountId>>(b"Staking", b"CurrentElected", b"").unwrap();
|
||||
let old_staker_0 = old_stakers[0];
|
||||
let old_staker_1 = old_stakers[1];
|
||||
let old_current_era =
|
||||
get_storage_value::<EraIndex>(b"Staking", b"CurrentEra", b"").unwrap();
|
||||
let old_staker_0_exposure = get_storage_value::<Exposure<AccountId, Balance>>(
|
||||
b"Staking", b"Stakers", &blake2_256(&old_staker_0.encode())
|
||||
).unwrap();
|
||||
let old_staker_1_exposure = get_storage_value::<Exposure<AccountId, Balance>>(
|
||||
b"Staking", b"Stakers", &blake2_256(&old_staker_1.encode())
|
||||
).unwrap();
|
||||
let (
|
||||
old_era_points_earned_total,
|
||||
old_era_points_earned_individual
|
||||
) = get_storage_value::<(u32, Vec<u32>)>(b"Staking", b"CurrentEraPointsEarned", b"")
|
||||
.unwrap_or((0, vec![]));
|
||||
|
||||
Staking::on_runtime_upgrade();
|
||||
assert!(<Staking as Store>::StorageVersion::get() == Releases::V2_0_0);
|
||||
|
||||
// Check ActiveEra and CurrentEra
|
||||
let active_era = Staking::active_era().unwrap().index;
|
||||
let current_era = Staking::current_era().unwrap();
|
||||
assert!(current_era == active_era);
|
||||
assert!(current_era == old_current_era);
|
||||
|
||||
// Check ErasStartSessionIndex
|
||||
let active_era_start = Staking::eras_start_session_index(active_era).unwrap();
|
||||
let current_era_start = Staking::eras_start_session_index(current_era).unwrap();
|
||||
let current_session_index = Session::current_index();
|
||||
assert!(current_era_start == active_era_start);
|
||||
assert!(active_era_start <= current_session_index);
|
||||
assert_eq!(<Staking as Store>::ErasStartSessionIndex::iter().count(), 1);
|
||||
|
||||
// Check ErasStakers
|
||||
assert_eq!(<Staking as Store>::ErasStakers::iter().count(), 2);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakers::get(current_era, old_staker_0),
|
||||
old_staker_0_exposure
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakers::get(current_era, old_staker_1),
|
||||
old_staker_1_exposure
|
||||
);
|
||||
|
||||
// Check ErasStakersClipped
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::iter().count(), 2);
|
||||
assert!(<Staking as Store>::ErasStakersClipped::iter().all(|exposure_clipped| {
|
||||
let max = <Test as Trait>::MaxNominatorRewardedPerValidator::get() as usize;
|
||||
exposure_clipped.others.len() <= max
|
||||
}));
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakersClipped::get(current_era, old_staker_0),
|
||||
old_staker_0_exposure
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakersClipped::get(current_era, old_staker_1),
|
||||
old_staker_1_exposure
|
||||
);
|
||||
|
||||
// Check ErasValidatorPrefs
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::iter().count(), 2);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasValidatorPrefs::get(current_era, old_staker_0),
|
||||
Staking::validators(old_staker_0)
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasValidatorPrefs::get(current_era, old_staker_1),
|
||||
Staking::validators(old_staker_1)
|
||||
);
|
||||
|
||||
// Check ErasTotalStake
|
||||
assert_eq!(<Staking as Store>::ErasTotalStake::iter().count(), 1);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasTotalStake::get(current_era),
|
||||
old_staker_0_exposure.total + old_staker_1_exposure.total
|
||||
);
|
||||
|
||||
// Check ErasRewardPoints
|
||||
assert_eq!(<Staking as Store>::ErasRewardPoints::iter().count(), 1);
|
||||
let mut individual = BTreeMap::new();
|
||||
if let Some(p) = old_era_points_earned_individual.get(0) {
|
||||
individual.insert(old_staker_0, p.clone());
|
||||
}
|
||||
if let Some(p) = old_era_points_earned_individual.get(1) {
|
||||
individual.insert(old_staker_1, p.clone());
|
||||
}
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasRewardPoints::get(current_era),
|
||||
EraRewardPoints {
|
||||
total: old_era_points_earned_total,
|
||||
individual,
|
||||
}
|
||||
);
|
||||
|
||||
// Check ErasValidatorReward
|
||||
assert_eq!(<Staking as Store>::ErasValidatorReward::iter().count(), 0);
|
||||
});
|
||||
}
|
||||
}
|
||||
@@ -16,8 +16,6 @@
|
||||
|
||||
//! Tests for the module.
|
||||
|
||||
mod test_upgrade_from_master_dataset;
|
||||
|
||||
use super::*;
|
||||
use mock::*;
|
||||
use sp_runtime::{assert_eq_error_rate, traits::{OnInitialize, BadOrigin}};
|
||||
@@ -26,10 +24,8 @@ use frame_support::{
|
||||
assert_ok, assert_noop,
|
||||
traits::{Currency, ReservableCurrency},
|
||||
StorageMap,
|
||||
storage::migration::{put_storage_value, get_storage_value},
|
||||
};
|
||||
use pallet_balances::Error as BalancesError;
|
||||
use sp_io::hashing::blake2_256;
|
||||
use substrate_test_utils::assert_eq_uvec;
|
||||
use crate::Store;
|
||||
|
||||
@@ -2843,97 +2839,6 @@ fn claim_reward_at_the_last_era_and_no_double_claim_and_invalid_claim() {
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn upgrade_works() {
|
||||
ExtBuilder::default().build().execute_with(|| {
|
||||
start_era(3);
|
||||
|
||||
assert_eq!(Session::validators(), vec![21, 11]);
|
||||
|
||||
// Insert fake data to check the migration
|
||||
put_storage_value::<Vec<AccountId>>(b"Staking", b"CurrentElected", b"", vec![21, 31]);
|
||||
put_storage_value::<SessionIndex>(b"Staking", b"CurrentEraStartSessionIndex", b"", 5);
|
||||
put_storage_value::<MomentOf<Test>>(b"Staking", b"CurrentEraStart", b"", 777);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&11u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 10,
|
||||
own: 10,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&21u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value(
|
||||
b"Staking", b"Stakers", &blake2_256(&31u64.encode()),
|
||||
Exposure::<AccountId, Balance> {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
}
|
||||
);
|
||||
put_storage_value::<(u32, Vec<u32>)>(b"Staking", b"CurrentEraPointsEarned", b"", (12, vec![2, 10]));
|
||||
<Staking as Store>::ErasStakers::remove_all();
|
||||
<Staking as Store>::ErasStakersClipped::remove_all();
|
||||
|
||||
<Staking as Store>::IsUpgraded::put(false);
|
||||
|
||||
// Perform upgrade
|
||||
Staking::ensure_storage_upgraded();
|
||||
|
||||
assert_eq!(<Staking as Store>::IsUpgraded::get(), true);
|
||||
|
||||
// Check migration
|
||||
assert_eq!(<Staking as Store>::ErasStartSessionIndex::get(3).unwrap(), 5);
|
||||
assert_eq!(<Staking as Store>::ErasRewardPoints::get(3), EraRewardPoints {
|
||||
total: 12,
|
||||
individual: vec![(21, 2), (31, 10)].into_iter().collect(),
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ActiveEra::get().unwrap().index, 3);
|
||||
assert_eq!(<Staking as Store>::ActiveEra::get().unwrap().start, Some(777));
|
||||
assert_eq!(<Staking as Store>::CurrentEra::get().unwrap(), 3);
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 11), Exposure {
|
||||
total: 0,
|
||||
own: 0,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 21), Exposure {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakers::get(3, 31), Exposure {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 11), Exposure {
|
||||
total: 0,
|
||||
own: 0,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 21), Exposure {
|
||||
total: 20,
|
||||
own: 20,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::get(3, 31), Exposure {
|
||||
total: 30,
|
||||
own: 30,
|
||||
others: vec![],
|
||||
});
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::get(3, 21), Staking::validators(21));
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::get(3, 31), Staking::validators(31));
|
||||
assert_eq!(<Staking as Store>::ErasTotalStake::get(3), 50);
|
||||
})
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn zero_slash_keeps_nominators() {
|
||||
ExtBuilder::default().build().execute_with(|| {
|
||||
@@ -3071,129 +2976,6 @@ fn test_max_nominator_rewarded_per_validator_and_cant_steal_someone_else_reward(
|
||||
});
|
||||
}
|
||||
|
||||
// Test that an upgrade from previous test environment works.
|
||||
#[test]
|
||||
fn test_upgrade_from_master_works() {
|
||||
let data_sets = &[
|
||||
test_upgrade_from_master_dataset::_0,
|
||||
test_upgrade_from_master_dataset::_1,
|
||||
test_upgrade_from_master_dataset::_2,
|
||||
test_upgrade_from_master_dataset::_3,
|
||||
test_upgrade_from_master_dataset::_4,
|
||||
test_upgrade_from_master_dataset::_5,
|
||||
test_upgrade_from_master_dataset::_6,
|
||||
test_upgrade_from_master_dataset::_7,
|
||||
test_upgrade_from_master_dataset::_8,
|
||||
];
|
||||
for data_set in data_sets.iter() {
|
||||
let mut storage = sp_runtime::Storage::default();
|
||||
for (key, value) in data_set.iter() {
|
||||
storage.top.insert(key.to_vec(), value.to_vec());
|
||||
}
|
||||
let mut ext = sp_io::TestExternalities::from(storage);
|
||||
ext.execute_with(|| {
|
||||
let old_stakers =
|
||||
get_storage_value::<Vec<AccountId>>(b"Staking", b"CurrentElected", b"").unwrap();
|
||||
let old_staker_0 = old_stakers[0];
|
||||
let old_staker_1 = old_stakers[1];
|
||||
let old_current_era =
|
||||
get_storage_value::<EraIndex>(b"Staking", b"CurrentEra", b"").unwrap();
|
||||
let old_staker_0_exposure = get_storage_value::<Exposure<AccountId, Balance>>(
|
||||
b"Staking", b"Stakers", &blake2_256(&old_staker_0.encode())
|
||||
).unwrap();
|
||||
let old_staker_1_exposure = get_storage_value::<Exposure<AccountId, Balance>>(
|
||||
b"Staking", b"Stakers", &blake2_256(&old_staker_1.encode())
|
||||
).unwrap();
|
||||
let (
|
||||
old_era_points_earned_total,
|
||||
old_era_points_earned_individual
|
||||
) = get_storage_value::<(u32, Vec<u32>)>(b"Staking", b"CurrentEraPointsEarned", b"")
|
||||
.unwrap_or((0, vec![]));
|
||||
|
||||
Staking::ensure_storage_upgraded();
|
||||
assert!(<Staking as Store>::IsUpgraded::get());
|
||||
|
||||
// Check ActiveEra and CurrentEra
|
||||
let active_era = Staking::active_era().unwrap().index;
|
||||
let current_era = Staking::current_era().unwrap();
|
||||
assert!(current_era == active_era);
|
||||
assert!(current_era == old_current_era);
|
||||
|
||||
// Check ErasStartSessionIndex
|
||||
let active_era_start = Staking::eras_start_session_index(active_era).unwrap();
|
||||
let current_era_start = Staking::eras_start_session_index(current_era).unwrap();
|
||||
let current_session_index = Session::current_index();
|
||||
assert!(current_era_start == active_era_start);
|
||||
assert!(active_era_start <= current_session_index);
|
||||
assert_eq!(<Staking as Store>::ErasStartSessionIndex::iter().count(), 1);
|
||||
|
||||
// Check ErasStakers
|
||||
assert_eq!(<Staking as Store>::ErasStakers::iter().count(), 2);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakers::get(current_era, old_staker_0),
|
||||
old_staker_0_exposure
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakers::get(current_era, old_staker_1),
|
||||
old_staker_1_exposure
|
||||
);
|
||||
|
||||
// Check ErasStakersClipped
|
||||
assert_eq!(<Staking as Store>::ErasStakersClipped::iter().count(), 2);
|
||||
assert!(<Staking as Store>::ErasStakersClipped::iter().all(|exposure_clipped| {
|
||||
let max = <Test as Trait>::MaxNominatorRewardedPerValidator::get() as usize;
|
||||
exposure_clipped.others.len() <= max
|
||||
}));
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakersClipped::get(current_era, old_staker_0),
|
||||
old_staker_0_exposure
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasStakersClipped::get(current_era, old_staker_1),
|
||||
old_staker_1_exposure
|
||||
);
|
||||
|
||||
// Check ErasValidatorPrefs
|
||||
assert_eq!(<Staking as Store>::ErasValidatorPrefs::iter().count(), 2);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasValidatorPrefs::get(current_era, old_staker_0),
|
||||
Staking::validators(old_staker_0)
|
||||
);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasValidatorPrefs::get(current_era, old_staker_1),
|
||||
Staking::validators(old_staker_1)
|
||||
);
|
||||
|
||||
// Check ErasTotalStake
|
||||
assert_eq!(<Staking as Store>::ErasTotalStake::iter().count(), 1);
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasTotalStake::get(current_era),
|
||||
old_staker_0_exposure.total + old_staker_1_exposure.total
|
||||
);
|
||||
|
||||
// Check ErasRewardPoints
|
||||
assert_eq!(<Staking as Store>::ErasRewardPoints::iter().count(), 1);
|
||||
let mut individual = BTreeMap::new();
|
||||
if let Some(p) = old_era_points_earned_individual.get(0) {
|
||||
individual.insert(old_staker_0, p.clone());
|
||||
}
|
||||
if let Some(p) = old_era_points_earned_individual.get(1) {
|
||||
individual.insert(old_staker_1, p.clone());
|
||||
}
|
||||
assert_eq!(
|
||||
<Staking as Store>::ErasRewardPoints::get(current_era),
|
||||
EraRewardPoints {
|
||||
total: old_era_points_earned_total,
|
||||
individual,
|
||||
}
|
||||
);
|
||||
|
||||
// Check ErasValidatorReward
|
||||
assert_eq!(<Staking as Store>::ErasValidatorReward::iter().count(), 0);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn set_history_depth_works() {
|
||||
ExtBuilder::default().build().execute_with(|| {
|
||||
|
||||
@@ -196,6 +196,10 @@ impl<T> Parameter for T where T: Codec + EncodeLike + Clone + Eq + fmt::Debug {}
|
||||
///
|
||||
/// The following reserved functions also take the block number (with type `T::BlockNumber`) as an optional input:
|
||||
///
|
||||
/// * `on_runtime_upgrade`: Executes at the beginning of a block prior to on_initialize when there
|
||||
/// is a runtime upgrade. This allows each module to upgrade its storage before the storage items are used.
|
||||
/// As such, **calling other modules must be avoided**!! Using this function will implement the
|
||||
/// [`OnRuntimeUpgrade`](../sp_runtime/traits/trait.OnRuntimeUpgrade.html) trait.
|
||||
/// * `on_initialize`: Executes at the beginning of a block. Using this function will
|
||||
/// implement the [`OnInitialize`](../sp_runtime/traits/trait.OnInitialize.html) trait.
|
||||
/// * `on_finalize`: Executes at the end of a block. Using this function will
|
||||
@@ -229,6 +233,7 @@ macro_rules! decl_module {
|
||||
{}
|
||||
{}
|
||||
{}
|
||||
{}
|
||||
[]
|
||||
$($t)*
|
||||
);
|
||||
@@ -261,6 +266,7 @@ macro_rules! decl_module {
|
||||
{}
|
||||
{}
|
||||
{}
|
||||
{}
|
||||
[]
|
||||
$($t)*
|
||||
);
|
||||
@@ -274,6 +280,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{}
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -290,6 +297,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $vis fn deposit_event() = default; }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -305,6 +313,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{}
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -326,6 +335,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{}
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -342,6 +352,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{
|
||||
#[weight = $crate::dispatch::SimpleDispatchInfo::zero()]
|
||||
fn on_finalize( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
@@ -361,6 +372,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{}
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -378,6 +390,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{
|
||||
#[weight = $weight]
|
||||
fn on_finalize( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
@@ -389,6 +402,85 @@ macro_rules! decl_module {
|
||||
$($rest)*
|
||||
);
|
||||
};
|
||||
// Add on_runtime_upgrade, without a given weight.
|
||||
(@normalize
|
||||
$(#[$attr:meta])*
|
||||
pub struct $mod_type:ident<
|
||||
$trait_instance:ident: $trait_name:ident$(<I>, I: $instantiable:path $(= $module_default_instance:path)?)?
|
||||
>
|
||||
for enum $call_type:ident where origin: $origin_type:ty, system = $system:ident
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{}
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
{ $( $error_type:tt )* }
|
||||
[ $( $dispatchables:tt )* ]
|
||||
$(#[doc = $doc_attr:tt])*
|
||||
fn on_runtime_upgrade( $( $param_name:ident : $param:ty ),* $(,)? ) { $( $impl:tt )* }
|
||||
$($rest:tt)*
|
||||
) => {
|
||||
$crate::decl_module!(@normalize
|
||||
$(#[$attr])*
|
||||
pub struct $mod_type<$trait_instance: $trait_name$(<I>, I: $instantiable $(= $module_default_instance)?)?>
|
||||
for enum $call_type where origin: $origin_type, system = $system
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{
|
||||
#[weight = $crate::dispatch::SimpleDispatchInfo::zero()]
|
||||
fn on_runtime_upgrade( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
}
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
{ $( $error_type )* }
|
||||
[ $( $dispatchables )* ]
|
||||
$($rest)*
|
||||
);
|
||||
};
|
||||
// Add on_runtime_upgrade, given weight.
|
||||
(@normalize
|
||||
$(#[$attr:meta])*
|
||||
pub struct $mod_type:ident<
|
||||
$trait_instance:ident: $trait_name:ident$(<I>, I: $instantiable:path $(= $module_default_instance:path)?)?
|
||||
>
|
||||
for enum $call_type:ident where origin: $origin_type:ty, system = $system:ident
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{}
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
{ $( $error_type:tt )* }
|
||||
[ $( $dispatchables:tt )* ]
|
||||
$(#[doc = $doc_attr:tt])*
|
||||
#[weight = $weight:expr]
|
||||
fn on_runtime_upgrade( $( $param_name:ident : $param:ty ),* $(,)? ) { $( $impl:tt )* }
|
||||
$($rest:tt)*
|
||||
) => {
|
||||
$crate::decl_module!(@normalize
|
||||
$(#[$attr])*
|
||||
pub struct $mod_type<$trait_instance: $trait_name$(<I>, I: $instantiable $(= $module_default_instance)?)?>
|
||||
for enum $call_type where origin: $origin_type, system = $system
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{
|
||||
#[weight = $weight]
|
||||
fn on_runtime_upgrade( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
}
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
{ $( $error_type )* }
|
||||
[ $( $dispatchables )* ]
|
||||
$($rest)*
|
||||
);
|
||||
};
|
||||
// Add on_initialize, without a given weight.
|
||||
(@normalize
|
||||
$(#[$attr:meta])*
|
||||
@@ -399,6 +491,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{}
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -418,6 +511,7 @@ macro_rules! decl_module {
|
||||
#[weight = $crate::dispatch::SimpleDispatchInfo::zero()]
|
||||
fn on_initialize( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
}
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -436,6 +530,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{}
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -456,6 +551,7 @@ macro_rules! decl_module {
|
||||
#[weight = $weight]
|
||||
fn on_initialize( $( $param_name : $param ),* ) { $( $impl )* }
|
||||
}
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -474,6 +570,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -492,6 +589,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ fn offchain_worker( $( $param_name : $param ),* ) { $( $impl )* } }
|
||||
{ $( $constants )* }
|
||||
@@ -512,6 +610,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -531,6 +630,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{
|
||||
@@ -555,6 +655,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -573,6 +674,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -592,6 +694,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -608,6 +711,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -628,6 +732,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -649,6 +754,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -676,6 +782,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -696,6 +803,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -717,6 +825,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -742,6 +851,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -767,6 +877,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -793,6 +904,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -808,6 +920,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds )* }
|
||||
{ $( $deposit_event )* }
|
||||
{ $( $on_initialize )* }
|
||||
{ $( $on_runtime_upgrade )* }
|
||||
{ $( $on_finalize )* }
|
||||
{ $( $offchain )* }
|
||||
{ $( $constants )* }
|
||||
@@ -903,6 +1016,39 @@ macro_rules! decl_module {
|
||||
{}
|
||||
};
|
||||
|
||||
(@impl_on_runtime_upgrade
|
||||
$module:ident<$trait_instance:ident: $trait_name:ident$(<I>, $instance:ident: $instantiable:path)?>;
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
#[weight = $weight:expr]
|
||||
fn on_runtime_upgrade() { $( $impl:tt )* }
|
||||
) => {
|
||||
impl<$trait_instance: $trait_name$(<I>, $instance: $instantiable)?>
|
||||
$crate::sp_runtime::traits::OnRuntimeUpgrade
|
||||
for $module<$trait_instance$(, $instance)?> where $( $other_where_bounds )*
|
||||
{
|
||||
fn on_runtime_upgrade() {
|
||||
use $crate::sp_std::if_std;
|
||||
if_std! {
|
||||
use $crate::tracing;
|
||||
let span = tracing::span!(tracing::Level::DEBUG, "on_runtime_upgrade");
|
||||
let _enter = span.enter();
|
||||
}
|
||||
{ $( $impl )* }
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
(@impl_on_runtime_upgrade
|
||||
$module:ident<$trait_instance:ident: $trait_name:ident$(<I>, $instance:ident: $instantiable:path)?>;
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
) => {
|
||||
impl<$trait_instance: $trait_name$(<I>, $instance: $instantiable)?>
|
||||
$crate::sp_runtime::traits::OnRuntimeUpgrade
|
||||
for $module<$trait_instance$(, $instance)?> where $( $other_where_bounds )*
|
||||
{}
|
||||
};
|
||||
|
||||
|
||||
(@impl_on_finalize
|
||||
$module:ident<$trait_instance:ident: $trait_name:ident$(<I>, $instance:ident: $instantiable:path)?>;
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
@@ -961,6 +1107,10 @@ macro_rules! decl_module {
|
||||
(@impl_block_hooks_weight
|
||||
$module:ident<$trait_instance:ident: $trait_name:ident$(<I>, $instance:ident: $instantiable:path)?>;
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
@runtime_upgrade $(
|
||||
#[weight = $weight_runtime_update:expr]
|
||||
fn on_runtime_upgrade($( $param_runtime_upgrade:ident : $param_ty_runtime_upgrade:ty )*) { $( $impl_runtime_upgrade:tt )* }
|
||||
)?
|
||||
@init $(
|
||||
#[weight = $weight_initialize:expr]
|
||||
fn on_initialize($( $param_initialize:ident : $param_ty_initialize:ty )*) { $( $impl_initialize:tt )* }
|
||||
@@ -974,6 +1124,11 @@ macro_rules! decl_module {
|
||||
$crate::dispatch::WeighBlock<$trait_instance::BlockNumber> for $module<$trait_instance$(, $instance)?> where
|
||||
$( $other_where_bounds )*
|
||||
{
|
||||
$(
|
||||
fn on_runtime_upgrade() -> $crate::dispatch::Weight {
|
||||
<dyn $crate::dispatch::WeighData<()>>::weigh_data(&$weight_initialize, ())
|
||||
}
|
||||
)?
|
||||
$(
|
||||
fn on_initialize(n: $trait_instance::BlockNumber) -> $crate::dispatch::Weight {
|
||||
<dyn $crate::dispatch::WeighData<$trait_instance::BlockNumber>>::weigh_data(&$weight_initialize, n)
|
||||
@@ -1208,6 +1363,7 @@ macro_rules! decl_module {
|
||||
{ $( $other_where_bounds:tt )* }
|
||||
{ $( $deposit_event:tt )* }
|
||||
{ $( $on_initialize:tt )* }
|
||||
{ $( $on_runtime_upgrade:tt )* }
|
||||
{ $( $on_finalize:tt )* }
|
||||
{ $( $offchain:tt )* }
|
||||
{ $( $constants:tt )* }
|
||||
@@ -1231,6 +1387,14 @@ macro_rules! decl_module {
|
||||
$( $on_initialize )*
|
||||
}
|
||||
|
||||
$crate::decl_module! {
|
||||
@impl_on_runtime_upgrade
|
||||
$mod_type<$trait_instance: $trait_name $(<I>, $instance: $instantiable)?>;
|
||||
{ $( $other_where_bounds )* }
|
||||
$( $on_runtime_upgrade )*
|
||||
}
|
||||
|
||||
|
||||
$crate::decl_module! {
|
||||
@impl_on_finalize
|
||||
$mod_type<$trait_instance: $trait_name $(<I>, $instance: $instantiable)?>;
|
||||
@@ -1242,6 +1406,7 @@ macro_rules! decl_module {
|
||||
@impl_block_hooks_weight
|
||||
$mod_type<$trait_instance: $trait_name $(<I>, $instance: $instantiable)?>;
|
||||
{ $( $other_where_bounds )* }
|
||||
@runtime_upgrade $( $on_runtime_upgrade )*
|
||||
@init $( $on_initialize )*
|
||||
@fin $( $on_finalize )*
|
||||
}
|
||||
@@ -1869,15 +2034,12 @@ macro_rules! __check_reserved_fn_name {
|
||||
(on_initialize $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error on_initialize);
|
||||
};
|
||||
(on_initialise $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error_renamed on_initialise on_initialize);
|
||||
(on_runtime_upgrade $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error on_runtime_upgrade);
|
||||
};
|
||||
(on_finalize $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error on_finalize);
|
||||
};
|
||||
(on_finalise $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error_renamed on_finalise on_finalize);
|
||||
};
|
||||
(offchain_worker $( $rest:ident )*) => {
|
||||
$crate::__check_reserved_fn_name!(@compile_error offchain_worker);
|
||||
};
|
||||
@@ -1914,7 +2076,7 @@ macro_rules! __check_reserved_fn_name {
|
||||
#[allow(dead_code)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::sp_runtime::traits::{OnInitialize, OnFinalize};
|
||||
use crate::sp_runtime::traits::{OnInitialize, OnFinalize, OnRuntimeUpgrade};
|
||||
use crate::weights::{DispatchInfo, DispatchClass};
|
||||
use crate::traits::{CallMetadata, GetCallMetadata, GetCallName};
|
||||
|
||||
@@ -1936,8 +2098,8 @@ mod tests {
|
||||
}
|
||||
}
|
||||
|
||||
struct BLockWeight;
|
||||
impl<BlockNumber: Into<u32>> WeighData<BlockNumber> for BLockWeight {
|
||||
struct BlockWeight;
|
||||
impl<BlockNumber: Into<u32>> WeighData<BlockNumber> for BlockWeight {
|
||||
fn weigh_data(&self, target: BlockNumber) -> Weight {
|
||||
let target: u32 = target.into();
|
||||
if target % 2 == 0 { 10 } else { 0 }
|
||||
@@ -1957,8 +2119,10 @@ mod tests {
|
||||
|
||||
#[weight = SimpleDispatchInfo::FixedNormal(7)]
|
||||
fn on_initialize(n: T::BlockNumber,) { if n.into() == 42 { panic!("on_initialize") } }
|
||||
#[weight = BLockWeight]
|
||||
#[weight = BlockWeight]
|
||||
fn on_finalize(n: T::BlockNumber) { if n.into() == 42 { panic!("on_finalize") } }
|
||||
#[weight = SimpleDispatchInfo::FixedOperational(69)]
|
||||
fn on_runtime_upgrade() { }
|
||||
fn offchain_worker() {}
|
||||
|
||||
#[weight = SimpleDispatchInfo::FixedOperational(5)]
|
||||
@@ -2100,6 +2264,11 @@ mod tests {
|
||||
<Module<TraitImpl> as OnFinalize<u32>>::on_finalize(42);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn on_runtime_upgrade_should_work() {
|
||||
<Module<TraitImpl> as OnRuntimeUpgrade>::on_runtime_upgrade();
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn weight_should_attach_to_call_enum() {
|
||||
// operational.
|
||||
|
||||
@@ -65,8 +65,8 @@ pub trait StorageValue<T: FullCodec> {
|
||||
///
|
||||
/// # Usage
|
||||
///
|
||||
/// This would typically be called inside the module implementation of on_initialize, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_initialize`**. (More
|
||||
/// This would typically be called inside the module implementation of on_runtime_upgrade, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_runtime_upgrade`**. (More
|
||||
/// precisely prior initialized modules doesn't make use of this storage).
|
||||
fn translate<O: Decode, F: FnOnce(Option<O>) -> Option<T>>(f: F) -> Result<Option<T>, ()>;
|
||||
|
||||
@@ -265,8 +265,8 @@ pub trait StorageLinkedMap<K: FullCodec, V: FullCodec> {
|
||||
///
|
||||
/// # Usage
|
||||
///
|
||||
/// This would typically be called inside the module implementation of on_initialize, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_initialize`**. (More
|
||||
/// This would typically be called inside the module implementation of on_runtime_upgrade, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_runtime_upgrade`**. (More
|
||||
/// precisely prior initialized modules doesn't make use of this storage).
|
||||
fn translate<K2, V2, TK, TV>(translate_key: TK, translate_val: TV) -> Result<(), Option<K2>>
|
||||
where K2: FullCodec + Clone, V2: Decode, TK: Fn(K2) -> K, TV: Fn(V2) -> V;
|
||||
@@ -460,8 +460,8 @@ pub trait StoragePrefixedMap<Value: FullCodec> {
|
||||
///
|
||||
/// # Usage
|
||||
///
|
||||
/// This would typically be called inside the module implementation of on_initialize, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_initialize`**. (More
|
||||
/// This would typically be called inside the module implementation of on_runtime_upgrade, while
|
||||
/// ensuring **no usage of this storage are made before the call to `on_runtime_upgrade`**. (More
|
||||
/// precisely prior initialized modules doesn't make use of this storage).
|
||||
fn translate_values<OldValue, TV>(translate_val: TV) -> Result<(), u32>
|
||||
where OldValue: Decode, TV: Fn(OldValue) -> Value
|
||||
|
||||
@@ -101,6 +101,10 @@ pub fn get_raw(key: &[u8]) -> Option<Vec<u8>> {
|
||||
}
|
||||
|
||||
/// Put a raw byte slice into storage.
|
||||
///
|
||||
/// **WARNING**: If you set the storage of the Substrate Wasm (`well_known_keys::CODE`),
|
||||
/// you should also call `frame_system::RuntimeUpgraded::put(true)` to trigger the
|
||||
/// `on_runtime_upgrade` logic.
|
||||
pub fn put_raw(key: &[u8], value: &[u8]) {
|
||||
sp_io::storage::set(key, value)
|
||||
}
|
||||
|
||||
@@ -67,9 +67,11 @@ pub trait ClassifyDispatch<T> {
|
||||
fn classify_dispatch(&self, target: T) -> DispatchClass;
|
||||
}
|
||||
|
||||
/// Means of determining the weight of a block's life cycle hooks: `on_initialize`, `on_finalize` and
|
||||
/// such.
|
||||
/// Means of determining the weight of a block's life cycle hooks: `on_initialize`, `on_finalize`,
|
||||
/// `on_runtime_upgrade`, and such.
|
||||
pub trait WeighBlock<BlockNumber> {
|
||||
/// Return the weight of the block's on_runtime_upgrade hook.
|
||||
fn on_runtime_upgrade() -> Weight { Zero::zero() }
|
||||
/// Return the weight of the block's on_initialize hook.
|
||||
fn on_initialize(_: BlockNumber) -> Weight { Zero::zero() }
|
||||
/// Return the weight of the block's on_finalize hook.
|
||||
@@ -87,6 +89,14 @@ pub trait PaysFee<T> {
|
||||
/// Maybe I can do something to remove the duplicate code here.
|
||||
#[impl_for_tuples(30)]
|
||||
impl<BlockNumber: Copy> WeighBlock<BlockNumber> for SingleModule {
|
||||
fn on_runtime_upgrade() -> Weight {
|
||||
let mut accumulated_weight: Weight = Zero::zero();
|
||||
for_tuples!(
|
||||
#( accumulated_weight = accumulated_weight.saturating_add(SingleModule::on_runtime_upgrade()); )*
|
||||
);
|
||||
accumulated_weight
|
||||
}
|
||||
|
||||
fn on_initialize(n: BlockNumber) -> Weight {
|
||||
let mut accumulated_weight: Weight = Zero::zero();
|
||||
for_tuples!(
|
||||
|
||||
@@ -27,6 +27,6 @@ macro_rules! reserved {
|
||||
}
|
||||
}
|
||||
|
||||
reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
|
||||
fn main() {}
|
||||
|
||||
@@ -1,47 +1,39 @@
|
||||
error: Invalid call fn name: `on_finalize`, name is reserved and doesn't match expected signature, please refer to `decl_module!` documentation to see the appropriate usage, or rename it to an unreserved keyword.
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
30 | reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
error: Invalid call fn name: `on_initialize`, name is reserved and doesn't match expected signature, please refer to `decl_module!` documentation to see the appropriate usage, or rename it to an unreserved keyword.
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
30 | reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
error: `on_finalise` was renamed to `on_finalize`. Please rename your function accordingly.
|
||||
error: Invalid call fn name: `on_runtime_upgrade`, name is reserved and doesn't match expected signature, please refer to `decl_module!` documentation to see the appropriate usage, or rename it to an unreserved keyword.
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
error: `on_initialise` was renamed to `on_initialize`. Please rename your function accordingly.
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
30 | reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
error: Invalid call fn name: `offchain_worker`, name is reserved and doesn't match expected signature, please refer to `decl_module!` documentation to see the appropriate usage, or rename it to an unreserved keyword.
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
30 | reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
error: `deposit_event` function is reserved and must follow the syntax: `$vis:vis fn deposit_event() = default;`
|
||||
--> $DIR/on_initialize.rs:30:1
|
||||
|
|
||||
30 | reserved!(on_finalize on_initialize on_finalise on_initialise offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
30 | reserved!(on_finalize on_initialize on_runtime_upgrade offchain_worker deposit_event);
|
||||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ in this macro invocation
|
||||
|
|
||||
= note: this error originates in a macro outside of the current crate (in Nightly builds, run with -Z external-macro-backtrace for more info)
|
||||
|
||||
@@ -367,6 +367,9 @@ decl_storage! {
|
||||
/// the `EventIndex` then in case if the topic has the same contents on the next block
|
||||
/// no notification will be triggered thus the event might be lost.
|
||||
EventTopics get(fn event_topics): map hasher(blake2_256) T::Hash => Vec<(T::BlockNumber, EventIndex)>;
|
||||
|
||||
/// A bool to track if the runtime was upgraded last block.
|
||||
pub RuntimeUpgraded: bool;
|
||||
}
|
||||
add_extra_genesis {
|
||||
config(changes_trie_config): Option<ChangesTrieConfiguration>;
|
||||
@@ -486,6 +489,7 @@ decl_module! {
|
||||
}
|
||||
|
||||
storage::unhashed::put_raw(well_known_keys::CODE, &code);
|
||||
RuntimeUpgraded::put(true);
|
||||
Self::deposit_event(RawEvent::CodeUpdated);
|
||||
}
|
||||
|
||||
@@ -494,6 +498,7 @@ decl_module! {
|
||||
pub fn set_code_without_checks(origin, code: Vec<u8>) {
|
||||
ensure_root(origin)?;
|
||||
storage::unhashed::put_raw(well_known_keys::CODE, &code);
|
||||
RuntimeUpgraded::put(true);
|
||||
Self::deposit_event(RawEvent::CodeUpdated);
|
||||
}
|
||||
|
||||
@@ -521,6 +526,9 @@ decl_module! {
|
||||
ensure_root(origin)?;
|
||||
for i in &items {
|
||||
storage::unhashed::put_raw(&i.0, &i.1);
|
||||
if i.0 == well_known_keys::CODE {
|
||||
RuntimeUpgraded::put(true);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1971,6 +1979,26 @@ mod tests {
|
||||
System::events(),
|
||||
vec![EventRecord { phase: Phase::ApplyExtrinsic(0), event: 102u16, topics: vec![] }],
|
||||
);
|
||||
|
||||
assert_eq!(RuntimeUpgraded::get(), true);
|
||||
});
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn runtime_upgraded_with_set_storage() {
|
||||
let executor = substrate_test_runtime_client::new_native_executor();
|
||||
let mut ext = new_test_ext();
|
||||
ext.register_extension(sp_core::traits::CallInWasmExt::new(executor));
|
||||
ext.execute_with(|| {
|
||||
System::set_storage(
|
||||
RawOrigin::Root.into(),
|
||||
vec![(
|
||||
well_known_keys::CODE.to_vec(),
|
||||
substrate_test_runtime_client::runtime::WASM_BINARY.to_vec()
|
||||
)],
|
||||
).unwrap();
|
||||
|
||||
assert_eq!(RuntimeUpgraded::get(), true);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
@@ -345,7 +345,7 @@ mod tests {
|
||||
use sp_runtime::{
|
||||
Perbill,
|
||||
testing::Header,
|
||||
traits::{BlakeTwo256, IdentityLookup, Identity, OnInitialize},
|
||||
traits::{BlakeTwo256, IdentityLookup, Identity, OnRuntimeUpgrade},
|
||||
};
|
||||
use sp_storage::Storage;
|
||||
|
||||
@@ -484,7 +484,7 @@ mod tests {
|
||||
];
|
||||
s.top = data.into_iter().collect();
|
||||
sp_io::TestExternalities::new(s).execute_with(|| {
|
||||
Balances::on_initialize(1);
|
||||
Balances::on_runtime_upgrade();
|
||||
assert_eq!(Balances::free_balance(6), 60);
|
||||
assert_eq!(Balances::usable_balance(&6), 30);
|
||||
System::set_block_number(2);
|
||||
|
||||
@@ -23,11 +23,11 @@ use sp_runtime::traits::{Block as BlockT, NumberFor};
|
||||
/// specific chain build.
|
||||
///
|
||||
/// The Strategy can be customized for the two use cases of authoring new blocks
|
||||
/// upon the best chain or which fork to finalise. Unless implemented differently
|
||||
/// upon the best chain or which fork to finalize. Unless implemented differently
|
||||
/// by default finalization methods fall back to use authoring, so as a minimum
|
||||
/// `_authoring`-functions must be implemented.
|
||||
///
|
||||
/// Any particular user must make explicit, however, whether they intend to finalise
|
||||
/// Any particular user must make explicit, however, whether they intend to finalize
|
||||
/// or author through the using the right function call, as these might differ in
|
||||
/// some implementations.
|
||||
///
|
||||
|
||||
@@ -241,7 +241,7 @@ impl<'a, I: AsRef<[u8]>, T: IntoIterator<Item=I>> Request<'a, T> {
|
||||
sp_io::offchain::http_request_write_body(id, chunk.as_ref(), self.deadline)?;
|
||||
}
|
||||
|
||||
// finalise the request
|
||||
// finalize the request
|
||||
sp_io::offchain::http_request_write_body(id, &[], self.deadline)?;
|
||||
|
||||
Ok(PendingRequest {
|
||||
|
||||
@@ -345,6 +345,14 @@ pub trait OnInitialize<BlockNumber> {
|
||||
fn on_initialize(_n: BlockNumber) {}
|
||||
}
|
||||
|
||||
/// The runtime upgrade trait. Implementing this lets you express what should happen
|
||||
/// when the runtime upgrades, and changes may need to occur to your module.
|
||||
#[impl_for_tuples(30)]
|
||||
pub trait OnRuntimeUpgrade {
|
||||
/// Perform a module upgrade.
|
||||
fn on_runtime_upgrade() {}
|
||||
}
|
||||
|
||||
/// Off-chain computation trait.
|
||||
///
|
||||
/// Implementing this trait on a module allows you to perform long-running tasks
|
||||
|
||||
Reference in New Issue
Block a user