d-bis/dodo-contractV2
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update withleftOver

Browse Source
This commit is contained in:
owen05
2021-04-10 16:05:21 +08:00
parent 0e59f643b8
commit b3d5d1090c
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
contracts/DODOToken/DODOMineV2/BaseMine.sol
View File

@@ -183,11 +183,11 @@ contract BaseMine is InitializableOwnable {
emit UpdateReward(i, newRewardPerBlock);
}
function withdrawLeftOver(uint256 i) external onlyOwner {
function withdrawLeftOver(uint256 i, uint256 amount) external onlyOwner {
RewardTokenInfo storage rt = rewardTokenInfos[i];
require(block.number > rt.endBlock, "DODOMineV2: MINING_NOT_FINISHED");
IRewardVault(rt.rewardVault).withdrawLeftOver(msg.sender);
IRewardVault(rt.rewardVault).withdrawLeftOver(msg.sender,amount);
emit WithdrawLeftOver(msg.sender, i);
}

7
contracts/DODOToken/DODOMineV2/RewardVault.sol
View File

@@ -14,7 +14,7 @@ import {IERC20} from "../../intf/IERC20.sol";
interface IRewardVault {
function reward(address to, uint256 amount) external;
function withdrawLeftOver(address to) external;
function withdrawLeftOver(address to, uint256 amount) external;
}
contract RewardVault is Ownable {
@@ -30,8 +30,9 @@ contract RewardVault is Ownable {
IERC20(rewardToken).safeTransfer(to, amount);
}
function withdrawLeftOver(address to) external onlyOwner {
function withdrawLeftOver(address to,uint256 amount) external onlyOwner {
uint256 leftover = IERC20(rewardToken).balanceOf(address(this));
IERC20(rewardToken).safeTransfer(to, leftover);
require(amount <= leftover, "VAULT_NOT_ENOUGH");
IERC20(rewardToken).safeTransfer(to, amount);
}
}