Refactor claimAllETHGains to use helper func

liquity · May 8, 2024 · 1b2f4f7 · 1b2f4f7
1 parent 4d3f91f
commit 1b2f4f7
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/contracts/src/StabilityPool.sol b/contracts/src/StabilityPool.sol
@@ -376,12 +376,7 @@ contract StabilityPool is LiquityBase, Ownable, CheckContract, IStabilityPool {
             _updateDepositAndSnapshots(msg.sender, compoundedBoldDeposit);
         }
 
-        uint256 stashedETHGain = stashedETH[msg.sender];
-        uint256 ETHToSend = currentETHGain + stashedETHGain;
-
-        stashedETH[msg.sender] = 0;
-
-        emit ETHGainWithdrawn(msg.sender, ETHToSend, boldLoss);
+        uint256 ETHToSend = _getTotalETHGainAndZeroStash(msg.sender,  currentETHGain);
 
         _sendETHGainToDepositor(ETHToSend);
         assert(getDepositorETHGain(msg.sender) == 0);