Implement deletion lifecycle hooks

Author: michaelgugino

api/v1alpha3/machine_types.go

@@ -37,6 +37,16 @@ const (
 
 	// MachineDeploymentLabelName is the label set on machines if they're controlled by MachineDeployment
 	MachineDeploymentLabelName = "cluster.x-k8s.io/deployment-name"
+
+	// PreDrainDeleteHookAnnotionPrefix annotation specifies the prefix we
+	// search each annotation for during the pre-drain.delete lifecycle hook
+	// to pause reconciliation.
+	PreDrainDeleteHookAnnotionPrefix = "pre-drain.delete.hook.machine.cluster.x-k8s.io"
+
+	// PreTermDeleteHookAnnotionPrefix annotation specifies the prefix we
+	// search each annotation for during the pre-drain.delete lifecycle hook
+	// to pause reconciliation.
+	PreTermDeleteHookAnnotionPrefix = "pre-term.delete.hook.machine.cluster.x-k8s.io"
 )
 
 // ANCHOR: MachineSpec

controllers/machine_controller.go

@@ -19,6 +19,7 @@ package controllers
 import (
 	"context"
 	"fmt"
+	"strings"
 	"time"
 
 	"github.com/go-logr/logr"
@@ -290,6 +291,11 @@ func (r *MachineReconciler) reconcileDelete(ctx context.Context, cluster *cluste
 	}
 
 	if isDeleteNodeAllowed {
+		 // pre-drain.delete lifecycle hook
+		// Return early without error, will requeue if/when the Hook is removed.
+		if waitForHook(clusterv1.PreDrainDeleteHookAnnotionPrefix, m.ObjectMeta.Annotations) {
+			return ctrl.Result{}, nil
+		}
 		// Drain node before deletion.
 		if _, exists := m.ObjectMeta.Annotations[clusterv1.ExcludeNodeDrainingAnnotation]; !exists {
 			logger.Info("Draining node", "node", m.Status.NodeRef.Name)
@@ -301,6 +307,12 @@ func (r *MachineReconciler) reconcileDelete(ctx context.Context, cluster *cluste
 		}
 	}
 
+	// pre-term.delete lifecycle hook
+   // Return early without error, will requeue if/when the Hook is removed.
+   if waitForHook(clusterv1.PreTermDeleteHookAnnotionPrefix, m.ObjectMeta.Annotations) {
+	   return ctrl.Result{}, nil
+   }
+
 	if ok, err := r.reconcileDeleteExternal(ctx, m); !ok || err != nil {
 		// Return early and don't remove the finalizer if we got an error or
 		// the external reconciliation deletion isn't ready.
@@ -491,6 +503,14 @@ func (r *MachineReconciler) shouldAdopt(m *clusterv1.Machine) bool {
 	return metav1.GetControllerOf(m) == nil && !util.HasOwner(m.OwnerReferences, clusterv1.GroupVersion.String(), []string{"Cluster"})
 }
 
+func waitForHook(hookName string, annotations map[string]string) bool {
+	for key := range annotations {
+		strings.HasPrefix(key, hookName)
+		return true
+	}
+	return false
+}
+
 // writer implements io.Writer interface as a pass-through for klog.
 type writer struct {
 	logFunc func(args ...interface{})