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

Added change to the stabledebttoken

Browse Source
This commit is contained in:
The3D 2020-09-16 20:00:13 +02:00
parent 570a81a1b2
commit bcdef6fa7e
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
contracts/tokenization/StableDebtToken.sol
View File

@ -110,7 +110,8 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
uint256 balanceIncrease
) = _calculateBalanceIncrease(user);
vars.currentSupply = totalSupply();
//accrueing the interest accumulation to the stored total supply and caching it
vars.currentSupply = _totalSupply = totalSupply();
vars.currentAvgStableRate = _avgStableRate;
vars.nextSupply = vars.currentSupply.add(amount);
@ -125,12 +126,13 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
require(vars.newStableRate < (1 << 128), 'Debt token: stable rate overflow');
_usersData[user] = vars.newStableRate;
//updating the user and supply timestamp
//solium-disable-next-line
_totalSupplyTimestamp = _timestamps[user] = uint40(block.timestamp);
//calculates the updated average stable rate
_avgStableRate = vars.currentAvgStableRate
.rayMul(vars.currentPrincipalSupply.wadToRay())
.rayMul(vars.currentSupply.wadToRay())
.add(rate.rayMul(vars.amountInRay))
.rayDiv(vars.nextSupply.wadToRay());
@ -224,8 +226,9 @@ contract StableDebtToken is IStableDebtToken, DebtTokenBase {
);
}
function getPrincipalSupplyAndAvgRate() public override view returns (uint256, uint256) {
return (super.totalSupply(), _avgStableRate);
function getSupplyData() public override view returns (uint256, uint256, uint256) {
uint256 avgRate = _avgStableRate;
return (super.totalSupply(), _calcTotalSupply(avgRate), avgRate);
}
function getTotalSupplyAndAvgRate() public override view returns (uint256, uint256) {

4
contracts/tokenization/interfaces/IStableDebtToken.sol
View File

@ -86,9 +86,9 @@ interface IStableDebtToken {
function getUserLastUpdated(address user) external view returns (uint40);
/**
* @dev returns the principal total supply and the average stable rate
* @dev returns the principal, the total supply and the average stable rate
**/
function getPrincipalSupplyAndAvgRate() external view returns (uint256, uint256);
function getSupplyData() external view returns (uint256, uint256, uint256);
/**
* @dev returns the timestamp of the last update of the total supply