Files
onepanel/manager/workflow_manager.go
2019-12-13 13:19:05 -08:00

63 lines
2.0 KiB
Go

package manager
import (
"github.com/onepanelio/core/argo"
"github.com/onepanelio/core/model"
"github.com/onepanelio/core/util"
"google.golang.org/grpc/codes"
)
func (r *ResourceManager) CreateWorkflow(namespace string, workflow *model.Workflow) (createdWorkflow *model.Workflow, err error) {
workflowTemplate, err := r.workflowRepository.GetWorkflowTemplate(workflow.WorkflowTemplate.UID, workflow.WorkflowTemplate.Version)
if err != nil {
return nil, util.NewUserError(codes.NotFound, "Workflow template not found.")
}
opts := &argo.Options{
Namespace: namespace,
}
for _, param := range workflow.Parameters {
opts.Parameters = append(opts.Parameters, argo.Parameter{
Name: param.Name,
Value: param.Value,
})
}
createdWorkflows, err := r.argClient.Create(workflowTemplate.GetManifestBytes(), opts)
if err != nil {
return
}
createdWorkflow = workflow
createdWorkflow.Name = createdWorkflows[0].Name
createdWorkflow.UID = string(createdWorkflows[0].ObjectMeta.UID)
return
}
func (r *ResourceManager) CreateWorkflowTemplate(namespace string, workflowTemplate *model.WorkflowTemplate) (createdWorkflowTemplate *model.WorkflowTemplate, err error) {
createdWorkflowTemplate, err = r.workflowRepository.CreateWorkflowTemplate(workflowTemplate)
if err != nil {
return nil, util.NewUserErrorWrap(err, "Workflow template")
}
return
}
func (r *ResourceManager) GetWorkflowTemplate(namespace, uid string, version int32) (workflowTemplate *model.WorkflowTemplate, err error) {
workflowTemplate, err = r.workflowRepository.GetWorkflowTemplate(uid, version)
if err != nil {
return nil, util.NewUserError(codes.NotFound, "Workflow template not found.")
}
return
}
func (r *ResourceManager) ListWorkflowTemplateVersions(namespace, uid string) (workflowTemplateVersions []*model.WorkflowTemplate, err error) {
workflowTemplateVersions, err = r.workflowRepository.ListWorkflowTemplateVersions(uid)
if err != nil {
return nil, util.NewUserError(codes.NotFound, "Workflow template not found.")
}
return
}