add debt tokens in reward calculation

This commit is contained in:
Mubaris NK 2021-04-19 15:04:14 +05:30
parent 42045b293c
commit cf0d5f2eba
No known key found for this signature in database
GPG Key ID: 9AC09AD0F8D68561

View File

@ -270,10 +270,11 @@ contract AaveHelpers is DSMath {
} }
function getPendingRewards(address[] memory _tokens, address user) internal view returns (uint rewards) { function getPendingRewards(address[] memory _tokens, address user) internal view returns (uint rewards) {
address[] memory _atokens = new address[](_tokens.length); uint arrLength = 2 * _tokens.length;
address[] memory _atokens = new address[](arrLength);
AaveProtocolDataProvider aaveData = AaveProtocolDataProvider(getAaveProtocolDataProvider()); AaveProtocolDataProvider aaveData = AaveProtocolDataProvider(getAaveProtocolDataProvider());
for (uint i = 0; i < _tokens.length; i++) { for (uint i = 0; i < _tokens.length; i++) {
(_atokens[i],,) = aaveData.getReserveTokensAddresses(_tokens[i]); (_atokens[2*i],,_atokens[2*i + 1]) = aaveData.getReserveTokensAddresses(_tokens[i]);
} }
rewards = AaveIncentivesInterface(getAaveIncentivesAddress()).getRewardsBalance(_atokens, user); rewards = AaveIncentivesInterface(getAaveIncentivesAddress()).getRewardsBalance(_atokens, user);
} }