zishuo/onepanel
1
0
Fork 0
mirror of https://github.com/onepanelio/onepanel.git synced 2025-10-20 20:25:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add terminate workflow service

Browse Source
This commit is contained in:
rushtehrani
2020-02-06 14:29:21 -08:00
parent 3c0122582c
commit c66a7f6c50
7 changed files with 355 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
kube/workflow.go
View File

@@ -1,14 +1,17 @@
package kube
import (
"encoding/json"
"fmt"
wfv1 "github.com/argoproj/argo/pkg/apis/workflow/v1alpha1"
"github.com/argoproj/argo/workflow/common"
"github.com/argoproj/pkg/json"
argojson "github.com/argoproj/pkg/json"
"github.com/onepanelio/core/util/env"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/fields"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/watch"
_ "k8s.io/client-go/plugin/pkg/client/auth"
)
@@ -34,11 +37,11 @@ type WorkflowOptions struct {
func unmarshalWorkflows(wfBytes []byte, strict bool) (wfs []Workflow, err error) {
var wf Workflow
var jsonOpts []json.JSONOpt
var jsonOpts []argojson.JSONOpt
if strict {
jsonOpts = append(jsonOpts, json.DisallowUnknownFields)
jsonOpts = append(jsonOpts, argojson.DisallowUnknownFields)
}
err = json.Unmarshal(wfBytes, &wf, jsonOpts...)
err = argojson.Unmarshal(wfBytes, &wf, jsonOpts...)
if err == nil {
return []Workflow{wf}, nil
}
@@ -168,3 +171,21 @@ func (c *Client) WatchWorkflow(namespace, name string) (watcher watch.Interface,
return
}
func (c *Client) TerminateWorkflow(namespace, name string) (err error) {
obj := map[string]interface{}{
"spec": map[string]interface{}{
"activeDeadlineSeconds": 0,
},
}
patch, err := json.Marshal(obj)
if err != nil {
return
}
_, err = c.ArgoprojV1alpha1().Workflows(namespace).Patch(name, types.MergePatchType, patch)
if err != nil {
return
}
return
}