/
nodecadaemon.go
118 lines (98 loc) · 3.52 KB
/
nodecadaemon.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
package resource
import (
"context"
"os"
appsv1 "k8s.io/api/apps/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
appsv1client "k8s.io/client-go/kubernetes/typed/apps/v1"
appsv1listers "k8s.io/client-go/listers/apps/v1"
corev1listers "k8s.io/client-go/listers/core/v1"
operatorv1 "github.com/openshift/api/operator/v1"
"github.com/openshift/library-go/pkg/operator/events"
"github.com/openshift/library-go/pkg/operator/resource/resourceapply"
"github.com/openshift/library-go/pkg/operator/resource/resourcemerge"
"github.com/openshift/library-go/pkg/operator/resource/resourceread"
"github.com/openshift/library-go/pkg/operator/v1helpers"
assets "github.com/openshift/cluster-image-registry-operator/bindata"
"github.com/openshift/cluster-image-registry-operator/pkg/defaults"
)
var _ Mutator = &generatorNodeCADaemonSet{}
type generatorNodeCADaemonSet struct {
eventRecorder events.Recorder
daemonSetLister appsv1listers.DaemonSetNamespaceLister
serviceLister corev1listers.ServiceNamespaceLister
client appsv1client.AppsV1Interface
operatorClient v1helpers.OperatorClient
}
func NewGeneratorNodeCADaemonSet(eventRecorder events.Recorder, daemonSetLister appsv1listers.DaemonSetNamespaceLister, serviceLister corev1listers.ServiceNamespaceLister, client appsv1client.AppsV1Interface, operatorClient v1helpers.OperatorClient) Mutator {
return &generatorNodeCADaemonSet{
eventRecorder: eventRecorder,
daemonSetLister: daemonSetLister,
serviceLister: serviceLister,
client: client,
operatorClient: operatorClient,
}
}
func (ds *generatorNodeCADaemonSet) Type() runtime.Object {
return &appsv1.DaemonSet{}
}
func (ds *generatorNodeCADaemonSet) GetNamespace() string {
return defaults.ImageRegistryOperatorNamespace
}
func (ds *generatorNodeCADaemonSet) GetName() string {
return "node-ca"
}
func (ds *generatorNodeCADaemonSet) Get() (runtime.Object, error) {
return ds.daemonSetLister.Get(ds.GetName())
}
func (ds *generatorNodeCADaemonSet) expected() *appsv1.DaemonSet {
daemonSet := resourceread.ReadDaemonSetV1OrDie(assets.MustAsset("nodecadaemon.yaml"))
daemonSet.Spec.Template.Spec.Containers[0].Image = os.Getenv("IMAGE")
return daemonSet
}
func (ds *generatorNodeCADaemonSet) Create() (runtime.Object, error) {
dep, _, err := ds.Update(nil)
return dep, err
}
func (ds *generatorNodeCADaemonSet) Update(o runtime.Object) (runtime.Object, bool, error) {
desiredDaemonSet := ds.expected()
_, opStatus, _, err := ds.operatorClient.GetOperatorState()
if err != nil {
return nil, false, err
}
actualDaemonSet, updated, err := resourceapply.ApplyDaemonSet(
context.TODO(),
ds.client,
ds.eventRecorder,
desiredDaemonSet,
resourcemerge.ExpectedDaemonSetGeneration(desiredDaemonSet, opStatus.Generations),
)
if err != nil {
return o, updated, err
}
if updated {
updateStatusFn := func(newStatus *operatorv1.OperatorStatus) error {
resourcemerge.SetDaemonSetGeneration(&newStatus.Generations, actualDaemonSet)
return nil
}
_, _, err = v1helpers.UpdateStatus(
context.TODO(),
ds.operatorClient,
updateStatusFn,
)
if err != nil {
return actualDaemonSet, updated, err
}
}
return actualDaemonSet, updated, nil
}
func (ds *generatorNodeCADaemonSet) Delete(opts metav1.DeleteOptions) error {
return ds.client.DaemonSets(ds.GetNamespace()).Delete(
context.TODO(), ds.GetName(), opts,
)
}
func (ds *generatorNodeCADaemonSet) Owned() bool {
// the nodeca daemon's lifecycle is not tied to the lifecycle of the registry
return false
}