feat(vi): create volume snapshot for vi on the pvc

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

Author: Isteb4k

images/virtualization-artifact/pkg/controller/service/disk_service.go

@@ -238,6 +238,34 @@ func (s DiskService) CheckProvisioning(ctx context.Context, pvc *corev1.Persiste
 	return nil
 }
 
+func (s DiskService) CreateVolumeSnapshot(ctx context.Context, pvc *corev1.PersistentVolumeClaim) error {
+	if pvc == nil || pvc.Status.Phase == corev1.ClaimBound {
+		return errors.New("pvc not Bound")
+	}
+
+	vs := &vsv1.VolumeSnapshot{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      pvc.Name,
+			Namespace: pvc.Namespace,
+			OwnerReferences: []metav1.OwnerReference{
+				MakeOwnerReference(pvc),
+			},
+		},
+		Spec: vsv1.VolumeSnapshotSpec{
+			Source: vsv1.VolumeSnapshotSource{
+				PersistentVolumeClaimName: &pvc.Name,
+			},
+		},
+	}
+
+	err := s.client.Create(ctx, vs)
+	if err != nil && !k8serrors.IsAlreadyExists(err) {
+		return fmt.Errorf("create vs: %w", err)
+	}
+
+	return nil
+}
+
 func (s DiskService) CreatePersistentVolumeClaim(ctx context.Context, pvc *corev1.PersistentVolumeClaim) error {
 	err := s.client.Create(ctx, pvc)
 	if err != nil && !k8serrors.IsAlreadyExists(err) {

images/virtualization-artifact/pkg/controller/vi/internal/source/http.go

@@ -22,6 +22,7 @@ import (
 	"fmt"
 	"time"
 
+	vsv1 "github.com/kubernetes-csi/external-snapshotter/client/v6/apis/volumesnapshot/v1"
 	corev1 "k8s.io/api/core/v1"
 	storagev1 "k8s.io/api/storage/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
@@ -379,6 +380,48 @@ func (ds HTTPDataSource) StoreToPVC(ctx context.Context, vi *v1alpha2.VirtualIma
 			"The HTTP DataSource import has completed",
 		)
 
+		_, exists := vi.Annotations["virtualization.deckhouse.io/use-volume-snapshot"]
+		if exists {
+			var vs *vsv1.VolumeSnapshot
+			vs, err = ds.diskService.GetVolumeSnapshot(ctx, pvc.Name, pvc.Namespace)
+			if err != nil {
+				vi.Status.Phase = v1alpha2.ImageFailed
+				cb.
+					Status(metav1.ConditionFalse).
+					Reason(vicondition.ProvisioningFailed).
+					Message(err.Error())
+				return reconcile.Result{}, nil
+			}
+
+			if vs == nil {
+				err = ds.diskService.CreateVolumeSnapshot(ctx, pvc)
+				if err != nil {
+					vi.Status.Phase = v1alpha2.ImageFailed
+					cb.
+						Status(metav1.ConditionFalse).
+						Reason(vicondition.ProvisioningFailed).
+						Message(err.Error())
+					return reconcile.Result{}, nil
+				}
+
+				vi.Status.Phase = v1alpha2.ImageProvisioning
+				cb.
+					Status(metav1.ConditionFalse).
+					Reason(vicondition.Provisioning).
+					Message("The VolumeSnapshot has been created.")
+				return reconcile.Result{RequeueAfter: time.Second}, nil
+			}
+
+			if vs.Status.ReadyToUse == nil || !(*vs.Status.ReadyToUse) {
+				vi.Status.Phase = v1alpha2.ImageProvisioning
+				cb.
+					Status(metav1.ConditionFalse).
+					Reason(vicondition.Provisioning).
+					Message("Waiting for the VolumeSnapshot to be ready to use.")
+				return reconcile.Result{RequeueAfter: time.Second}, nil
+			}
+		}
+
 		vi.Status.Phase = v1alpha2.ImageReady
 		cb.
 			Status(metav1.ConditionTrue).