Instadapp/dsa-connectors-old
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors-old.git synced 2024-07-29 22:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:InstaDApp/dsa-connectors

Browse Source
This commit is contained in:
Thrilok Kumar 2020-06-21 00:34:03 +05:30
commit a30934566d
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
contracts/connectors/compound.sol
View File

@ -132,10 +132,10 @@ contract CompoundHelpers is Helpers {
} }
/** /**
* @dev Return COMP Token. * @dev Return COMP Token Address.
*/ */
function getCompToken() internal pure returns (TokenInterface) { function getCompTokenAddress() internal pure returns (address) {
return TokenInterface(0xc00e94Cb662C3520282E6f5717214004A7f26888); return 0xc00e94Cb662C3520282E6f5717214004A7f26888;
} }
/** /**
@ -301,9 +301,10 @@ contract ExtraResolver is BasicResolver {
* @param setId Set ctoken amount at this ID in `InstaMemory` Contract. * @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
*/ */
function ClaimComp(uint setId) external payable { function ClaimComp(uint setId) external payable {
uint intialBal = getCompToken().balanceOf(address(this)); TokenInterface compToken = TokenInterface(getCompTokenAddress());
uint intialBal = compToken.balanceOf(address(this));
ComptrollerInterface(getComptrollerAddress()).claimComp(address(this)); ComptrollerInterface(getComptrollerAddress()).claimComp(address(this));
uint finalBal = getCompToken().balanceOf(address(this)); uint finalBal = compToken.balanceOf(address(this));
uint amt = sub(finalBal, intialBal); uint amt = sub(finalBal, intialBal);
setUint(setId, amt); setUint(setId, amt);