minor changes

This commit is contained in:
Thrilok Kumar 2020-08-16 02:54:21 +05:30
parent 755298095f
commit 3dcb3bea93

View File

@ -26,7 +26,7 @@ contract CurveVestingHelpers is Stores, DSMath{
} }
contract CurveVestingProtocol is CurveVestingHelpers { contract CurveVestingProtocol is CurveVestingHelpers {
event LogClaim(address account, uint256 claimable, uint256 getId, uint256 setId); event LogClaim(address account, uint256 claimAmount, uint256 getId, uint256 setId);
/** /**
* @dev Claim Curve DAO Token. * @dev Claim Curve DAO Token.
@ -35,10 +35,9 @@ contract CurveVestingProtocol is CurveVestingHelpers {
*/ */
function claim(uint getId, uint setId) external{ function claim(uint getId, uint setId) external{
TokenInterface curveTokenContract = TokenInterface(getCurveTokenAddr()); TokenInterface curveTokenContract = TokenInterface(getCurveTokenAddr());
uint initialCurveBal = curveTokenContract.balanceOf(address(this)); uint initialCurveBal = curveTokenContract.balanceOf(address(this));
ICurve(getCurveVestingAddr()).claim(address(this)); ICurve(getCurveVestingAddr()).claim(address(this));
uint finalCurveBal = curveTokenContract.balanceOf(address(this)); uint finalCurveBal = curveTokenContract.balanceOf(address(this));
uint claimedAmt = sub(finalCurveBal, initialCurveBal); uint claimedAmt = sub(finalCurveBal, initialCurveBal);