-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
f84a7e0
commit 8297940
Showing
11 changed files
with
231 additions
and
87 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,3 +1,4 @@ | ||
| pub mod v10; | ||
| pub mod v11; | ||
| pub mod v12; | ||
| pub mod v13; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
131 changes: 131 additions & 0 deletions
131
substrate-node/pallets/pallet-tfgrid/src/migrations/v13.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,131 @@ | ||
| use crate::{ | ||
| types::StorageVersion, Config, InterfaceOf, Nodes, PalletVersion, PubConfigOf, LocationOf, SerialNumberOf, | ||
| TFGRID_NODE_VERSION, | ||
| }; | ||
| use frame_support::{pallet_prelude::Weight, traits::Get, traits::OnRuntimeUpgrade}; | ||
| use log::info; | ||
| use sp_std::marker::PhantomData; | ||
| use tfchain_support::resources::Resources; | ||
| use tfchain_support::types::{ConsumableResources, Node, Power, PowerState, PowerTarget}; | ||
|
|
||
| pub mod deprecated { | ||
| use crate::Config; | ||
| use codec::{Decode, Encode}; | ||
| use frame_support::decl_module; | ||
|
|
||
| use scale_info::TypeInfo; | ||
| use sp_std::prelude::*; | ||
|
|
||
| use tfchain_support::{resources::Resources, types::NodeCertification}; | ||
|
|
||
| #[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Encode, Decode, Default, Debug, TypeInfo)] | ||
| pub struct NodeV11Struct<Location, PubConfig, If, SerialNumber> { | ||
| pub version: u32, | ||
| pub id: u32, | ||
| pub farm_id: u32, | ||
| pub twin_id: u32, | ||
| pub resources: Resources, | ||
| pub location: Location, | ||
| pub country: Vec<u8>, | ||
| pub city: Vec<u8>, | ||
| // optional public config | ||
| pub public_config: Option<PubConfig>, | ||
| pub created: u64, | ||
| pub farming_policy_id: u32, | ||
| pub interfaces: Vec<If>, | ||
| pub certification: NodeCertification, | ||
| pub secure_boot: bool, | ||
| pub virtualized: bool, | ||
| pub serial_number: SerialNumber, | ||
| pub connection_price: u32, | ||
| } | ||
|
|
||
| decl_module! { | ||
| pub struct Module<T: Config> for enum Call where origin: T::Origin { } | ||
| } | ||
| } | ||
| pub struct NodeMigrationV11<T: Config>(PhantomData<T>); | ||
|
|
||
| impl<T: Config> OnRuntimeUpgrade for NodeMigrationV11<T> { | ||
| #[cfg(feature = "try-runtime")] | ||
| fn pre_upgrade() -> Result<(), &'static str> { | ||
| info!("current pallet version: {:?}", PalletVersion::<T>::get()); | ||
| info!("👥 Tfgrid pallet to V11 passes PRE migrate checks ✅"); | ||
| Ok(()) | ||
| } | ||
|
|
||
| fn on_runtime_upgrade() -> Weight { | ||
| if PalletVersion::<T>::get() == StorageVersion::V11Struct { | ||
| migrate_to_version_12::<T>() | ||
| } else { | ||
| info!(" >>> Unused migration"); | ||
| 0 | ||
| } | ||
| } | ||
|
|
||
| #[cfg(feature = "try-runtime")] | ||
| fn post_upgrade() -> Result<(), &'static str> { | ||
| info!("current pallet version: {:?}", PalletVersion::<T>::get()); | ||
| info!( | ||
| "👥 Tfgrid pallet to {:?} passes POST migrate checks ✅", | ||
| PalletVersion::<T>::get() | ||
| ); | ||
|
|
||
| Ok(()) | ||
| } | ||
| } | ||
|
|
||
| pub fn migrate_to_version_12<T: Config>() -> frame_support::weights::Weight { | ||
| info!( | ||
| " >>> Starting tfgrid pallet migration, pallet version: {:?}", | ||
| PalletVersion::<T>::get() | ||
| ); | ||
|
|
||
| let mut migrated_count = 0; | ||
|
|
||
| Nodes::<T>::translate::<deprecated::NodeV11Struct<LocationOf<T>, PubConfigOf<T>, InterfaceOf<T>, SerialNumberOf<T> >, _>( | ||
| |k, n| { | ||
| migrated_count += 1; | ||
| let migrated_contract = Node { | ||
| version: TFGRID_NODE_VERSION, | ||
| id: n.id, | ||
| farm_id: n.farm_id, | ||
| twin_id: n.twin_id, | ||
| resources: ConsumableResources { | ||
| total_resources: n.resources, | ||
| used_resources: Resources::empty(), | ||
| }, | ||
| location: n.location, | ||
| power: Power { | ||
| target: PowerTarget::Up, | ||
| state: PowerState::Up, | ||
| last_uptime: 0, | ||
| }, | ||
| // optional public config | ||
| public_config: n.public_config, | ||
| created: n.created, | ||
| farming_policy_id: n.farming_policy_id, | ||
| interfaces: n.interfaces, | ||
| certification: n.certification, | ||
| secure_boot: n.secure_boot, | ||
| virtualized: n.virtualized, | ||
| serial_number: n.serial_number, | ||
| connection_price: n.connection_price, | ||
| }; | ||
| info!("Node: {:?} succesfully migrated", k); | ||
| Some(migrated_contract) | ||
| }, | ||
| ); | ||
|
|
||
| info!( | ||
| " <<< Node storage updated! Migrated {} Nodes ✅", | ||
| migrated_count | ||
| ); | ||
|
|
||
| // Update pallet storage version | ||
| PalletVersion::<T>::set(StorageVersion::V12Struct); | ||
| info!(" <<< Storage version upgraded"); | ||
|
|
||
| // Return the weight consumed by the migration. | ||
| T::DbWeight::get().reads_writes(migrated_count as Weight + 1, migrated_count as Weight + 1) | ||
| } |
Oops, something went wrong.