fix: revise type of kghNum to uint128

kroma-network · Jul 19, 2024 · f9df342 · f9df342
1 parent 9b380e2
commit f9df342
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/contracts/contracts/L1/AssetManager.sol b/packages/contracts/contracts/L1/AssetManager.sol
@@ -838,10 +838,10 @@ contract AssetManager is ISemver, IERC721Receiver, IAssetManager {
 
  Vault storage vault = _vaults[validator];
  KghDelegator storage kghDelegator = vault.kghDelegators[msg.sender];
- if (kghDelegator.kghNum < tokenIds.length) revert InvalidTokenIdsInput();
+ if (kghDelegator.kghNum < uint128(tokenIds.length)) revert InvalidTokenIdsInput();
 
  // Auto compounding is not applied for claim function.
- uint128 claimedRewards = _claimBoostedReward(validator, msg.sender, tokenIds.length, false);
+ uint128 claimedRewards = _claimBoostedReward(validator, msg.sender, uint128(tokenIds.length), false);
 
  // Burn kroShares to claim the base reward.
  uint128 kroSharesToBurn;
@@ -883,7 +883,7 @@ contract AssetManager is ISemver, IERC721Receiver, IAssetManager {
  function _claimBoostedReward(
  address validator,
  address delegator,
- uint256 kghNum,
+ uint128 kghNum,
  bool isCompounding
  ) internal returns (uint128) {
  Vault storage vault = _vaults[validator];

diff --git a/packages/contracts/contracts/L1/interfaces/IAssetManager.sol b/packages/contracts/contracts/L1/interfaces/IAssetManager.sol
@@ -54,7 +54,7 @@ interface IAssetManager {
  */
  struct KghDelegator {
  uint128 rewardPerKghPaid;
- uint256 kghNum;
+ uint128 kghNum;
  mapping(uint256 => uint128) delegatedAt;
  mapping(uint256 => uint128) kroShares;
  }