This repository has been archived by the owner on Mar 4, 2019. It is now read-only.
forked from kmodules/client-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
statefulset.go
119 lines (106 loc) · 4.17 KB
/
statefulset.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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package v1
import (
. "github.com/appscode/go/types"
atypes "github.com/appscode/go/types"
"github.com/appscode/kutil"
core_util "github.com/appscode/kutil/core/v1"
"github.com/golang/glog"
"github.com/pkg/errors"
apps "k8s.io/api/apps/v1"
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/strategicpatch"
"k8s.io/apimachinery/pkg/util/wait"
"k8s.io/client-go/kubernetes"
)
func CreateOrPatchStatefulSet(c kubernetes.Interface, meta metav1.ObjectMeta, transform func(*apps.StatefulSet) *apps.StatefulSet) (*apps.StatefulSet, kutil.VerbType, error) {
cur, err := c.AppsV1().StatefulSets(meta.Namespace).Get(meta.Name, metav1.GetOptions{})
if kerr.IsNotFound(err) {
glog.V(3).Infof("Creating StatefulSet %s/%s.", meta.Namespace, meta.Name)
out, err := c.AppsV1().StatefulSets(meta.Namespace).Create(transform(&apps.StatefulSet{
TypeMeta: metav1.TypeMeta{
Kind: "StatefulSet",
APIVersion: apps.SchemeGroupVersion.String(),
},
ObjectMeta: meta,
}))
return out, kutil.VerbCreated, err
} else if err != nil {
return nil, kutil.VerbUnchanged, err
}
return PatchStatefulSet(c, cur, transform)
}
func PatchStatefulSet(c kubernetes.Interface, cur *apps.StatefulSet, transform func(*apps.StatefulSet) *apps.StatefulSet) (*apps.StatefulSet, kutil.VerbType, error) {
return PatchStatefulSetObject(c, cur, transform(cur.DeepCopy()))
}
func PatchStatefulSetObject(c kubernetes.Interface, cur, mod *apps.StatefulSet) (*apps.StatefulSet, 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 := strategicpatch.CreateTwoWayMergePatch(curJson, modJson, apps.StatefulSet{})
if err != nil {
return nil, kutil.VerbUnchanged, err
}
if len(patch) == 0 || string(patch) == "{}" {
return cur, kutil.VerbUnchanged, nil
}
glog.V(3).Infof("Patching StatefulSet %s/%s with %s.", cur.Namespace, cur.Name, string(patch))
out, err := c.AppsV1().StatefulSets(cur.Namespace).Patch(cur.Name, types.StrategicMergePatchType, patch)
return out, kutil.VerbPatched, err
}
func TryUpdateStatefulSet(c kubernetes.Interface, meta metav1.ObjectMeta, transform func(*apps.StatefulSet) *apps.StatefulSet) (result *apps.StatefulSet, err error) {
attempt := 0
err = wait.PollImmediate(kutil.RetryInterval, kutil.RetryTimeout, func() (bool, error) {
attempt++
cur, e2 := c.AppsV1().StatefulSets(meta.Namespace).Get(meta.Name, metav1.GetOptions{})
if kerr.IsNotFound(e2) {
return false, e2
} else if e2 == nil {
result, e2 = c.AppsV1().StatefulSets(cur.Namespace).Update(transform(cur.DeepCopy()))
return e2 == nil, nil
}
glog.Errorf("Attempt %d failed to update StatefulSet %s/%s due to %v.", attempt, cur.Namespace, cur.Name, e2)
return false, nil
})
if err != nil {
err = errors.Errorf("failed to update StatefulSet %s/%s after %d attempts due to %v", meta.Namespace, meta.Name, attempt, err)
}
return
}
func WaitUntilStatefulSetReady(kubeClient kubernetes.Interface, meta metav1.ObjectMeta) error {
return wait.PollImmediate(kutil.RetryInterval, kutil.ReadinessTimeout, func() (bool, error) {
if obj, err := kubeClient.AppsV1().StatefulSets(meta.Namespace).Get(meta.Name, metav1.GetOptions{}); err == nil {
return Int32(obj.Spec.Replicas) == obj.Status.ReadyReplicas, nil
}
return false, nil
})
}
func DeleteStatefulSet(kubeClient kubernetes.Interface, meta metav1.ObjectMeta) error {
statefulSet, err := kubeClient.AppsV1().StatefulSets(meta.Namespace).Get(meta.Name, metav1.GetOptions{})
if err != nil {
if kerr.IsNotFound(err) {
return nil
} else {
return err
}
}
// Update StatefulSet
_, _, err = PatchStatefulSet(kubeClient, statefulSet, func(in *apps.StatefulSet) *apps.StatefulSet {
in.Spec.Replicas = atypes.Int32P(0)
return in
})
if err != nil {
return err
}
err = core_util.WaitUntilPodDeletedBySelector(kubeClient, statefulSet.Namespace, statefulSet.Spec.Selector)
if err != nil {
return err
}
return kubeClient.AppsV1().StatefulSets(statefulSet.Namespace).Delete(statefulSet.Name, nil)
}