provider: add server side implementation of rpc calls

Signed-off-by: Rewant Soni <[email protected]>

controllers/storageconsumer/storageconsumer_controller.go

@@ -51,6 +51,7 @@ const (
 	StorageProfileLabel           = "ocs.openshift.io/storageprofile"
 	ConsumerUUIDLabel             = "ocs.openshift.io/storageconsumer-uuid"
 	StorageConsumerNameLabel      = "ocs.openshift.io/storageconsumer-name"
+	MaintenanceModeLabel          = "ocs.openshift.io/maintenanceMode"
 )
 
 // StorageConsumerReconciler reconciles a StorageConsumer object

services/provider/server/consumer.go

@@ -227,3 +227,11 @@ func (c *ocsConsumerManager) UpdateConsumerStatus(ctx context.Context, id string
 	klog.Infof("successfully updated Status for StorageConsumer %v", consumerObj.Name)
 	return nil
 }
+
+func (c *ocsConsumerManager) Update(ctx context.Context, consumer *ocsv1alpha1.StorageConsumer) error {
+	if err := c.client.Update(ctx, consumer); err != nil {
+		return fmt.Errorf("failed to update StorageConsumer %v: %v", consumer.Name, err)
+	}
+	klog.Infof("successfully updated StorageConsumer %v", consumer.Name)
+	return nil
+}

services/provider/server/server.go

@@ -11,6 +11,7 @@ import (
 	"encoding/json"
 	"encoding/pem"
 	"fmt"
+	appsv1 "k8s.io/api/apps/v1"
 	"math"
 	"net"
 	"slices"
@@ -949,3 +950,54 @@ func extractMonitorIps(data string) ([]string, error) {
 func (s *OCSProviderServer) PeerStorageCluster(_ context.Context, _ *pb.PeerStorageClusterRequest) (*pb.PeerStorageClusterResponse, error) {
 	return &pb.PeerStorageClusterResponse{}, nil
 }
+
+func (s *OCSProviderServer) StartMaintenanceMode(ctx context.Context, req *pb.StartMaintenanceModeRequest) (*pb.StartMaintenanceModeResponse, error) {
+	// Get storage consumer resource using UUID
+	consumer, err := s.consumerManager.Get(ctx, req.StorageConsumerUUID)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+
+	util.AddLabel(consumer, controllers.MaintenanceModeLabel, "")
+	err = s.consumerManager.Update(ctx, consumer)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+
+	return &pb.StartMaintenanceModeResponse{}, nil
+}
+
+func (s *OCSProviderServer) StopMaintenanceMode(ctx context.Context, req *pb.StopMaintenanceModeRequest) (*pb.StopMaintenanceModeResponse, error) {
+	// Get storage consumer resource using UUID
+	consumer, err := s.consumerManager.Get(ctx, req.StorageConsumerUUID)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+
+	//TODO: Remove Label
+
+	if err := s.consumerManager.Update(ctx, consumer); err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+
+	return &pb.StopMaintenanceModeResponse{}, nil
+}
+
+func (s *OCSProviderServer) GetMaintenanceModeStatus(ctx context.Context, req *pb.GetMaintenanceModeStatusRequest) (*pb.GetMaintenanceModeStatusResponse, error) {
+	// Get storage consumer resource using UUID
+	_, err := s.consumerManager.Get(ctx, req.StorageConsumerUUID)
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+
+	rbdMirrorDeployments := &appsv1.DeploymentList{}
+	err = s.client.List(ctx, rbdMirrorDeployments, client.InNamespace(s.namespace), client.MatchingLabels{"app": "rook-ceph-rbd-mirror"}, client.Limit(1))
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, err.Error())
+	}
+	replicas := rbdMirrorDeployments.Items[0].Status.Replicas
+	if replicas == 1 {
+		return &pb.GetMaintenanceModeStatusResponse{MmintenanceStatus: pb.GetMaintenanceModeStatusResponse_Progressing}, nil
+	}
+	return &pb.GetMaintenanceModeStatusResponse{MmintenanceStatus: pb.GetMaintenanceModeStatusResponse_Completed}, nil
+}