Instadapp/aave-protocol-v2
1
0
Fork 0
mirror of https://github.com/Instadapp/aave-protocol-v2.git synced 2024-07-29 21:47:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated stabledebttoken

Browse Source
This commit is contained in:
The3D 2020-09-15 18:49:53 +02:00
parent c6a3588792
commit 04a67d3df0
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
contracts/tokenization/StableDebtToken.sol
View File

@ -81,10 +81,11 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
} }
struct MintLocalVars { struct MintLocalVars {
uint256 supplyAfterMint; uint256 currentPrincipalSupply;
uint256 supplyBeforeMint; uint256 nextSupply;
uint256 amountInRay; uint256 amountInRay;
uint256 newStableRate; uint256 newStableRate;
uint256 currentAvgStableRate;
} }
/** /**
@ -108,8 +109,9 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
uint256 balanceIncrease uint256 balanceIncrease
) = _calculateBalanceIncrease(user); ) = _calculateBalanceIncrease(user);
vars.supplyBeforeMint = totalSupply().add(balanceIncrease); vars.currentPrincipalSupply = totalSupply();
vars.supplyAfterMint = vars.supplyBeforeMint.add(amount); vars.currentAvgStableRate = _avgStableRate;
vars.nextSupply = _totalSupply = _calcTotalSupply(vars.currentAvgStableRate).add(amount);
vars.amountInRay = amount.wadToRay(); vars.amountInRay = amount.wadToRay();
@ -126,10 +128,10 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
_totalSupplyTimestamp = _timestamps[user] = uint40(block.timestamp); _totalSupplyTimestamp = _timestamps[user] = uint40(block.timestamp);
//calculates the updated average stable rate //calculates the updated average stable rate
_avgStableRate = _avgStableRate _avgStableRate = vars.currentAvgStableRate
.rayMul(vars.supplyBeforeMint.wadToRay()) .rayMul(vars.currentPrincipalSupply.wadToRay())
.add(rate.rayMul(vars.amountInRay)) .add(rate.rayMul(vars.amountInRay))
.rayDiv(vars.supplyAfterMint.wadToRay()); .rayDiv(vars.nextSupply.wadToRay());
_mint(user, amount.add(balanceIncrease)); _mint(user, amount.add(balanceIncrease));
@ -237,6 +239,7 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
function getTotalSupplyLastUpdated() public override view returns(uint40) { function getTotalSupplyLastUpdated() public override view returns(uint40) {
return _totalSupplyTimestamp; return _totalSupplyTimestamp;
} }
/** /**
* @dev Returns the principal debt balance of the user from * @dev Returns the principal debt balance of the user from
* @return The debt balance of the user since the last burn/mint action * @return The debt balance of the user since the last burn/mint action