fix total rewards computation

ambrosus · Nov 17, 2024 · c0f2f8b · c0f2f8b
1 parent a27b24f
commit c0f2f8b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/gov-portal-db/src/rewards_manager/rewards_cache.rs b/gov-portal-db/src/rewards_manager/rewards_cache.rs
@@ -59,19 +59,19 @@ impl RewardsCache {
         let mut unclaimed_rewards = HashMap::<BatchId, HashMap<Address, Reward>>::default();
 
         for entry in rewards {
-            match entry.status {
-                RewardStatus::Granted => (),
-                RewardStatus::Reverted => continue,
-                RewardStatus::Claimed => {
-                    return Err(anyhow::anyhow!("Unexpected status for a batch {entry:?}").into());
-                }
+            if entry.status == RewardStatus::Claimed {
+                return Err(anyhow::anyhow!("Unexpected status for a batch {entry:?}").into());
             }
 
             // [`BatchId`] is a block number in which it was created
             let block_number = entry.id.0;
 
             for (wallet, reward) in entry.wallets {
                 match reward.status {
+                    // If batch was reverted
+                    RewardStatus::Granted if entry.status != RewardStatus::Granted => {
+                        continue;
+                    }
                     RewardStatus::Granted => {
                         unclaimed_rewards
                             .entry(entry.id)