-
Notifications
You must be signed in to change notification settings - Fork 6
/
reconciler-hooks.go
105 lines (89 loc) · 2.82 KB
/
reconciler-hooks.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
package aaq_operator
import (
"context"
"fmt"
"kubevirt.io/application-aware-quota/staging/src/kubevirt.io/application-aware-quota-api/pkg/apis/core/v1alpha1"
"time"
"k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/go-logr/logr"
corev1 "k8s.io/api/core/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/reconcile"
)
// watch registers AAQ-specific watches
func (r *ReconcileAAQ) watch() error {
if err := r.reconciler.WatchResourceTypes(&corev1.ConfigMap{}, &corev1.Secret{}); err != nil {
return err
}
if err := r.watchAAQCRD(); err != nil {
return err
}
if err := r.watchSecurityContextConstraints(); err != nil {
return err
}
return nil
}
// preCreate creates the operator config map
func (r *ReconcileAAQ) preCreate(cr client.Object) error {
// claim the configmap
if err := r.createOperatorConfig(cr); err != nil {
return err
}
return nil
}
// checkSanity verifies whether config map exists and is in proper relation with the cr
func (r *ReconcileAAQ) checkSanity(cr client.Object, reqLogger logr.Logger) (*reconcile.Result, error) {
configMap, err := r.getConfigMap()
if err != nil {
return &reconcile.Result{}, err
}
if !metav1.IsControlledBy(configMap, cr) {
ownerDeleted, err := r.configMapOwnerDeleted(configMap)
if err != nil {
return &reconcile.Result{}, err
}
if ownerDeleted || configMap.DeletionTimestamp != nil {
reqLogger.Info("Waiting for aaq-config to be deleted before reconciling", "AAQ", cr.GetName())
return &reconcile.Result{RequeueAfter: time.Second}, nil
}
reqLogger.Info("Reconciling to error state, unwanted AAQ object")
result, err := r.reconciler.ReconcileError(cr, "Reconciling to error state, unwanted AAQ object")
return &result, err
}
return nil, nil
}
// sync syncs certificates used by AAQ
func (r *ReconcileAAQ) sync(cr client.Object, logger logr.Logger) error {
aaq := cr.(*v1alpha1.AAQ)
if aaq.DeletionTimestamp != nil {
return nil
}
return r.certManager.Sync(r.getCertificateDefinitions(aaq))
}
func (r *ReconcileAAQ) configMapOwnerDeleted(cm *corev1.ConfigMap) (bool, error) {
ownerRef := metav1.GetControllerOf(cm)
if ownerRef != nil {
if ownerRef.Kind != "AAQ" {
return false, fmt.Errorf("unexpected configmap owner kind %q", ownerRef.Kind)
}
owner := &v1alpha1.AAQ{}
if err := r.client.Get(context.TODO(), client.ObjectKey{Name: ownerRef.Name}, owner); err != nil {
if errors.IsNotFound(err) {
return true, nil
}
return false, err
}
if owner.DeletionTimestamp == nil && owner.UID == ownerRef.UID {
return false, nil
}
}
return true, nil
}
func (r *ReconcileAAQ) registerHooks() {
r.reconciler.
WithPreCreateHook(r.preCreate).
WithWatchRegistrator(r.watch).
WithSanityChecker(r.checkSanity).
WithPerishablesSynchronizer(r.sync)
}