fix claim rewards calculate

AcalaNetwork · Sep 14, 2023 · 581cd15 · 581cd15
1 parent b6e7505
commit 581cd15
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/packages/earn-subql/src/handlers/claim-rewards.ts b/packages/earn-subql/src/handlers/claim-rewards.ts
@@ -25,20 +25,13 @@ export const handleClaimRewards = async (event: SubstrateEvent) => {
   // modify total_share, actural_amount, deduction_amount, user_share to bigint
   const totalShareBigInt = BigInt(totalShare);
   const userShareBigInt = BigInt(userShare);
-  const actualAmountBigInt = BigInt(actual_amount.toString());
-  const deductionAmountBigInt = BigInt(deduction_amount.toString());
-  // get total reward amount in current event
-  const totalRewardAmount = actualAmountBigInt * deductionAmountBigInt;
-  // get user withdraw reward amount form loyalty bonus pool
-  const userWithdrawRewardAmount = totalRewardAmount * userShareBigInt / totalShareBigInt;
 
   let poolEntity = await LoyaltyBonusPool.get(poolId);
 
   if (!poolEntity) {
     poolEntity = new LoyaltyBonusPool(poolId, [] as any, timestamp, BigInt(blockNumber));
   }
 
-  // update pool rewards
   // get reward token id
   const rewardTokenId = forceToCurrencyName(reward_currency_id);
   const deductionAmount = BigInt(deduction_amount.toString()); 
@@ -48,7 +41,11 @@ export const handleClaimRewards = async (event: SubstrateEvent) => {
     // get current reward field
     const currentRewardIndex = poolEntity.rewards.findIndex((reward) => reward.token === rewardTokenId);
     const prevRewardAmount = currentRewardIndex === -1 ? BigInt(0) : poolEntity.rewards[currentRewardIndex].amount;
-    const currentRewardAmount = prevRewardAmount - userWithdrawRewardAmount + deductionAmount;
+    // when prevRewardAmount is zero, currentRewardAmount is equal to deductionAmount, otherwise, calculate currentRewardAmount
+    const currentRewardAmount = prevRewardAmount === BigInt(0)
+      ? deductionAmount
+      // user will withdraw some rewards and then put new deductionAmount into pool, so we need to deduct the userShare from totalShare
+      : prevRewardAmount - prevRewardAmount * (userShareBigInt / totalShareBigInt) + deductionAmount;
 
     // if currentRewardIndex is -1, push new reward
     if (currentRewardIndex === -1) {