add dodoBalanceOf and fix balanceOf

This commit is contained in:
mingda
2021-02-07 15:32:31 +08:00
committed by owen05
parent 5b91969a68
commit 60f95d8d72

View File

@@ -221,12 +221,8 @@ contract vDODOToken is InitializableOwnable {
vDODOSupply = IERC20(_DODO_TOKEN_).balanceOf(address(this)) / _DODO_RATIO_;
}
function balanceOf(address account) public view returns (uint256 dodoAmount) {
UserInfo memory user = userInfo[account];
dodoAmount = DecimalMath
.mulFloor(uint256(user.stakingPower), getLatestAlpha())
.mul(_DODO_RATIO_)
.sub(user.credit);
function balanceOf(address account) public view returns (uint256 vDODOAmount) {
vDODOAmount = dodoBalanceOf(account) / _DODO_RATIO_;
}
function availableBalanceOf(address account) public view returns (uint256 balance) {
@@ -238,6 +234,13 @@ contract vDODOToken is InitializableOwnable {
}
}
function dodoBalanceOf(address account) public view returns (uint256 dodoAmount) {
UserInfo memory user = userInfo[account];
dodoAmount = DecimalMath.mulFloor(uint256(user.stakingPower), getLatestAlpha()).sub(
user.credit
);
}
function transfer(address to, uint256 vDODOAmount) public returns (bool) {
_updateAlpha();
_transfer(msg.sender, to, vDODOAmount);