controllers: Generates VGSClass

Signed-off-by: raaizik <[email protected]>

controllers/storageclaim_controller.go

@@ -32,6 +32,7 @@ import (
 	"github.com/red-hat-storage/ocs-client-operator/pkg/utils"
 
 	"github.com/go-logr/logr"
+	groupsnapapi "github.com/kubernetes-csi/external-snapshotter/client/v6/apis/volumegroupsnapshot/v1alpha1"
 	snapapi "github.com/kubernetes-csi/external-snapshotter/client/v6/apis/volumesnapshot/v1"
 	providerclient "github.com/red-hat-storage/ocs-operator/v4/services/provider/client"
 	corev1 "k8s.io/api/core/v1"
@@ -110,6 +111,7 @@ func (r *StorageClaimReconciler) SetupWithManager(mgr ctrl.Manager) error {
 		For(&v1alpha1.StorageClaim{}, builder.WithPredicates(predicate.GenerationChangedPredicate{})).
 		Owns(&storagev1.StorageClass{}).
 		Owns(&snapapi.VolumeSnapshotClass{}).
+		Owns(&groupsnapapi.VolumeGroupSnapshotClass{}).
 		Complete(r)
 }
 
@@ -118,7 +120,7 @@ func (r *StorageClaimReconciler) SetupWithManager(mgr ctrl.Manager) error {
 //+kubebuilder:rbac:groups=ocs.openshift.io,resources=storageclaims/finalizers,verbs=update
 //+kubebuilder:rbac:groups="",resources=secrets,verbs=get;list;watch;create;update;delete
 //+kubebuilder:rbac:groups=storage.k8s.io,resources=storageclasses,verbs=get;list;watch;create;update;patch;delete
-//+kubebuilder:rbac:groups=snapshot.storage.k8s.io,resources=volumesnapshotclasses,verbs=get;list;watch;create;delete
+//+kubebuilder:rbac:groups=groupsnapshot.storage.k8s.io,resources=volumegroupsnapshotclasses,verbs=get;list;watch;create;delete
 //+kubebuilder:rbac:groups=core,resources=persistentvolumes,verbs=get;list;watch
 //+kubebuilder:rbac:groups=snapshot.storage.k8s.io,resources=volumesnapshotcontents,verbs=get;list;watch
 
@@ -407,6 +409,22 @@ func (r *StorageClaimReconciler) reconcilePhases() (reconcile.Result, error) {
 				if err := r.createOrReplaceVolumeSnapshotClass(volumeSnapshotClass); err != nil {
 					return reconcile.Result{}, fmt.Errorf("failed to create or update VolumeSnapshotClass: %s", err)
 				}
+			case "VolumeGroupSnapshotClass":
+				var volumeGroupSnapshotClass *groupsnapapi.VolumeGroupSnapshotClass
+				data["csi.storage.k8s.io/snapshotter-secret-namespace"] = r.OperatorNamespace
+				// generate a new clusterID for cephfs subvolumegroup, as
+				// storageclaim is clusterscoped resources using its
+				// hash as the clusterID
+				data["clusterID"] = r.storageClaimHash
+				if resource.Name == "cephfs" {
+					volumeGroupSnapshotClass = r.getCephFSVolumeGroupSnapshotClass(data)
+				} else if resource.Name == "ceph-rbd" {
+					volumeGroupSnapshotClass = r.getCephRBDVolumeGroupSnapshotClass(data)
+				}
+				utils.AddAnnotation(volumeGroupSnapshotClass, storageClaimAnnotation, r.storageClaim.Name)
+				if err := r.createOrReplaceVolumeGroupSnapshotClass(volumeGroupSnapshotClass); err != nil {
+					return reconcile.Result{}, fmt.Errorf("failed to create or update volumeGroupSnapshot: %s", err)
+				}
 			}
 		}
 
@@ -651,6 +669,66 @@ func (r *StorageClaimReconciler) hasVolumeSnapshotContents() (bool, error) {
 	return false, nil
 }
 
+func (r *StorageClaimReconciler) getCephFSVolumeGroupSnapshotClass(data map[string]string) *groupsnapapi.VolumeGroupSnapshotClass {
+	volumgroupesnapshotclass := &groupsnapapi.VolumeGroupSnapshotClass{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: r.storageClaim.Name,
+		},
+		Driver:         csi.GetCephFSDriverName(),
+		DeletionPolicy: snapapi.VolumeSnapshotContentDelete,
+		Parameters:     data,
+	}
+	return volumgroupesnapshotclass
+}
+
+func (r *StorageClaimReconciler) getCephRBDVolumeGroupSnapshotClass(data map[string]string) *groupsnapapi.VolumeGroupSnapshotClass {
+	volumegroupsnapshotclass := &groupsnapapi.VolumeGroupSnapshotClass{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: r.storageClaim.Name,
+		},
+		Driver:         csi.GetRBDDriverName(),
+		DeletionPolicy: snapapi.VolumeSnapshotContentDelete,
+		Parameters:     data,
+	}
+	return volumegroupsnapshotclass
+}
+
+func (r *StorageClaimReconciler) createOrReplaceVolumeGroupSnapshotClass(volumeGroupSnapshotClass *groupsnapapi.VolumeGroupSnapshotClass) error {
+	existing := &groupsnapapi.VolumeGroupSnapshotClass{}
+	existing.Name = r.storageClaim.Name
+
+	if err := r.own(volumeGroupSnapshotClass); err != nil {
+		return fmt.Errorf("failed to own volumegroupsnapshotclass: %v", err)
+	}
+
+	if err := r.get(existing); err != nil && !errors.IsNotFound(err) {
+		return fmt.Errorf("failed to get VolumeGroupSnapshotClass: %v", err)
+	}
+
+	// If present then compare the existing VolumeGroupSnapshotClass parameters with
+	// the received VolumeGroupSnapshotClass parameters, and only proceed if they differ.
+	if reflect.DeepEqual(existing.Parameters, volumeGroupSnapshotClass.Parameters) {
+		return nil
+	}
+
+	// VolumeGroupSnapshotClass already exists, but parameters have changed. Delete the existing VolumeGroupSnapshotClass and create a new one.
+	if existing.UID != "" {
+		// Since we have to update the existing VolumeGroupSnapshotClass, so we will delete the existing VolumeGroupSnapshotClass and create a new one.
+		r.log.Info("VolumeGroupSnapshotClass needs to be updated, deleting it.", "Name", existing.Name)
+
+		// Delete the VolumeGroupSnapshotClass.
+		if err := r.delete(existing); err != nil {
+			r.log.Error(err, "Failed to delete VolumeGroupSnapshotClass.", "Name", existing.Name)
+			return fmt.Errorf("failed to delete VolumeGroupSnapshotClass: %v", err)
+		}
+	}
+	r.log.Info("Creating VolumeGroupSnapshotClass.", "Name", existing.Name)
+	if err := r.Client.Create(r.ctx, volumeGroupSnapshotClass); err != nil {
+		return fmt.Errorf("failed to create VolumeGroupSnapshotClass: %v", err)
+	}
+	return nil
+}
+
 func getMD5Hash(text string) string {
 	hash := md5.Sum([]byte(text))
 	return hex.EncodeToString(hash[:])