/
task.go
81 lines (72 loc) 路 2.65 KB
/
task.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package util
import (
"fmt"
api "github.com/appscode/stash/apis/stash/v1beta1"
cs "github.com/appscode/stash/client/clientset/versioned/typed/stash/v1beta1"
jsonpatch "github.com/evanphx/json-patch"
"github.com/golang/glog"
kerr "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/util/wait"
kutil "kmodules.xyz/client-go"
)
func CreateOrPatchTask(c cs.StashV1beta1Interface, meta metav1.ObjectMeta, transform func(in *api.Task) *api.Task) (*api.Task, kutil.VerbType, error) {
cur, err := c.Tasks().Get(meta.Name, metav1.GetOptions{})
if kerr.IsNotFound(err) {
glog.V(3).Infof("Creating Task %s/%s.", meta.Namespace, meta.Name)
out, err := c.Tasks().Create(transform(&api.Task{
TypeMeta: metav1.TypeMeta{
Kind: "Task",
APIVersion: api.SchemeGroupVersion.String(),
},
ObjectMeta: meta,
}))
return out, kutil.VerbCreated, err
} else if err != nil {
return nil, kutil.VerbUnchanged, err
}
return PatchTask(c, cur, transform)
}
func PatchTask(c cs.StashV1beta1Interface, cur *api.Task, transform func(*api.Task) *api.Task) (*api.Task, kutil.VerbType, error) {
return PatchTaskObject(c, cur, transform(cur.DeepCopy()))
}
func PatchTaskObject(c cs.StashV1beta1Interface, cur, mod *api.Task) (*api.Task, kutil.VerbType, error) {
curJson, err := json.Marshal(cur)
if err != nil {
return nil, kutil.VerbUnchanged, err
}
modJson, err := json.Marshal(mod)
if err != nil {
return nil, kutil.VerbUnchanged, err
}
patch, err := jsonpatch.CreateMergePatch(curJson, modJson)
if err != nil {
return nil, kutil.VerbUnchanged, err
}
if len(patch) == 0 || string(patch) == "{}" {
return cur, kutil.VerbUnchanged, nil
}
glog.V(3).Infof("Patching Task %s/%s with %s.", cur.Namespace, cur.Name, string(patch))
out, err := c.Tasks().Patch(cur.Name, types.MergePatchType, patch)
return out, kutil.VerbPatched, err
}
func TryUpdateTask(c cs.StashV1beta1Interface, meta metav1.ObjectMeta, transform func(*api.Task) *api.Task) (result *api.Task, err error) {
attempt := 0
err = wait.PollImmediate(kutil.RetryInterval, kutil.RetryTimeout, func() (bool, error) {
attempt++
cur, e2 := c.Tasks().Get(meta.Name, metav1.GetOptions{})
if kerr.IsNotFound(e2) {
return false, e2
} else if e2 == nil {
result, e2 = c.Tasks().Update(transform(cur.DeepCopy()))
return e2 == nil, nil
}
glog.Errorf("Attempt %d failed to update Task %s/%s due to %v.", attempt, cur.Namespace, cur.Name, e2)
return false, nil
})
if err != nil {
err = fmt.Errorf("failed to update Task %s/%s after %d attempts due to %v", meta.Namespace, meta.Name, attempt, err)
}
return
}