add disable workflow feature

Signed-off-by: Patrick Zhao <[email protected]>
koderover · Jun 27, 2024 · 762bab7 · 762bab7
1 parent 49e5098
commit 762bab7
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 0 deletions.
diff --git a/pkg/microservice/aslan/core/common/repository/models/workflow_v4.go b/pkg/microservice/aslan/core/common/repository/models/workflow_v4.go
@@ -42,6 +42,7 @@ type WorkflowV4 struct {
 	ID              primitive.ObjectID       `bson:"_id,omitempty"       yaml:"-"                   json:"id"`
 	Name            string                   `bson:"name"                yaml:"name"                json:"name"`
 	DisplayName     string                   `bson:"display_name"        yaml:"display_name"        json:"display_name"`
+	Disabled        bool                     `bson:"disabled"            yaml:"disabled"            json:"disabled"`
 	Category        setting.WorkflowCategory `bson:"category"            yaml:"category"            json:"category"`
 	KeyVals         []*KeyVal                `bson:"key_vals"            yaml:"key_vals"            json:"key_vals"`
 	Params          []*Param                 `bson:"params"              yaml:"params"              json:"params"`

diff --git a/pkg/microservice/aslan/core/workflow/service/workflow/workflow.go b/pkg/microservice/aslan/core/workflow/service/workflow/workflow.go
@@ -51,6 +51,7 @@ type Workflow struct {
 	Name                 string                     `json:"name"`
 	DisplayName          string                     `json:"display_name"`
 	ProjectName          string                     `json:"projectName"`
+	Disabled             bool                       `json:"disabled"`
 	UpdateTime           int64                      `json:"updateTime"`
 	CreateTime           int64                      `json:"createTime"`
 	UpdateBy             string                     `json:"updateBy,omitempty"`

diff --git a/pkg/microservice/aslan/core/workflow/service/workflow/workflow_task_v4.go b/pkg/microservice/aslan/core/workflow/service/workflow/workflow_task_v4.go
@@ -380,6 +380,20 @@ func CreateWorkflowTaskV4(args *CreateWorkflowTaskV4Args, workflow *commonmodels
 		return resp, err
 	}
 
+	if args.Type == config.WorkflowTaskTypeWorkflow || args.Type == "" {
+		orignalWorkflow, err := commonrepo.NewWorkflowV4Coll().Find(workflow.Name)
+		if err != nil {
+			return resp, e.ErrCreateTask.AddErr(fmt.Errorf("cannot find workflow %s, error: %v", workflow.Name, err))
+		}
+		if orignalWorkflow.Disabled {
+			return resp, e.ErrCreateTask.AddDesc("workflow is disabled")
+		}
+	} else {
+		if workflow.Disabled {
+			return resp, e.ErrCreateTask.AddDesc("workflow is disabled")
+		}
+	}
+
 	// if account is not set, use name as account
 	if args.Account == "" {
 		args.Account = args.Name

diff --git a/pkg/microservice/aslan/core/workflow/service/workflow/workflow_v4.go b/pkg/microservice/aslan/core/workflow/service/workflow/workflow_v4.go
@@ -450,6 +450,7 @@ func ListWorkflowV4(projectName, viewName, userID string, names, v4Names []strin
 			Name:          workflowModel.Name,
 			DisplayName:   workflowModel.DisplayName,
 			ProjectName:   workflowModel.Project,
+			Disabled:      workflowModel.Disabled,
 			EnabledStages: stages,
 			CreateTime:    workflowModel.CreateTime,
 			UpdateTime:    workflowModel.UpdateTime,