Merge pull request #2870 from rewantsoni/mmode-rpc

rpc: add rpc calls to create the maintenance mode reconciler in ocs-operator
red-hat-storage · Nov 20, 2024 · 2e7cf01 · 2e7cf01
2 parents d486fab + fdc4a8f
commit 2e7cf01
Show file tree

Hide file tree

Showing 13 changed files with 1,198 additions and 553 deletions.
diff --git a/controllers/util/k8sutil.go b/controllers/util/k8sutil.go
@@ -54,6 +54,8 @@ const (
 	ExitCodeThatShouldRestartTheProcess = 42
 
 	BlockPoolMirroringTargetIDAnnotation = "ocs.openshift.io/mirroring-target-id"
+	RequestMaintenanceModeAnnotation     = "ocs.openshift.io/request-maintenance-mode"
+	CephRBDMirrorName                    = "cephrbdmirror"
 )
 
 var podNamespace = os.Getenv(PodNamespaceEnvVar)

diff --git a/deploy/ocs-operator/manifests/provider-role.yaml b/deploy/ocs-operator/manifests/provider-role.yaml
@@ -17,6 +17,7 @@ rules:
       - cephfilesystemsubvolumegroups
       - cephblockpoolradosnamespaces
       - cephblockpools
+      - cephrbdmirrors
     verbs:
       - get
       - list

diff --git a/metrics/vendor/github.com/red-hat-storage/ocs-operator/v4/controllers/util/k8sutil.go b/metrics/vendor/github.com/red-hat-storage/ocs-operator/v4/controllers/util/k8sutil.go
diff --git a/rbac/provider-role.yaml b/rbac/provider-role.yaml
@@ -17,6 +17,7 @@ rules:
       - cephfilesystemsubvolumegroups
       - cephblockpoolradosnamespaces
       - cephblockpools
+      - cephrbdmirrors
     verbs:
       - get
       - list

diff --git a/services/provider/api/client/client.go b/services/provider/api/client/client.go
@@ -227,3 +227,19 @@ func (cc *OCSProviderClient) PeerStorageCluster(ctx context.Context, onboardingT
 
 	return cc.Client.PeerStorageCluster(apiCtx, req)
 }
+
+func (cc *OCSProviderClient) RequestMaintenanceMode(ctx context.Context, consumerUUID string, enable bool) (*pb.RequestMaintenanceModeResponse, error) {
+	if cc.Client == nil || cc.clientConn == nil {
+		return nil, fmt.Errorf("provider client is closed")
+	}
+
+	req := &pb.RequestMaintenanceModeRequest{
+		StorageConsumerUUID: consumerUUID,
+		Enable:              enable,
+	}
+
+	apiCtx, cancel := context.WithTimeout(ctx, cc.timeout)
+	defer cancel()
+
+	return cc.Client.RequestMaintenanceMode(apiCtx, req)
+}