-
Notifications
You must be signed in to change notification settings - Fork 250
/
reconciler-hooks.go
104 lines (87 loc) · 2.8 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
package controller
import (
"context"
"fmt"
"time"
"k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
cdiv1 "kubevirt.io/containerized-data-importer/pkg/apis/core/v1beta1"
"github.com/go-logr/logr"
corev1 "k8s.io/api/core/v1"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
"sigs.k8s.io/controller-runtime/pkg/reconcile"
)
// watch registers CDI-specific watches
func (r *ReconcileCDI) watch() error {
if err := r.reconciler.WatchResourceTypes(&corev1.ConfigMap{}, &corev1.Secret{}); err != nil {
return err
}
if err := r.watchRoutes(); err != nil {
return err
}
if err := r.watchSecurityContextConstraints(); err != nil {
return err
}
return nil
}
// preCreate creates the operator config map
func (r *ReconcileCDI) preCreate(cr controllerutil.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 *ReconcileCDI) checkSanity(cr controllerutil.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 cdi-config to be deleted before reconciling", "CDI", cr.GetName())
return &reconcile.Result{RequeueAfter: time.Second}, nil
}
reqLogger.Info("Reconciling to error state, unwanted CDI object")
result, err := r.reconciler.ReconcileError(cr, "Reconciling to error state, unwanted CDI object")
return &result, err
}
return nil, nil
}
// sync syncs certificates used by CDU
func (r *ReconcileCDI) sync(cr controllerutil.Object, logger logr.Logger) error {
cdi := cr.(*cdiv1.CDI)
return r.certManager.Sync(r.getCertificateDefinitions(cdi))
}
func (r *ReconcileCDI) configMapOwnerDeleted(cm *corev1.ConfigMap) (bool, error) {
ownerRef := metav1.GetControllerOf(cm)
if ownerRef != nil {
if ownerRef.Kind != "CDI" {
return false, fmt.Errorf("unexpected configmap owner kind %q", ownerRef.Kind)
}
owner := &cdiv1.CDI{}
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 *ReconcileCDI) registerHooks() {
r.reconciler.
WithPreCreateHook(r.preCreate).
WithWatchRegistrator(r.watch).
WithSanityChecker(r.checkSanity).
WithPerishablesSynchronizer(r.sync)
}