From 3597547ca6f91b58725a0c649e13d38970dc62e5 Mon Sep 17 00:00:00 2001
From: Liran Rotenberg <lrotenbe@redhat.com>
Date: Tue, 12 Sep 2023 18:35:43 +0300
Subject: [PATCH] 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 <lrotenbe@redhat.com>
---
 pkg/controller/plan/kubevirt.go  | 23 +++++++++++++++++++++++
 pkg/controller/plan/migration.go | 11 +++++++++++
 2 files changed, 34 insertions(+)

diff --git a/pkg/controller/plan/kubevirt.go b/pkg/controller/plan/kubevirt.go
index a2129d994..08e466449 100644
--- 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
index 66b771a53..984b90ce3 100644
--- a/pkg/controller/plan/migration.go
+++ b/pkg/controller/plan/migration.go
@@ -414,6 +414,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) {
@@ -421,6 +426,12 @@ 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.deleteImporterPods(vm)
 	if err != nil {