custom amount to withdraw

zeta-chain · Aug 21, 2024 · f52a4bd · f52a4bd
1 parent 172158c
commit f52a4bd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/zevm-app-contracts/contracts/instant-rewards/InstantRewards.sol b/packages/zevm-app-contracts/contracts/instant-rewards/InstantRewards.sol
@@ -79,9 +79,10 @@ contract InstantRewards is Ownable, Pausable, ReentrancyGuard {
         signerAddress = signerAddress_;
     }
 
-    function withdraw(address wallet) external onlyOwner {
+    function withdraw(address wallet, uint256 amount) external onlyOwner {
         if (wallet == address(0)) revert InvalidAddress();
-        payable(wallet).transfer(address(this).balance);
+        if (amount > address(this).balance) revert TransferFailed();
+        payable(wallet).transfer(amount);
     }
 
     receive() external payable {}