Helpers for linked-clone

Author: bn023052

tests/e2e/linked_clone/linked_clone_concurrent_util.go

@@ -0,0 +1,105 @@
+/*
+Copyright 2025 The Kubernetes Authors.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package linked_clone
+
+import (
+	"context"
+	"fmt"
+	"sync"
+
+	snapV1 "github.com/kubernetes-csi/external-snapshotter/client/v8/apis/volumesnapshot/v1"
+	"github.com/onsi/ginkgo/v2"
+	corev1 "k8s.io/api/core/v1"
+	v1 "k8s.io/api/storage/v1"
+	clientset "k8s.io/client-go/kubernetes"
+	fpv "k8s.io/kubernetes/test/e2e/framework/pv"
+	"sigs.k8s.io/vsphere-csi-driver/v3/tests/e2e/config"
+	"sigs.k8s.io/vsphere-csi-driver/v3/tests/e2e/constants"
+)
+
+// This function can be used to create given number of PVC/pod
+// Keeping this method here as it populates the array used for cleanup
+func CreateMultiplePvcPod(ctx context.Context, e2eTestConfig *config.E2eTestConfig, client clientset.Interface, namespace string, storageclass *v1.StorageClass, doCreatePod bool, doCreateDep bool, numOfPVc int) map[*corev1.PersistentVolumeClaim][]*corev1.PersistentVolume {
+
+	volumeMap := map[*corev1.PersistentVolumeClaim][]*corev1.PersistentVolume{}
+
+	fmt.Println("Create PVC and verify PVC is bound")
+	for i := 0; i < numOfPVc; i++ {
+		ginkgo.By(fmt.Sprintf("Creating PVC in iteration: %v",
+			i))
+		pvclaim, pv := CreateAndValidatePvc(ctx, client, namespace, storageclass)
+		volumeMap[pvclaim] = pv
+
+		fmt.Println("Create Pod and attach to PVC")
+		if doCreatePod || doCreateDep {
+			CreatePodForPvc(ctx, e2eTestConfig, client, namespace, []*corev1.PersistentVolumeClaim{pvclaim}, doCreatePod, doCreateDep)
+		}
+	}
+
+	return volumeMap
+}
+
+// This method can be used to create snapshot in parallel
+// Keeping this method here as it populates the array used for cleanup
+func CreateSnapshotInParallel(ctx context.Context, e2eTestConfig *config.E2eTestConfig, namespace string, volumeMap map[*corev1.PersistentVolumeClaim][]*corev1.PersistentVolume) chan *snapV1.VolumeSnapshot {
+
+	var wg sync.WaitGroup
+	snapshots := make(chan *snapV1.VolumeSnapshot, len(volumeMap))
+
+	for pvclaim, pvList := range volumeMap {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			snapshot, _ := CreateVolumeSnapshot(ctx, e2eTestConfig, namespace, pvclaim, pvList, constants.DiskSize)
+			snapshots <- snapshot
+		}()
+	}
+
+	wg.Wait()
+	close(snapshots)
+	fmt.Println("All threads completed")
+	return snapshots
+}
+
+// Expected to create few linked clones before calling this method
+func CreateDeleteLinkedClonesInParallel(ctx context.Context, client clientset.Interface, namespace string, storageclass *v1.StorageClass, snapshot *snapV1.VolumeSnapshot, pvcList []*corev1.PersistentVolumeClaim, iteration int) (chan *corev1.PersistentVolumeClaim, chan []*corev1.PersistentVolume) {
+	var wg sync.WaitGroup
+
+	lcPvcCreated := make(chan *corev1.PersistentVolumeClaim, iteration)
+	lcPvCreated := make(chan []*corev1.PersistentVolume, iteration)
+
+	for i := 0; i < iteration; i++ {
+		fmt.Printf("Iteration %d\n", i)
+
+		wg.Add(2) // We're launching 2 goroutines per iteration
+
+		go func(id int) {
+			defer wg.Done()
+			linkdeClonePvc, lcPv := CreateAndValidateLinkedClone(ctx, client, namespace, storageclass, snapshot.Name)
+			lcPvcCreated <- linkdeClonePvc
+			lcPvCreated <- lcPv
+		}(i)
+
+		go func(id int) {
+			defer wg.Done()
+			fpv.DeletePersistentVolumeClaim(ctx, client, pvcList[i].Name, namespace)
+		}(i)
+	}
+
+	wg.Wait()
+	close(lcPvcCreated)
+	close(lcPvCreated)
+
+	return lcPvcCreated, lcPvCreated
+}