git.delta.rocks / unique-network / refs/commits / 3539f20b1874

difftreelog

feat switch `collator-selection` from `Currency` trait to `fungible::*` traits

Grigoriy Simonov2023-05-22parent: #fcb9d1c.patch.diff
in: master

4 files changed

modifiedpallets/collator-selection/src/benchmarking.rsdiffbeforeafterboth
40use frame_support::{40use frame_support::{
41 assert_ok,41 assert_ok,
42 codec::Decode,42 codec::Decode,
43 traits::{Currency, EnsureOrigin, Get},43 traits::{
44 EnsureOrigin,
45 fungible::{Inspect, Mutate},
46 Get,
47 },
44};48};
45use frame_system::{EventRecord, RawOrigin};49use frame_system::{EventRecord, RawOrigin};
78) -> T::AccountId {82) -> T::AccountId {
79 let user = account(string, n, SEED);83 let user = account(string, n, SEED);
80 let balance = balance_unit::<T>() * balance_factor.into();84 let balance = balance_unit::<T>() * balance_factor.into();
81 let _ = T::Currency::make_free_balance_be(&user, balance);85 let _ = T::Currency::set_balance(&user, balance);
82 user86 user
83}87}
8488
137 );141 );
138142
139 for who in candidates {143 for who in candidates {
140 T::Currency::make_free_balance_be(&who, <LicenseBond<T>>::get() * 2u32.into());144 T::Currency::set_balance(&who, <LicenseBond<T>>::get() * 2u32.into());
141 <CollatorSelection<T>>::get_license(RawOrigin::Signed(who.clone()).into()).unwrap();145 <CollatorSelection<T>>::get_license(RawOrigin::Signed(who.clone()).into()).unwrap();
142 <CollatorSelection<T>>::onboard(RawOrigin::Signed(who).into()).unwrap();146 <CollatorSelection<T>>::onboard(RawOrigin::Signed(who).into()).unwrap();
143 }147 }
153 );157 );
154158
155 for who in candidates {159 for who in candidates {
156 T::Currency::make_free_balance_be(&who, <LicenseBond<T>>::get() * 2u32.into());160 T::Currency::set_balance(&who, <LicenseBond<T>>::get() * 2u32.into());
157 <CollatorSelection<T>>::get_license(RawOrigin::Signed(who.clone()).into()).unwrap();161 <CollatorSelection<T>>::get_license(RawOrigin::Signed(who.clone()).into()).unwrap();
158 }162 }
159}163}
160164
161/// `Currency::minimum_balance` was used originally, but in unique-chain, we have165/// `Currency::minimum_balance` was used originally, but in unique-chain, we have
162/// zero existential deposit, thus triggering zero bond assertion.166/// zero existential deposit, thus triggering zero bond assertion.
163fn balance_unit<T: Config>() -> <T::Currency as Currency<T::AccountId>>::Balance {167fn balance_unit<T: Config>() -> BalanceOf<T> {
164 200u32.into()168 200u32.into()
165}169}
166170
182188
183 let new_invulnerable: T::AccountId = whitelisted_caller();189 let new_invulnerable: T::AccountId = whitelisted_caller();
184 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();190 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();
185 T::Currency::make_free_balance_be(&new_invulnerable, bond.clone());191 T::Currency::set_balance(&new_invulnerable, bond.clone());
186192
187 <session::Pallet<T>>::set_keys(193 <session::Pallet<T>>::set_keys(
188 RawOrigin::Signed(new_invulnerable.clone()).into(),194 RawOrigin::Signed(new_invulnerable.clone()).into(),
227233
228 let caller: T::AccountId = whitelisted_caller();234 let caller: T::AccountId = whitelisted_caller();
229 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();235 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();
230 T::Currency::make_free_balance_be(&caller, bond.clone());236 T::Currency::set_balance(&caller, bond.clone());
231237
232 <session::Pallet<T>>::set_keys(238 <session::Pallet<T>>::set_keys(
233 RawOrigin::Signed(caller.clone()).into(),239 RawOrigin::Signed(caller.clone()).into(),
253259
254 let caller: T::AccountId = whitelisted_caller();260 let caller: T::AccountId = whitelisted_caller();
255 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();261 let bond: BalanceOf<T> = balance_unit::<T>() * 2u32.into();
256 T::Currency::make_free_balance_be(&caller, bond.clone());262 T::Currency::set_balance(&caller, bond.clone());
257263
258 let origin = RawOrigin::Signed(caller.clone());264 let origin = RawOrigin::Signed(caller.clone());
259265
329 // worst case is paying a non-existing candidate account.335 // worst case is paying a non-existing candidate account.
330 note_author {336 note_author {
331 <LicenseBond<T>>::put(balance_unit::<T>());337 <LicenseBond<T>>::put(balance_unit::<T>());
332 T::Currency::make_free_balance_be(338 T::Currency::set_balance(
333 &<CollatorSelection<T>>::account_id(),339 &<CollatorSelection<T>>::account_id(),
334 balance_unit::<T>() * 4u32.into(),340 balance_unit::<T>() * 4u32.into(),
335 );341 );
336 let author = account("author", 0, SEED);342 let author = account("author", 0, SEED);
337 let new_block: T::BlockNumber = 10u32.into();343 let new_block: T::BlockNumber = 10u32.into();
338344
339 frame_system::Pallet::<T>::set_block_number(new_block);345 frame_system::Pallet::<T>::set_block_number(new_block);
340 assert!(T::Currency::free_balance(&author) == 0u32.into());346 assert!(T::Currency::balance(&author) == 0u32.into());
341 }: {347 }: {
342 <CollatorSelection<T> as EventHandler<_, _>>::note_author(author.clone())348 <CollatorSelection<T> as EventHandler<_, _>>::note_author(author.clone())
343 } verify {349 } verify {
344 assert!(T::Currency::free_balance(&author) > 0u32.into());350 assert!(T::Currency::balance(&author) > 0u32.into());
345 assert_eq!(frame_system::Pallet::<T>::block_number(), new_block);351 assert_eq!(frame_system::Pallet::<T>::block_number(), new_block);
346 }352 }
347353
modifiedpallets/collator-selection/src/lib.rsdiffbeforeafterboth
9292
93#[frame_support::pallet]93#[frame_support::pallet]
94pub mod pallet {94pub mod pallet {
95 use super::*;
95 pub use crate::weights::WeightInfo;96 pub use crate::weights::WeightInfo;
96 use core::ops::Div;97 use core::ops::Div;
97 use frame_support::{98 use frame_support::{
100 pallet_prelude::*,101 pallet_prelude::*,
101 sp_runtime::traits::{AccountIdConversion, CheckedSub, Saturating, Zero},102 sp_runtime::traits::{AccountIdConversion, CheckedSub, Saturating, Zero},
102 traits::{103 traits::{
103 Currency, EnsureOrigin, ExistenceRequirement::KeepAlive, ReservableCurrency,104 EnsureOrigin,
105 fungible::{Balanced, BalancedHold, Inspect, InspectHold, Mutate, MutateHold},
104 ValidatorRegistration,106 ValidatorRegistration,
107 tokens::{Precision, Preservation},
105 },108 },
106 BoundedVec, PalletId,109 BoundedVec, PalletId,
107 };110 };
159 /// The weight information of this pallet.162 /// The weight information of this pallet.
160 type WeightInfo: WeightInfo;163 type WeightInfo: WeightInfo;
164
165 #[pallet::constant]
166 type LicenceBondIdentifier: Get<<<Self as pallet_configuration::Config>::Currency as InspectHold<Self::AccountId>>::Reason>;
161 }167 }
162168
163 #[pallet::pallet]169 #[pallet::pallet]
361367
362 let deposit = <LicenseBond<T>>::get();368 let deposit = <LicenseBond<T>>::get();
363369
364 T::Currency::reserve(&who, deposit)?;370 T::Currency::hold(&T::LicenceBondIdentifier::get(), &who, deposit)?;
365 LicenseDepositOf::<T>::insert(who.clone(), deposit);371 LicenseDepositOf::<T>::insert(who.clone(), deposit);
366372
367 Self::deposit_event(Event::LicenseObtained {373 Self::deposit_event(Event::LicenseObtained {
523 let slashed = T::SlashRatio::get() * deposit;529 let slashed = T::SlashRatio::get() * deposit;
524 let remaining = deposit - slashed;530 let remaining = deposit - slashed;
525531
526 let (imbalance, _) = T::Currency::slash_reserved(who, slashed);532 let (imbalance, _) =
533 T::Currency::slash(&T::LicenceBondIdentifier::get(), who, slashed);
527 //T::Currency::unreserve(who, remaining);534 //T::Currency::unreserve(who, remaining);
528 deposit_returned = remaining;535 deposit_returned = remaining;
529536
530 T::Currency::resolve_creating(&T::TreasuryAccountId::get(), imbalance);537 T::Currency::resolve(&T::TreasuryAccountId::get(), imbalance)
538 .map_err(|_| DispatchError::Other("Failed to deposit imbalance"))?;
531 } else {539 } else {
532 //T::Currency::unreserve(who, deposit);540 //T::Currency::unreserve(who, deposit);
533 deposit_returned = deposit;541 deposit_returned = deposit;
534 }542 }
535543
536 T::Currency::unreserve(who, deposit_returned);544 T::Currency::release(
545 &T::LicenceBondIdentifier::get(),
546 who,
547 deposit_returned,
548 Precision::Exact,
549 )?;
537 Ok(())550 Ok(())
538 } else {551 } else {
539 Err(Error::<T>::NoLicense.into())552 Err(Error::<T>::NoLicense.into())
594 fn note_author(author: T::AccountId) {607 fn note_author(author: T::AccountId) {
595 let pot = Self::account_id();608 let pot = Self::account_id();
596 // assumes an ED will be sent to pot.609 // assumes an ED will be sent to pot.
597 let reward = T::Currency::free_balance(&pot)610 let reward = T::Currency::balance(&pot)
598 .checked_sub(&T::Currency::minimum_balance())611 .checked_sub(&T::Currency::minimum_balance())
599 .unwrap_or_else(Zero::zero)612 .unwrap_or_else(Zero::zero)
600 .div(2u32.into());613 .div(2u32.into());
601 // `reward` is half of pot account minus ED, this should never fail.614 // `reward` is half of pot account minus ED, this should never fail.
602 let _success = T::Currency::transfer(&pot, &author, reward, KeepAlive);615 let _success = T::Currency::transfer(&pot, &author, reward, Preservation::Preserve);
603 debug_assert!(_success.is_ok());616 debug_assert!(_success.is_ok());
604 <LastAuthoredBlock<T>>::insert(author, frame_system::Pallet::<T>::block_number());617 <LastAuthoredBlock<T>>::insert(author, frame_system::Pallet::<T>::block_number());
605618
modifiedpallets/collator-selection/src/tests.rsdiffbeforeafterboth
417fn authorship_event_handler() {417fn authorship_event_handler() {
418 new_test_ext().execute_with(|| {418 new_test_ext().execute_with(|| {
419 // put 100 in the pot + 5 for ED419 // put 100 in the pot + 5 for ED
420 Balances::make_free_balance_be(&CollatorSelection::account_id(), 105);420 <pallet_balances::Pallet<T> as fungible::Mutate<T::AccountId>>::set_balance(
421 &CollatorSelection::account_id(),
422 105,
423 );
441 // Nothing panics, no reward when no ED in balance444 // Nothing panics, no reward when no ED in balance
442 Authorship::on_initialize(1);445 Authorship::on_initialize(1);
443 // put some money into the pot at ED446 // put some money into the pot at ED
444 Balances::make_free_balance_be(&CollatorSelection::account_id(), 5);447 <pallet_balances::Pallet<T> as fungible::Mutate<T::AccountId>>::set_balance(
448 &CollatorSelection::account_id(),
449 5,
450 );
modifiedpallets/configuration/src/lib.rsdiffbeforeafterboth
5050
51 pub use crate::weights::WeightInfo;51 pub use crate::weights::WeightInfo;
52 pub type BalanceOf<T> = 52 pub type BalanceOf<T> =
53 <<T as Config>::Balances as fungible::Inspect<<T as SystemConfig>::AccountId>>::Balance;53 <<T as Config>::Currency as fungible::Inspect<<T as SystemConfig>::AccountId>>::Balance;
5454
55 #[pallet::config]55 #[pallet::config]
56 pub trait Config: frame_system::Config {56 pub trait Config: frame_system::Config {
57 /// Overarching event type.57 /// Overarching event type.
58 type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;58 type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;
5959
60 type Balances:60 type Currency: fungible::Inspect<Self::AccountId>
61 fungible::Inspect<Self::AccountId>
62 + fungible::Mutate::<Self::AccountId>61 + fungible::Mutate<Self::AccountId>
63 + fungible::MutateFreeze::<Self::AccountId>62 + fungible::MutateFreeze<Self::AccountId>
64 + fungible::InspectHold::<Self::AccountId>63 + fungible::InspectHold<Self::AccountId>
65 + fungible::MutateHold::<Self::AccountId>64 + fungible::MutateHold<Self::AccountId>
66 + fungible::BalancedHold::<Self::AccountId>;65 + fungible::BalancedHold<Self::AccountId>;
6766
68 #[pallet::constant]67 #[pallet::constant]
69 type DefaultWeightToFeeCoefficient: Get<u64>;68 type DefaultWeightToFeeCoefficient: Get<u64>;