chore: add startNewCycle function

own-protocol · Jan 20, 2025 · 1033581 · 1033581
1 parent 162ae8c
commit 1033581
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/src/interfaces/IAssetPool.sol b/src/interfaces/IAssetPool.sol
@@ -208,6 +208,11 @@ interface IAssetPool {
      */
     function rebalancePool(address lp, uint256 rebalancePrice, uint256 amount, bool isDeposit) external;
 
+    /**
+     * @notice When there is nothing to rebalance, start the new cycle
+     */
+    function startNewCycle() external;
+
     // --------------------------------------------------------------------------------
     //                              GOVERNANCE ACTIONS
     // --------------------------------------------------------------------------------

diff --git a/src/protocol/AssetPool.sol b/src/protocol/AssetPool.sol
@@ -393,19 +393,25 @@ contract AssetPool is IAssetPool, Ownable, Pausable {
 
         // If all LPs have rebalanced, start next cycle
         if (rebalancedLPs == lpRegistry.getLPCount(address(this))) {
+            uint256 totalLiquidity = lpRegistry.getTotalLPLiquidity(address(this));
+            uint256 assetBalance = assetToken.balanceOf(address(this));
+            uint256 reserveBalanceInAssetToken = assetToken.reserveBalanceOf(address(this));
+            assetToken.burn(address(this), assetBalance, reserveBalanceInAssetToken);
+            cycleRebalancePrice[cycleIndex] = cycleWeightedSum[cycleIndex] / totalLiquidity;
+
             _startNewCycle();
         }
     }
 
-    function updateCycle() external {
+    /**
+     * @notice If there is nothing to rebalance, start the next cycle.
+     */
+    function startNewCycle() external {
         if (cycleState != CycleState.REBALANCING) revert InvalidCycleState();
         if (cycleTotalDepositRequests[cycleIndex] > 0) revert InvalidCycleRequest();
         if (cycleTotalRedemptionRequests[cycleIndex] > 0) revert InvalidCycleRequest();
 
-        cycleIndex++;
-        cycleState = CycleState.ACTIVE;
-        nextRebalanceStartDate = block.timestamp + cycleTime;
-        nextRebalanceEndDate = nextRebalanceStartDate + rebalanceTime;
+        _startNewCycle();
     }
 
     // --------------------------------------------------------------------------------
@@ -473,11 +479,6 @@ contract AssetPool is IAssetPool, Ownable, Pausable {
      * @notice Starts a new cycle after all LPs have rebalanced.
      */
     function _startNewCycle() internal {
-        uint256 totalLiquidity = lpRegistry.getTotalLPLiquidity(address(this));
-        uint256 assetBalance = assetToken.balanceOf(address(this));
-        uint256 reserveBalanceInAssetToken = assetToken.reserveBalanceOf(address(this));
-        assetToken.burn(address(this), assetBalance, reserveBalanceInAssetToken);
-        cycleRebalancePrice[cycleIndex] = cycleWeightedSum[cycleIndex] / totalLiquidity;
         cycleIndex++;
         cycleState = CycleState.ACTIVE;
         rebalancedLPs = 0;