Gas optimizations

kilnfi · Mar 27, 2023 · 2115675 · 2115675
1 parent eaaff69
commit 2115675
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/contracts/StakingContract.sol b/src/contracts/StakingContract.sol
@@ -556,9 +556,12 @@ contract StakingContract {
             revert InvalidPublicKeys();
         }
         uint256 keyCount = _publicKeys.length / PUBLIC_KEY_LENGTH;
-        for (uint256 i = 0; i < keyCount; i++) {
+        for (uint256 i = 0; i < keyCount; ) {
             bytes memory publicKey = BytesLib.slice(_publicKeys, i * PUBLIC_KEY_LENGTH, PUBLIC_KEY_LENGTH);
             _deployAndWithdraw(publicKey, EXECUTION_LAYER_SALT_PREFIX, StakingContractStorageLib.getELDispatcher());
+            unchecked {
+                ++i;
+            }
         }
     }
 
@@ -571,9 +574,12 @@ contract StakingContract {
             revert InvalidPublicKeys();
         }
         uint256 keyCount = _publicKeys.length / PUBLIC_KEY_LENGTH;
-        for (uint256 i = 0; i < keyCount; i++) {
+        for (uint256 i = 0; i < keyCount; ) {
             bytes memory publicKey = BytesLib.slice(_publicKeys, i * PUBLIC_KEY_LENGTH, PUBLIC_KEY_LENGTH);
             _deployAndWithdraw(publicKey, CONSENSUS_LAYER_SALT_PREFIX, StakingContractStorageLib.getCLDispatcher());
+            unchecked {
+                ++i;
+            }
         }
     }
 
@@ -586,10 +592,13 @@ contract StakingContract {
             revert InvalidPublicKeys();
         }
         uint256 keyCount = _publicKeys.length / PUBLIC_KEY_LENGTH;
-        for (uint256 i = 0; i < keyCount; i++) {
+        for (uint256 i = 0; i < keyCount; ) {
             bytes memory publicKey = BytesLib.slice(_publicKeys, i * PUBLIC_KEY_LENGTH, PUBLIC_KEY_LENGTH);
             _deployAndWithdraw(publicKey, EXECUTION_LAYER_SALT_PREFIX, StakingContractStorageLib.getELDispatcher());
             _deployAndWithdraw(publicKey, CONSENSUS_LAYER_SALT_PREFIX, StakingContractStorageLib.getCLDispatcher());
+            unchecked {
+                ++i;
+            }
         }
     }