fluxcd
/
notification-controller
mirror of https://github.com/fluxcd/notification-controller.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add finalizers to all the CRDs 

Signed-off-by: Somtochi Onyekwere <somtochionyekwere@gmail.com>
This commit is contained in:
Somtochi Onyekwere 2022-09-08 21:02:47 +01:00
parent 8151979f52
commit 071dda5e97
4 changed files with 41 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/v1beta1/condition_types.go
View File

@ -16,6 +16,8 @@ limitations under the License.
package v1beta1
const NotificationFinalizer = "finalizers.fluxcd.io"
const (
// InitializedReason represents the fact that a given resource has been initialized.
InitializedReason string = "Initialized"

13
controllers/alert_controller.go
View File

@ -29,6 +29,7 @@ import (
"sigs.k8s.io/controller-runtime/pkg/builder"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
"sigs.k8s.io/controller-runtime/pkg/handler"
"sigs.k8s.io/controller-runtime/pkg/predicate"
"sigs.k8s.io/controller-runtime/pkg/ratelimiter"
@ -154,6 +155,18 @@ func (r *AlertReconciler) Reconcile(ctx context.Context, req ctrl.Request) (resu
r.Metrics.RecordDuration(ctx, alert, start)
}()
if !controllerutil.ContainsFinalizer(alert, v1beta1.NotificationFinalizer) {
controllerutil.AddFinalizer(alert, v1beta1.NotificationFinalizer)
result = ctrl.Result{Requeue: true}
return
}
if !alert.ObjectMeta.DeletionTimestamp.IsZero() {
controllerutil.RemoveFinalizer(alert, v1beta1.NotificationFinalizer)
result = ctrl.Result{}
return
}
return r.reconcile(ctx, alert)
}

13
controllers/provider_controller.go
View File

@ -30,6 +30,7 @@ import (
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
"sigs.k8s.io/controller-runtime/pkg/predicate"
"sigs.k8s.io/controller-runtime/pkg/ratelimiter"
"sigs.k8s.io/yaml"
@ -133,6 +134,18 @@ func (r *ProviderReconciler) Reconcile(ctx context.Context, req ctrl.Request) (r
}()
if !controllerutil.ContainsFinalizer(provider, v1beta1.NotificationFinalizer) {
controllerutil.AddFinalizer(provider, v1beta1.NotificationFinalizer)
result = ctrl.Result{Requeue: true}
return
}
if !provider.ObjectMeta.DeletionTimestamp.IsZero() {
controllerutil.RemoveFinalizer(provider, v1beta1.NotificationFinalizer)
result = ctrl.Result{}
return
}
return r.reconcile(ctx, provider)
}

13
controllers/receiver_controller.go
View File

@ -30,6 +30,7 @@ import (
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/client"
"sigs.k8s.io/controller-runtime/pkg/controller"
"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
"sigs.k8s.io/controller-runtime/pkg/ratelimiter"
"github.com/fluxcd/pkg/apis/meta"
@ -128,6 +129,18 @@ func (r *ReceiverReconciler) Reconcile(ctx context.Context, req ctrl.Request) (r
}()
if !controllerutil.ContainsFinalizer(receiver, v1beta1.NotificationFinalizer) {
controllerutil.AddFinalizer(receiver, v1beta1.NotificationFinalizer)
result = ctrl.Result{Requeue: true}
return
}
if !receiver.ObjectMeta.DeletionTimestamp.IsZero() {
controllerutil.RemoveFinalizer(receiver, v1beta1.NotificationFinalizer)
result = ctrl.Result{}
return
}
return r.reconcile(ctx, receiver)
}