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

edited small thing on COMP contract.

Browse Source
This commit is contained in:
Samyak Jain 2020-06-20 18:01:45 +10:00
parent 1031c34937
commit 47b096e7ef
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) {
return TokenInterface(0xc00e94Cb662C3520282E6f5717214004A7f26888);
function getCompTokenAddress() internal pure returns (address) {
return 0xc00e94Cb662C3520282E6f5717214004A7f26888;
}
/**
@ -301,9 +301,10 @@ contract ExtraResolver is BasicResolver {
* @param setId Set ctoken amount at this ID in `InstaMemory` Contract.
*/
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));
uint finalBal = getCompToken().balanceOf(address(this));
uint finalBal = compToken.balanceOf(address(this));
uint amt = sub(finalBal, intialBal);
setUint(setId, amt);