fix: vesting contract gas optimize

Author: PavelNaydanov

src/tokenDistribution/vesting/Vesting.sol

@@ -85,34 +85,30 @@ contract Vesting is IVesting, Initializable {
     }
 
     /// @notice Returns the total amount of unlocked tokens
-    function totalUnlocked() public view returns (uint256) {
+    function totalUnlocked() external view returns (uint256) {
         return _computeUnlocked(_initialTotalLocked);
     }
 
     /// @notice Returns the total amount of locked tokens
     function totalLocked() external view returns (uint256) {
-        // TODO: заменить на
-        // uint256 initialTotalLocked = _initialTotalLocked;
-        // return initialTotalLocked - _computeUnlocked(initialTotalLocked);
-        return _initialTotalLocked - totalUnlocked();
+        uint256 initialTotalLocked = _initialTotalLocked;
+        return initialTotalLocked - _computeUnlocked(initialTotalLocked);
     }
 
     /// @notice Returns the amount of unlocked tokens for a specific account
-    function unlockedOf(address account) public view returns (uint256) {
+    function unlockedOf(address account) external view returns (uint256) {
         return _computeUnlocked(_initialLocked[account]);
     }
 
     /// @notice Returns the amount of locked tokens for a specific account
     function lockedOf(address account) external view returns (uint256) {
-        // TODO: заменить на
-        // uint256 initialLocked = _initialLocked[account];
-        // return initialLocked - _computeUnlocked(initialLocked);
-        return _initialLocked[account] - unlockedOf(account);
+        uint256 initialLocked = _initialLocked[account];
+        return initialLocked - _computeUnlocked(initialLocked);
     }
 
     /// @notice Returns the available amount of unlocked tokens that can be claimed by a specific account
     function availableToClaim(address account) public view returns (uint256) {
-        return unlockedOf(account) - _released[account];
+        return _computeUnlocked(_initialLocked[account]) - _released[account];
     }
 
     // endregion
@@ -174,6 +170,7 @@ contract Vesting is IVesting, Initializable {
             revert ZeroBeneficiaries();
         }
 
+        uint256 initialTotalLocked = 0;
         for (uint256 i = 0; i < beneficiaries.length; i++) {
             Beneficiary memory beneficiary = beneficiaries[i];
 
@@ -189,15 +186,16 @@ contract Vesting is IVesting, Initializable {
                 revert DuplicateBeneficiary(beneficiary.account);
             }
 
+            initialTotalLocked += beneficiary.amount;
             _initialLocked[beneficiary.account] = beneficiary.amount;
-            // TODO: переделать на чтение из memory и последующая запись в storage
-            _initialTotalLocked += beneficiary.amount;
         }
 
-        if (_initialTotalLocked != baseToken.balanceOf(address(this))) {
+        if (initialTotalLocked != baseToken.balanceOf(address(this))) {
             revert IncorrectAmountOfBeneficiaries();
         }
 
+        _initialTotalLocked = initialTotalLocked;
+
         emit BeneficiariesInitialized(beneficiaries);
     }
 
@@ -207,16 +205,15 @@ contract Vesting is IVesting, Initializable {
      * @return unlockedAmount The total amount of unlocked tokens
      */
     function _computeUnlocked(uint256 initialLockedAmount) private view returns (uint256 unlockedAmount) {
-        // TODO: добавить _schedule в memory
-        if (block.timestamp < _schedule.startTime) {
+        Schedule memory schedule = _schedule;
+
+        if (block.timestamp < schedule.startTime) {
             return 0;
         }
 
-        uint256 startTime = _schedule.startTime;
-        Period[] memory periods = _schedule.periods;
-
-        for (uint256 i = 0; i < periods.length; i++) {
-            Period memory period = _schedule.periods[i];
+        uint256 startTime = schedule.startTime;
+        for (uint256 i = 0; i < schedule.periods.length; i++) {
+            Period memory period = schedule.periods[i];
 
             uint256 endTime = period.endTime;
             uint256 portion = period.portion;