add server side implementation

Signed-off-by: Rewant Soni <resoni@redhat.com> add generated changes Signed-off-by: Rewant Soni <resoni@redhat.com>
red-hat-storage · Nov 11, 2024 · be79221 · be79221
1 parent 130ce55
commit be79221
Show file tree

Hide file tree

Showing 3 changed files with 96 additions and 3 deletions.
diff --git a/deploy/ocs-operator/manifests/provider-role.yaml b/deploy/ocs-operator/manifests/provider-role.yaml
@@ -16,6 +16,7 @@ rules:
     resources:
       - cephfilesystemsubvolumegroups
       - cephblockpoolradosnamespaces
+      - cephblockpools
     verbs:
       - get
       - list

diff --git a/rbac/provider-role.yaml b/rbac/provider-role.yaml
@@ -16,6 +16,7 @@ rules:
     resources:
       - cephfilesystemsubvolumegroups
       - cephblockpoolradosnamespaces
+      - cephblockpools
     verbs:
       - get
       - list

diff --git a/services/provider/server/server.go b/services/provider/server/server.go
@@ -58,9 +58,10 @@ const (
 	storageRequestNameLabel   = "ocs.openshift.io/storagerequest-name"
 	notAvailable              = "N/A"
 
-	ramenDRStorageIDKey     = "ramendr.openshift.io/storageID"
-	ramenDRReplicationIDKey = "ramendr.openshift.io/replicationid"
-	ramenDRFlattenModeKey   = "replication.storage.openshift.io/flatten-mode"
+	ramenDRStorageIDKey              = "ramendr.openshift.io/storageID"
+	ramenDRReplicationIDKey          = "ramendr.openshift.io/replicationid"
+	ramenDRFlattenModeKey            = "replication.storage.openshift.io/flatten-mode"
+	rbdMirrorBootstrapPeerSecretName = "rbdMirrorBootstrapPeerSecretName"
 )
 
 const (
@@ -1042,3 +1043,93 @@ func (s *OCSProviderServer) PeerStorageCluster(ctx context.Context, req *pb.Peer
 
 	return &pb.PeerStorageClusterResponse{}, nil
 }
+
+func (s *OCSProviderServer) GetStorageClientsInfo(ctx context.Context, req *pb.StorageClientsInfoRequest) (*pb.StorageClientsInfoResponse, error) {
+	storageCluster, err := util.GetStorageClusterInNamespace(ctx, s.client, s.namespace)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get storageCluster. %v", err)
+	}
+
+	if storageCluster.UID != types.UID(req.StorageClusterUID) {
+		return nil, status.Errorf(codes.InvalidArgument, "storageClusterUID specified on the req does not match any existing storage cluster")
+	}
+
+	var clientsInfo []*pb.ClientInfo
+
+	for i := range req.ClientIDs {
+		consumer, err := s.consumerManager.Get(ctx, req.ClientIDs[i])
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "failed to get consumer. %v", err)
+		}
+		rnsList := &rookCephv1.CephBlockPoolRadosNamespaceList{}
+		err = s.client.List(
+			ctx,
+			rnsList,
+			client.InNamespace(s.namespace),
+			client.MatchingLabels{controllers.StorageConsumerNameLabel: consumer.Name},
+			client.Limit(1),
+		)
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "failed to list rados namespace. %v", err)
+		}
+		if len(rnsList.Items) == 0 {
+			return nil, status.Errorf(codes.Internal, "rados namespace for the client not found: %v", consumer.UID)
+		}
+		clientsInfo = append(clientsInfo, &pb.ClientInfo{ClientID: req.ClientIDs[i], RadosNamespace: rnsList.Items[0].Name})
+	}
+
+	return &pb.StorageClientsInfoResponse{ClientsInfo: clientsInfo}, nil
+}
+
+func (s *OCSProviderServer) GetBlockPoolsInfo(ctx context.Context, req *pb.BlockPoolsInfoRequest) (*pb.BlockPoolsInfoResponse, error) {
+	storageCluster, err := util.GetStorageClusterInNamespace(ctx, s.client, s.namespace)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get storageCluster. %v", err)
+	}
+
+	if storageCluster.UID != types.UID(req.StorageClusterUID) {
+		return nil, status.Errorf(codes.InvalidArgument, "storageClusterUID specified on the req does not match any existing storage cluster")
+	}
+
+	var blockPoolsInfo []*pb.BlockPoolInfo
+	for i := range req.BlockPoolNames {
+		cephBlockPool := &rookCephv1.CephBlockPool{}
+		cephBlockPool.Name = req.BlockPoolNames[i]
+		cephBlockPool.Namespace = s.namespace
+		err = s.client.Get(ctx, client.ObjectKeyFromObject(cephBlockPool), cephBlockPool)
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "failed to get CephBlockPool: %v. %v", cephBlockPool.Name, err)
+		}
+
+		var mirroringToken string
+
+		if cephBlockPool.Spec.Mirroring.Enabled &&
+			cephBlockPool.Status.Info != nil &&
+			cephBlockPool.Status.Info[rbdMirrorBootstrapPeerSecretName] != "" {
+			secret := &corev1.Secret{}
+			secret.Name = cephBlockPool.Status.Info[rbdMirrorBootstrapPeerSecretName]
+			secret.Namespace = s.namespace
+			err := s.client.Get(ctx, client.ObjectKeyFromObject(secret), secret)
+			if err != nil {
+				errMsg := fmt.Sprintf(
+					"failed to get bootstrap secret %s for CephBlockPool %s: %v",
+					cephBlockPool.Status.Info[rbdMirrorBootstrapPeerSecretName],
+					cephBlockPool.Name,
+					err,
+				)
+				klog.Error(errMsg)
+				return nil, status.Errorf(codes.Internal, "failed to get bootstrap secret for CephBlockPool %s.", cephBlockPool.Name)
+			}
+			mirroringToken = string(secret.Data["token"])
+		}
+
+		blockPoolsInfo = append(blockPoolsInfo, &pb.BlockPoolInfo{
+			BlockPoolName:  cephBlockPool.Name,
+			MirroringToken: mirroringToken,
+			BlockPoolID:    strconv.Itoa(cephBlockPool.Status.PoolID),
+		})
+
+	}
+
+	return &pb.BlockPoolsInfoResponse{BlockPoolsInfo: blockPoolsInfo}, nil
+}