Metro volume expand support

pkg/controller/controller.go

@@ -1286,9 +1286,26 @@
 	}, nil
 }
 
+func pauseSessionIfMetro(isMetroSession bool, powerStoreClient gopowerstore.Client, ctx context.Context, id string) (gopowerstore.EmptyResponse, error) {
+	if isMetroSession {
+		var params *gopowerstore.FailoverParams
+		return powerStoreClient.ExecuteActionOnReplicationSession(ctx, id, "pause", params)
+	}
+
+	return "", nil
+}
+
+func resumeSessionIfMetro(isMetroSession bool, powerStoreClient gopowerstore.Client, ctx context.Context, id string) (gopowerstore.EmptyResponse, error) {
+	if isMetroSession {
+		var params *gopowerstore.FailoverParams
+		powerStoreClient.ExecuteActionOnReplicationSession(ctx, id, "resume", params)
+	}
+	return "", nil
+}
+
 // ControllerExpandVolume resizes Volume or FileSystem by increasing available volume capacity in the storage array.
 func (s *Service) ControllerExpandVolume(ctx context.Context, req *csi.ControllerExpandVolumeRequest) (*csi.ControllerExpandVolumeResponse, error) {
-	id, arrayID, protocol, _, _, err := array.ParseVolumeID(ctx, req.VolumeId, s.DefaultArray(), nil)
+	id, arrayID, protocol, _, remoteArrayID, err := array.ParseVolumeID(ctx, req.VolumeId, s.DefaultArray(), nil)
 	if err != nil {
 		return nil, status.Errorf(codes.OutOfRange, "unable to parse the volume id")
 	}
@@ -1298,24 +1315,58 @@
 		return nil, status.Errorf(codes.OutOfRange, "volume exceeds allowed limit")
 	}
 
+	powerStoreClient := s.Arrays()[arrayID].Client
+
 	if protocol == "scsi" {
-		vol, err := s.Arrays()[arrayID].Client.GetVolume(ctx, id)
+		vol, err := powerStoreClient.GetVolume(ctx, id)
 		if err != nil {
 			return nil, status.Errorf(codes.OutOfRange, "detected SCSI protocol but wasn't able to fetch the volume info")
 		}
+		isMetroSession := remoteArrayID != ""
+
+		// get session id
+		replicationSessionId := ""
+		if isMetroSession {
+			vgs, err := powerStoreClient.GetVolumeGroupsByVolumeID(ctx, req.GetVolumeId())
+			if err != nil {
+				return nil, err
+			}
+			if len(vgs.VolumeGroup) == 0 {
+				replicationSessionId = vol.MetroReplicationSessionID
+			} else {
+				vg := vgs.VolumeGroup[0]
+				rs, err := powerStoreClient.GetReplicationSessionByLocalResourceID(ctx, vg.ID)
+				if err != nil {
+					return nil, err
+				}
+				replicationSessionId = rs.ID
+			}
+		}
+
 		if vol.Size < requiredBytes {
-			_, err = s.Arrays()[arrayID].Client.ModifyVolume(context.Background(), &gopowerstore.VolumeModify{Size: requiredBytes}, id)
+			_, err = pauseSessionIfMetro(isMetroSession, powerStoreClient, ctx, replicationSessionId)
+			if err != nil {
+				return nil, err
+			}
+
+			_, err = powerStoreClient.ModifyVolume(context.Background(), &gopowerstore.VolumeModify{Size: requiredBytes}, id)
+
+			_, resumeSessionErr := resumeSessionIfMetro(isMetroSession, powerStoreClient, ctx, replicationSessionId)
+			if resumeSessionErr != nil {
+				return nil, resumeSessionErr
+			}
+
 			if err != nil {
 				return nil, err
 			}
 			return &csi.ControllerExpandVolumeResponse{CapacityBytes: requiredBytes, NodeExpansionRequired: true}, nil
 		}
 		return &csi.ControllerExpandVolumeResponse{}, nil
 	}
-	fs, err := s.Arrays()[arrayID].Client.GetFS(ctx, id)
+	fs, err := powerStoreClient.GetFS(ctx, id)
 	if err == nil {
 		if fs.SizeTotal < requiredBytes {
-			_, err = s.Arrays()[arrayID].Client.ModifyFS(context.Background(), &gopowerstore.FSModify{Size: int(requiredBytes + ReservedSize)}, id)
+			_, err = powerStoreClient.ModifyFS(context.Background(), &gopowerstore.FSModify{Size: int(requiredBytes + ReservedSize)}, id)
 			if err != nil {
 				return nil, err
 			}