Skip to content

Commit f8f830c

Browse files
committed
feat: also add amm
1 parent e24c5a7 commit f8f830c

File tree

1 file changed

+5
-1
lines changed
  • src/1amm/AToken/AToken/contracts/protocol/tokenization

1 file changed

+5
-1
lines changed

src/1amm/AToken/AToken/contracts/protocol/tokenization/AToken.sol

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ contract AToken is
3131
keccak256('Permit(address owner,address spender,uint256 value,uint256 nonce,uint256 deadline)');
3232

3333
uint256 public constant ATOKEN_REVISION = 0x1;
34+
uint256 internal constant RELEASE_MARGIN = 10;
3435

3536
/// @dev owner => next valid nonce to submit with permit()
3637
mapping(address => uint256) public _nonces;
@@ -127,7 +128,10 @@ contract AToken is
127128
require(amountScaled != 0, Errors.CT_INVALID_BURN_AMOUNT);
128129
_burn(user, amountScaled);
129130

130-
IERC20(_underlyingAsset).safeTransfer(receiverOfUnderlying, amount);
131+
uint256 releaseAmount = amount > RELEASE_MARGIN ? amount - RELEASE_MARGIN : 0;
132+
if(releaseAmount != 0) {
133+
IERC20(_underlyingAsset).safeTransfer(receiverOfUnderlying, releaseAmount);
134+
}
131135

132136
emit Transfer(user, address(0), amount);
133137
emit Burn(user, receiverOfUnderlying, amount, index);

0 commit comments

Comments
 (0)