fix: fix suspend judgement

Signed-off-by: FogDong <fog@bentoml.com>
kubevela · Sep 24, 2024 · 3dd8afc · 3dd8afc
1 parent 76db4ac
commit 3dd8afc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pkg/tasks/custom/action.go b/pkg/tasks/custom/action.go
@@ -85,7 +85,7 @@ func (exec *executor) Terminate(message string) {
 // Wait let workflow wait.
 func (exec *executor) Wait(message string) {
 	exec.wait = true
-	if exec.wfStatus.Phase != v1alpha1.WorkflowStepPhaseFailed {
+	if exec.wfStatus.Phase != v1alpha1.WorkflowStepPhaseFailed && exec.wfStatus.Phase != v1alpha1.WorkflowStepPhaseSuspending {
 		exec.wfStatus.Phase = v1alpha1.WorkflowStepPhaseRunning
 		exec.wfStatus.Reason = types.StatusReasonWait
 		if message != "" {

diff --git a/pkg/tasks/custom/task.go b/pkg/tasks/custom/task.go
@@ -226,6 +226,10 @@ func (t *TaskLoader) makeTaskGenerator(templ string) (types.TaskGenerator, error
 
 			if status, ok := options.StepStatus[wfStep.Name]; ok {
 				exec.stepStatus = status
+				exec.wfStatus = status
+				if status.Phase == v1alpha1.WorkflowStepPhaseSuspending {
+					exec.suspend = true
+				}
 			}
 			basicTempl, err := util.ToString(basicVal)
 			if err != nil {