Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-01-03 09:29:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Ensure TotalAllocatedStake is set for the first set

Browse source
This commit is contained in:
Luke Parker 2024-07-04 02:11:04 -04:00
parent ba244e8090
commit 52bb918cc9
No known key found for this signature in database
1 changed files with 15 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
substrate/validator-sets/pallet/src/lib.rs
View file

@ -691,6 +691,15 @@ pub mod pallet {
} }
} }
fn set_total_allocated_stake(network: NetworkId) {
let participants = Participants::<T>::get(network)
.expect("setting TotalAllocatedStake for a network without participants");
let total_stake = participants.iter().fold(0, |acc, (addr, _)| {
acc + Allocations::<T>::get((network, addr)).unwrap_or(Amount(0)).0
});
TotalAllocatedStake::<T>::set(network, Some(Amount(total_stake)));
}
// TODO: This is called retire_set, yet just starts retiring the set // TODO: This is called retire_set, yet just starts retiring the set
// Update the nomenclature within this function // Update the nomenclature within this function
pub fn retire_set(set: ValidatorSet) { pub fn retire_set(set: ValidatorSet) {
@ -715,12 +724,7 @@ pub mod pallet {
}); });
// Update the total allocated stake to be for the current set // Update the total allocated stake to be for the current set
let participants = self.set_total_allocated_stake(set.network);
Participants::<T>::get(set.network).expect("set retired without a new set");
let total_stake = participants.iter().fold(0, |acc, (addr, _)| {
acc + Allocations::<T>::get((set.network, addr)).unwrap_or(Amount(0)).0
});
TotalAllocatedStake::<T>::set(set.network, Some(Amount(total_stake)));
} }
/// Take the amount deallocatable. /// Take the amount deallocatable.
@ -907,6 +911,11 @@ pub mod pallet {
Keys::<T>::set(set, Some(key_pair.clone())); Keys::<T>::set(set, Some(key_pair.clone()));
// If this is the first ever set for this network, set TotalAllocatedStake now
if session == Session(0) {
self.set_total_allocated_stake(network);
}
// This does not remove from TotalAllocatedStake or InSet in order to: // This does not remove from TotalAllocatedStake or InSet in order to:
// 1) Not decrease the stake present in this set. This means removed participants are // 1) Not decrease the stake present in this set. This means removed participants are
// still liable for the economic security of the external network. This prevents // still liable for the economic security of the external network. This prevents