feat: Implement functionalities to update grandpa ghost and estimate.…

… Set round as completable and finalize in some specific conditions.
LimeChain · Feb 12, 2025 · 87f90ec · 87f90ec
1 parent 5a8129d
commit 87f90ec
Show file tree

Hide file tree

Showing 7 changed files with 249 additions and 148 deletions.
diff --git a/src/main/java/com/limechain/exception/grandpa/EstimateExecutionException.java b/src/main/java/com/limechain/exception/grandpa/EstimateExecutionException.java
@@ -0,0 +1,7 @@
+package com.limechain.exception.grandpa;
+
+public class EstimateExecutionException extends RuntimeException {
+    public EstimateExecutionException(String message) {
+        super(message);
+    }
+}
diff --git a/src/main/java/com/limechain/grandpa/round/FinalizeStage.java b/src/main/java/com/limechain/grandpa/round/FinalizeStage.java
@@ -28,10 +28,10 @@ public void start(GrandpaRound round) {
     public void end(GrandpaRound round) {
         log.info(String.format("Round %d met finalization conditions and will be finalized.", round.getRoundNumber()));
         round.setOnFinalizeHandler(null);
-        round.attemptToFinalizeAt();
+        round.attemptToFinalize();
         log.fine(String.format("Round %d exits Finalize stage.", round.getRoundNumber()));
 
-        round.end();
+        round.complete();
     }
 
     private boolean isRoundReadyToBeFinalized(GrandpaRound round) {