From 904a5ee140659d69ed654d9911dc7418e0fb85de Mon Sep 17 00:00:00 2001
From: Rewant Soni <resoni@redhat.com>
Date: Mon, 10 Feb 2025 22:07:12 +0530
Subject: [PATCH] controllers: update maintenance mode controller

update the maintenance mode controller to put all the storage
clients in maintenance mode when a request is received

Signed-off-by: Rewant Soni <resoni@redhat.com>
---
 .../controller/maintenancemode_controller.go  | 50 +++++++++----------
 1 file changed, 23 insertions(+), 27 deletions(-)

diff --git a/internal/controller/maintenancemode_controller.go b/internal/controller/maintenancemode_controller.go
index 1c03d125..f38db22c 100644
--- a/internal/controller/maintenancemode_controller.go
+++ b/internal/controller/maintenancemode_controller.go
@@ -77,7 +77,10 @@ func (r *MaintenanceModeReconciler) Reconcile(ctx context.Context, _ ctrl.Reques
 	r.log = log.FromContext(ctx)
 	r.log.Info("Starting reconcile")
 
-	nameToStorageClient := map[string]*v1alpha1.StorageClient{}
+	//TODO: This controller assumes that there is only one client and that the maintenance mode is created only for that
+	// client. The correct way to fix this would be to use the mmode's targetID (which is the replicationID) and to
+	// fetch the corresponding VRC as the VRC is labeled with targetID and then fetch the StorageClient and put it to
+	// maintenanceMode
 
 	maintenanceModes := &ramenv1alpha1.MaintenanceModeList{}
 	if err := r.list(maintenanceModes); err != nil {
@@ -85,42 +88,35 @@ func (r *MaintenanceModeReconciler) Reconcile(ctx context.Context, _ ctrl.Reques
 		return reconcile.Result{}, err
 	}
 
+	storageClients := &v1alpha1.StorageClientList{}
+	if err := r.list(storageClients); err != nil {
+		r.log.Error(err, "failed to list the Storage Clients")
+		return reconcile.Result{}, err
+	}
+
+	if len(storageClients.Items) != 1 {
+		r.log.Error(
+			fmt.Errorf("expected 1 StorageClient found %d", len(storageClients.Items)),
+			"invalid number of StorageClients found",
+		)
+		return reconcile.Result{}, fmt.Errorf("expected 1 StorageClient found %d", len(storageClients.Items))
+	}
+
+	needsMaintenanceMode := false
+
 	for i := range maintenanceModes.Items {
 		mm := &maintenanceModes.Items[i]
-		sc := &v1alpha1.StorageClaim{}
-		// MMode's TargetID is replicationID, which in our case is storageClaim name
-		sc.Name = mm.Spec.TargetID
-		if err := r.get(sc); err != nil {
-			return ctrl.Result{}, err
-		}
-		clientName := sc.Spec.StorageClient
-		if clientName == "" {
-			return ctrl.Result{}, fmt.Errorf("StorageClaim %s does not have a StorageClient defined", sc.Name)
-		}
-		if nameToStorageClient[clientName] == nil {
-			storageClient := &v1alpha1.StorageClient{}
-			storageClient.Name = clientName
-			if err := r.get(storageClient); err != nil {
-				return ctrl.Result{}, err
-			}
-			nameToStorageClient[clientName] = storageClient
-		}
-		if nameToStorageClient[clientName].Status.InMaintenanceMode {
+		storageClient := &storageClients.Items[0]
+		if storageClient.Status.InMaintenanceMode {
+			needsMaintenanceMode = true
 			if err := r.updateStatusCompletedForMM(mm); err != nil {
 				return ctrl.Result{}, fmt.Errorf("failed to update status for MaintenanceMode %s: %w", mm.Name, err)
 			}
 		}
 	}
 
-	storageClients := &v1alpha1.StorageClientList{}
-	if err := r.list(storageClients); err != nil {
-		r.log.Error(err, "failed to list the Storage Clients")
-		return reconcile.Result{}, err
-	}
-
 	for i := range storageClients.Items {
 		storageClient := &storageClients.Items[i]
-		_, needsMaintenanceMode := nameToStorageClient[storageClient.Name]
 		if needsMaintenanceMode != storageClient.Status.InMaintenanceMode {
 			if err := r.toggleMaintenanceModeForClient(storageClient, needsMaintenanceMode); err != nil {
 				return ctrl.Result{}, err