Inline and remove isRunningOrDone

Inline and remove DistributedStagesSchedulerState.isRunningOrDone
trinodb · Jan 4, 2022 · 331770e · 331770e
1 parent 36c8f9e
commit 331770e
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/core/trino-main/src/main/java/io/trino/execution/scheduler/SqlQueryScheduler.java b/core/trino-main/src/main/java/io/trino/execution/scheduler/SqlQueryScheduler.java
@@ -321,7 +321,7 @@ private synchronized Optional<DistributedStagesScheduler> createDistributedStage
                 attempt);
         this.distributedStagesScheduler.set(distributedStagesScheduler);
         distributedStagesScheduler.addStateChangeListener(state -> {
-            if (queryStateMachine.getQueryState() == QueryState.STARTING && state.isRunningOrDone()) {
+            if (queryStateMachine.getQueryState() == QueryState.STARTING && (state == DistributedStagesSchedulerState.RUNNING || state.isDone())) {
                 queryStateMachine.transitionToRunning();
             }
 
@@ -1676,11 +1676,6 @@ public boolean isFailure()
         {
             return failureState;
         }
-
-        public boolean isRunningOrDone()
-        {
-            return this == RUNNING || isDone();
-        }
     }
 
     private static class DistributedStagesSchedulerStateMachine