fix: reward_calculation

gnoswap-labs · Jan 9, 2025 · 30fb984 · 30fb984
1 parent 09bfc7c
commit 30fb984
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/gov/staker/reward_calculation.gno b/gov/staker/reward_calculation.gno
@@ -349,9 +349,11 @@ func calculateProtocolFee() {
 		return
 	}
 
-	registered := common.ListRegisteredTokens()
-	// get gov staker's grc20 balance
 	for _, tokenPath := range common.ListRegisteredTokens() {
+		tokenBalance, exists := accuProtocolFee.Get(tokenPath)
+		if !exists || tokenBalance == uint64(0) {
+			continue
+		}
 
 		leftValue, exists := leftProtocolFeeFromLast.Get(tokenPath)
 		leftFromLast := uint64(0)
@@ -399,4 +401,4 @@ func calculateProtocolFee() {
 		alreadyCalculatedProtocolFee.Set(tokenPath, current+tokenBalance.(uint64))
 		leftProtocolFeeFromLast.Set(tokenPath, tokenBalance.(uint64)-calculated)
 	}
-}
+}