StateMachineProcedure getCurrentState() return null if state is EOF_S…

…TATE (apache#13704)
Caideyipi · Oct 12, 2024 · b9214a5 · b9214a5
1 parent 6b50f13
commit b9214a5
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...gnode/src/main/java/org/apache/iotdb/confignode/procedure/impl/StateMachineProcedure.java b/...gnode/src/main/java/org/apache/iotdb/confignode/procedure/impl/StateMachineProcedure.java
@@ -24,6 +24,7 @@
 import org.apache.iotdb.confignode.procedure.exception.ProcedureSuspendedException;
 import org.apache.iotdb.confignode.procedure.exception.ProcedureYieldException;
 
+import org.apache.thrift.annotation.Nullable;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -275,8 +276,15 @@ private boolean hasMoreState() {
     return stateFlow != Flow.NO_MORE_STATE;
   }
 
+  @Nullable
   protected TState getCurrentState() {
-    return stateCount > 0 ? getState(states[stateCount - 1]) : getInitialState();
+    if (stateCount > 0) {
+      if (states[stateCount - 1] == EOF_STATE) {
+        return null;
+      }
+      return getState(states[stateCount - 1]);
+    }
+    return getInitialState();
   }
 
   /**