return if available parallelism is 0

Signed-off-by: Paul Dittamo <[email protected]>
flyteorg · Apr 3, 2024 · 65ab929 · 65ab929
1 parent 97c7c32
commit 65ab929
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flytepropeller/pkg/controller/nodes/array/handler.go b/flytepropeller/pkg/controller/nodes/array/handler.go
@@ -271,6 +271,10 @@ func (a *arrayNodeHandler) Handle(ctx context.Context, nCtx interfaces.NodeExecu
 
 		nodeExecutionRequests := make([]*nodeExecutionRequest, 0, availableParallelism)
 		for i, nodePhaseUint64 := range arrayNodeState.SubNodePhases.GetItems() {
+			if availableParallelism == 0 {
+				break
+			}
+
 			nodePhase := v1alpha1.NodePhase(nodePhaseUint64)
 			taskPhase := int(arrayNodeState.SubNodeTaskPhases.GetItem(i))
 
@@ -315,9 +319,6 @@ func (a *arrayNodeHandler) Handle(ctx context.Context, nCtx interfaces.NodeExecu
 				nCtx.ExecutionContext().IncrementParallelism()
 			}
 			availableParallelism--
-			if availableParallelism == 0 {
-				break
-			}
 		}
 
 		workerErrorCollector := errorcollector.NewErrorMessageCollector()