provider: enable csiomap generator for client

send a configMap from provider to client to enable csi-omap generator required for DR on client cluster Signed-off-by: Rewant Soni <[email protected]>
red-hat-storage · Nov 8, 2024 · bd2c11d · bd2c11d
1 parent 70fe2f2
commit bd2c11d
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 1 deletion.
diff --git a/controllers/storageclusterpeer/storageclusterpeer_controller.go b/controllers/storageclusterpeer/storageclusterpeer_controller.go
@@ -27,6 +27,10 @@ import (
 	v1 "github.com/red-hat-storage/ocs-operator/api/v4/v1"
 )
 
+const (
+	ClientMappingConfigMapName = "client-mapping-config"
+)
+
 // StorageClusterPeerReconciler reconciles a StorageClusterPeer object
 // nolint:revive
 type StorageClusterPeerReconciler struct {

diff --git a/services/provider/server/server.go b/services/provider/server/server.go
@@ -11,6 +11,7 @@ import (
 	"encoding/json"
 	"encoding/pem"
 	"fmt"
+	"github.com/red-hat-storage/ocs-operator/v4/controllers/storageclusterpeer"
 	"math"
 	"net"
 	"slices"
@@ -200,10 +201,17 @@ func (s *OCSProviderServer) GetStorageConfig(ctx context.Context, req *pb.Storag
 			return nil, err
 		}
 
+		clientMappingConfig, err := s.getClientMapping(ctx)
+		if err != nil {
+			return nil, err
+		}
+		_, hasClientId := clientMappingConfig.Data[req.StorageConsumerUUID]
+
 		desiredClientConfigHash := getDesiredClientConfigHash(
 			channelName,
 			consumerObj,
 			isEncryptionInTransitEnabled(storageCluster.Spec.Network),
+			hasClientId,
 		)
 
 		klog.Infof("successfully returned the config details to the consumer.")
@@ -404,6 +412,21 @@ func (s *OCSProviderServer) getExternalResources(ctx context.Context, consumerRe
 
 	}
 
+	//Enable CSI-OMAP is storageClusterPeer is present
+	clientMappingConfig, err := s.getClientMapping(ctx)
+	if err != nil {
+		return nil, err
+	}
+	_, hasClientId := clientMappingConfig.Data[string(consumerResource.UID)]
+
+	extR = append(extR, &pb.ExternalResource{
+		Name: "ocs-ceph-csi-config",
+		Kind: "ConfigMap",
+		Data: mustMarshal(map[string]string{
+			"CSI_ENABLE_OMAP_GENERATOR": strconv.FormatBool(hasClientId),
+		}),
+	})
+
 	// Fetch noobaa remote secret and management address and append to extResources
 	consumerName := consumerResource.Name
 	noobaaOperatorSecret := &v1.Secret{}
@@ -860,10 +883,17 @@ func (s *OCSProviderServer) ReportStatus(ctx context.Context, req *pb.ReportStat
 		return nil, err
 	}
 
+	clientMappingConfig, err := s.getClientMapping(ctx)
+	if err != nil {
+		return nil, err
+	}
+	_, hasClientId := clientMappingConfig.Data[req.StorageConsumerUUID]
+
 	desiredClientConfigHash := getDesiredClientConfigHash(
 		channelName,
 		storageConsumer,
 		isEncryptionInTransitEnabled(storageCluster.Spec.Network),
+		hasClientId,
 	)
 
 	return &pb.ReportStatusResponse{
@@ -872,11 +902,12 @@ func (s *OCSProviderServer) ReportStatus(ctx context.Context, req *pb.ReportStat
 	}, nil
 }
 
-func getDesiredClientConfigHash(channelName string, storageConsumer *ocsv1alpha1.StorageConsumer, encryptionInTransit bool) string {
+func getDesiredClientConfigHash(channelName string, storageConsumer *ocsv1alpha1.StorageConsumer, encryptionInTransit bool, enableOmap bool) string {
 	var arr = []any{
 		channelName,
 		storageConsumer.Spec.StorageQuotaInGiB,
 		encryptionInTransit,
+		enableOmap,
 	}
 	return util.CalculateMD5Hash(arr)
 }
@@ -924,6 +955,17 @@ func (s *OCSProviderServer) getStorageCluster(ctx context.Context) (*ocsv1.Stora
 	return foundSc, nil
 }
 
+func (s *OCSProviderServer) getClientMapping(ctx context.Context) (*corev1.ConfigMap, error) {
+	clientMappingConfig := &corev1.ConfigMap{}
+	clientMappingConfig.Name = storageclusterpeer.ClientMappingConfigMapName
+	clientMappingConfig.Namespace = s.namespace
+	err := s.client.Get(ctx, client.ObjectKeyFromObject(clientMappingConfig), clientMappingConfig)
+	if err != nil {
+		return nil, fmt.Errorf("failed to list storage cluster peers. %v", err)
+	}
+	return clientMappingConfig, nil
+}
+
 func isEncryptionInTransitEnabled(networkSpec *rookCephv1.NetworkSpec) bool {
 	return networkSpec != nil &&
 		networkSpec.Connections != nil &&

diff --git a/services/provider/server/server_test.go b/services/provider/server/server_test.go
@@ -95,6 +95,13 @@ var mockExtR = map[string]*externalResource{
 			"QuotaForConsumer": fmt.Sprintf("%+v\n", clusterResourceQuotaSpec),
 		},
 	},
+	"ocs-ceph-csi-config": {
+		Name: "ocs-ceph-csi-config",
+		Kind: "ConfigMap",
+		Data: map[string]string{
+			"CSI_ENABLE_OMAP_GENERATOR": "false",
+		},
+	},
 	"noobaa-remote-join-secret": {
 		Name: "noobaa-remote-join-secret",
 		Kind: "Secret",