fix error handling in DeletePVCs

To return an error when we can't get a prime PVC rather than continue to the next PVC. Also refactored the code a bit to reduce cognitive complexity. Signed-off-by: Arik Hadas <[email protected]>
kubev2v · Oct 19, 2023 · d2f09cd · d2f09cd
1 parent 12d417b
commit d2f09cd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 15 deletions.
diff --git a/pkg/controller/plan/kubevirt.go b/pkg/controller/plan/kubevirt.go
@@ -937,25 +937,22 @@ func (r *KubeVirt) DeletePVCs(vm *plan.VMStatus) (err error) {
 	for _, pvc := range pvcs {
 		primePVC := core.PersistentVolumeClaim{}
 		err = r.Destination.Client.Get(context.TODO(), client.ObjectKey{Namespace: r.Plan.Spec.TargetNamespace, Name: fmt.Sprintf("prime-%s", string(pvc.UID))}, &primePVC)
-		if err != nil {
-			if k8serr.IsNotFound(err) {
-				err = nil
-			}
-			continue
+		if err != nil && k8serr.IsNotFound(err) {
+			err = nil
+		} else if err != nil {
+			return
 		}
 		// Best effort deletion
 		err = r.DeleteObject(&primePVC, vm, "Deleted prime PVC.", "pvc")
 		if err != nil && k8serr.IsNotFound(err) {
 			err = nil
-		}
-		if err != nil {
+		} else if err != nil {
 			return
 		}
 		err = r.DeleteObject(&pvc, vm, "Deleted PVC.", "pvc")
 		if err != nil && k8serr.IsNotFound(err) {
 			err = nil
-		}
-		if err != nil {
+		} else if err != nil {
 			return
 		}
 		pvcCopy := pvc.DeepCopy()

diff --git a/pkg/controller/plan/migration.go b/pkg/controller/plan/migration.go
@@ -424,7 +424,9 @@ func (r *Migration) Cancel() (err error) {
 }
 
 func (r *Migration) cleanUpPopulatorPVCs(vm *plan.VMStatus) (err error) {
-	err = r.kubevirt.DeletePVCs(vm)
+	if r.builder.SupportsVolumePopulators() {
+		err = r.kubevirt.DeletePVCs(vm)
+	}
 	return
 }
 
@@ -435,11 +437,9 @@ func (r *Migration) cleanup(vm *plan.VMStatus) (err error) {
 		if err != nil {
 			return
 		}
-		if r.builder.SupportsVolumePopulators() {
-			err = r.cleanUpPopulatorPVCs(vm)
-			if err != nil {
-				return
-			}
+		err = r.cleanUpPopulatorPVCs(vm)
+		if err != nil {
+			return
 		}
 	}
 	err = r.deleteImporterPods(vm)