fix: Unable to abort pre-cd and post-cd workflow

pkg/pipeline/CdHandler.go

@@ -52,6 +52,7 @@ import (
 	"github.com/go-pg/pg"
 	"go.opentelemetry.io/otel"
 	"go.uber.org/zap"
+	"k8s.io/client-go/rest"
 	"os"
 	"path/filepath"
 	"strconv"
@@ -600,10 +601,13 @@ func (impl *CdHandlerImpl) CancelStage(workflowRunnerId int, userId int32) (int,
 	} else if workflowRunner.WorkflowType == POST {
 		isExtCluster = pipeline.RunPostStageInEnv
 	}
-	restConfig, err := impl.k8sUtil.GetRestConfigByCluster(clusterConfig)
-	if err != nil {
-		impl.Logger.Errorw("error in getting rest config by cluster id", "err", err)
-		return 0, err
+	var restConfig *rest.Config
+	if isExtCluster {
+		restConfig, err = impl.k8sUtil.GetRestConfigByCluster(clusterConfig)
+		if err != nil {
+			impl.Logger.Errorw("error in getting rest config by cluster id", "err", err)
+			return 0, err
+		}
 	}
 	// Terminate workflow
 	err = impl.workflowService.TerminateWorkflow(workflowRunner.ExecutorType, workflowRunner.Name, workflowRunner.Namespace, restConfig, isExtCluster, nil)

pkg/pipeline/WorkflowService.go

@@ -295,6 +295,9 @@ func (impl *WorkflowServiceImpl) TerminateWorkflow(executorType pipelineConfig.W
 	var err error
 	if executorType != "" {
 		workflowExecutor := impl.getWorkflowExecutor(executorType)
+		if restConfig == nil {
+			restConfig = impl.config
+		}
 		err = workflowExecutor.TerminateWorkflow(name, namespace, restConfig)
 	} else {
 		wfClient, err := impl.getWfClient(environment, namespace, isExt)