MGMT-22728: Add backup label to AgentCluster and AgentMachine (#804)

Author: CrystalChun

controllers/agentcluster_controller.go

@@ -122,6 +122,9 @@ func (r *AgentClusterReconciler) Reconcile(ctx context.Context, req ctrl.Request
 		return ctrl.Result{}, nil
 	}
 
+	// Set backup label on the agentCluster
+	ensureBackupLabel(&agentCluster.ObjectMeta)
+
 	// If the agentCluster has no reference to a ClusterDeployment, find or create one
 	if agentCluster.Status.ClusterDeploymentRef.Name == "" {
 		return r.findOrCreateClusterDeployment(ctx, log, agentCluster)

controllers/agentmachine_controller.go

@@ -134,6 +134,9 @@ func (r *AgentMachineReconciler) Reconcile(ctx context.Context, req ctrl.Request
 		return ctrl.Result{}, nil
 	}
 
+	// Set backup label on the agentMachine
+	ensureBackupLabel(&agentMachine.ObjectMeta)
+
 	machineConfigPool, ignitionTokenSecretRef, ignitionEndpointHTTPHeaders, err := r.processBootstrapDataSecret(ctx, log, machine, agentMachine.Status.Ready)
 	if err != nil {
 		return ctrl.Result{}, err

controllers/common.go

@@ -52,6 +52,15 @@ func ensureSecretLabel(ctx context.Context, c client.Client, secret *corev1.Secr
 	return nil
 }
 
+func ensureBackupLabel(obj *metav1.ObjectMeta) {
+	if obj == nil {
+		return
+	}
+	if !metav1.HasLabel(*obj, BackupLabel) {
+		metav1.SetMetaDataLabel(obj, BackupLabel, BackupLabelValue)
+	}
+}
+
 func GetKubeClientSchemes(schemes *runtime.Scheme) *runtime.Scheme {
 	utilruntime.Must(scheme.AddToScheme(schemes))
 	utilruntime.Must(corev1.AddToScheme(schemes))