program: Always distribute full amount (#6)

paladin-bladesmith · Jul 22, 2024 · 53a89fc · OliverNChalk · Jul 26, 2024 · buffalojoec
1 parent 4f32315
commit 53a89fc
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/program/src/processor.rs b/program/src/processor.rs
@@ -26,16 +26,18 @@ struct RewardDistribution {
 // * 40% to all token stakers.
 // * 50% to all token holders.
 fn calculate_distribution(amount: u64) -> Result<RewardDistribution, ProgramError> {
-    let treasury_reward = amount
-        .checked_div(10)
-        .ok_or(ProgramError::InvalidInstructionData)?;
     let stakers_reward = amount
         .checked_mul(2)
         .and_then(|v| v.checked_div(5))
         .ok_or(ProgramError::InvalidInstructionData)?;
     let holders_reward = amount
         .checked_div(2)
         .ok_or(ProgramError::InvalidInstructionData)?;
+    let treasury_reward = amount
+        .checked_sub(holders_reward)
+        .unwrap()
+        .checked_sub(stakers_reward)
+        .unwrap();
 
     Ok(RewardDistribution {
         treasury_reward,