Clean up PVCs when canceling populator flows

This patch will clean up the PVCs (final and prime) associated with VMs that the migration was cancled for them. Signed-off-by: Liran Rotenberg <[email protected]>
kubev2v · Sep 12, 2023 · dd54aa5 · dd54aa5
1 parent ea39abb
commit dd54aa5
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 0 deletions.
diff --git a/pkg/controller/plan/kubevirt.go b/pkg/controller/plan/kubevirt.go
@@ -893,6 +893,29 @@ func (r *KubeVirt) SetPopulatorPodOwnership(vm *plan.VMStatus) (err error) {
 	return
 }
 
+// DeletePVCs deletes PVCs associated with the VM, including prime PVCs (populator flows)
+func (r *KubeVirt) DeletePVCs(vm *plan.VMStatus) (err error) {
+	pvcs, err := r.getPVCs(vm.Ref)
+	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 {
+			continue
+		}
+		// Best effort deletion
+		err = r.DeleteObject(&primePVC, vm, "Deleted prime PVC.", "pvc")
+		if err != nil && k8serr.IsNotFound(err) {
+			err = nil
+		}
+		err = r.DeleteObject(&pvc, vm, "Deleted PVC.", "pvc")
+		pvcCopy := pvc.DeepCopy()
+		pvc.Finalizers = nil
+		patch := client.MergeFrom(pvcCopy)
+		err = r.Destination.Client.Patch(context.TODO(), &pvc, patch)
+	}
+	return
+}
+
 // Delete any populator pods that belong to a VM's migration.
 func (r *KubeVirt) DeletePopulatorPods(vm *plan.VMStatus) (err error) {
 	list, err := r.getPopulatorPods()

diff --git a/pkg/controller/plan/migration.go b/pkg/controller/plan/migration.go
@@ -392,6 +392,16 @@ func (r *Migration) Cancel() (err error) {
 					vm.String())
 				err = nil
 			}
+			if r.builder.SupportsVolumePopulators() {
+				err = r.CleanUpPopulatorPVCs(vm)
+				if err != nil {
+					r.Log.Error(err,
+						"Couldn't clean up after canceled VM migration.",
+						"vm",
+						vm.String())
+					err = nil
+				}
+			}
 			if vm.RestorePowerState == On {
 				err = r.provider.PowerOn(vm.Ref)
 				if err != nil {
@@ -414,6 +424,11 @@ func (r *Migration) Cancel() (err error) {
 	return
 }
 
+func (r *Migration) CleanUpPopulatorPVCs(vm *plan.VMStatus) (err error) {
+	err = r.kubevirt.DeletePVCs(vm)
+	return
+}
+
 // Delete left over migration resources associated with a VM.
 func (r *Migration) CleanUp(vm *plan.VMStatus) (err error) {
 	if !vm.HasCondition(Succeeded) {