feat: Improve subnet and VPC finalizer handling

- Add finalizer to subnet and VPC resources if not present
- Remove VPC finalizer when no subnets exist
- Enhance VPC status management by adding subnets to queue when empty
- Use slices.Contains for more idiomatic finalizer checks

Signed-off-by: Mengxin Liu <[email protected]>

pkg/controller/subnet.go

@@ -361,7 +361,7 @@ func (c *Controller) syncSubnetFinalizer(cl client.Client) error {
 }
 
 func (c *Controller) handleSubnetFinalizer(subnet *kubeovnv1.Subnet) (*kubeovnv1.Subnet, bool, error) {
-	if subnet.DeletionTimestamp.IsZero() && len(subnet.GetFinalizers()) == 0 {
+	if subnet.DeletionTimestamp.IsZero() && !slices.Contains(subnet.GetFinalizers(), util.KubeOVNControllerFinalizer) {
 		newSubnet := subnet.DeepCopy()
 		controllerutil.AddFinalizer(newSubnet, util.KubeOVNControllerFinalizer)
 		patch, err := util.GenerateMergePatchPayload(subnet, newSubnet)

pkg/controller/vpc.go

@@ -21,6 +21,7 @@ import (
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/client-go/tools/cache"
 	"k8s.io/klog/v2"
+	"sigs.k8s.io/controller-runtime/pkg/controller/controllerutil"
 
 	kubeovnv1 "github.com/kubeovn/kube-ovn/pkg/apis/kubeovn/v1"
 	"github.com/kubeovn/kube-ovn/pkg/ovsdb/ovnnb"
@@ -156,6 +157,7 @@ func (c *Controller) handleUpdateVpcStatus(key string) error {
 
 	vpc.Status.DefaultLogicalSwitch = defaultSubnet
 	vpc.Status.Subnets = subnets
+
 	if !vpc.Spec.BFDPort.IsEnabled() && !vpc.Status.BFDPort.IsEmpty() {
 		vpc.Status.BFDPort.Clear()
 	}
@@ -170,6 +172,12 @@ func (c *Controller) handleUpdateVpcStatus(key string) error {
 		klog.Error(err)
 		return err
 	}
+
+	if len(vpc.Status.Subnets) == 0 {
+		klog.Infof("vpc %s has no subnets, add to queue", vpc.Name)
+		c.addOrUpdateVpcQueue.Add(vpc.Name)
+	}
+
 	if change {
 		for _, ns := range vpc.Spec.Namespaces {
 			c.addNamespaceQueue.Add(ns)
@@ -262,6 +270,7 @@ func (c *Controller) handleAddOrUpdateVpc(key string) error {
 		klog.Errorf("failed to format vpc %s: %v", key, err)
 		return err
 	}
+
 	if err = c.createVpcRouter(key); err != nil {
 		klog.Errorf("failed to create vpc router for vpc %s: %v", key, err)
 		return err
@@ -1094,6 +1103,16 @@ func (c *Controller) formatVpc(vpc *kubeovnv1.Vpc) (*kubeovnv1.Vpc, error) {
 		}
 	}
 
+	if vpc.DeletionTimestamp.IsZero() && !slices.Contains(vpc.GetFinalizers(), util.KubeOVNControllerFinalizer) {
+		controllerutil.AddFinalizer(vpc, util.KubeOVNControllerFinalizer)
+		changed = true
+	}
+
+	if !vpc.DeletionTimestamp.IsZero() && len(vpc.Status.Subnets) == 0 {
+		controllerutil.RemoveFinalizer(vpc, util.KubeOVNControllerFinalizer)
+		changed = true
+	}
+
 	if changed {
 		newVpc, err := c.config.KubeOvnClient.KubeovnV1().Vpcs().Update(context.Background(), vpc, metav1.UpdateOptions{})
 		if err != nil {

test/e2e/kube-ovn/pod/vpc_pod_probe.go

@@ -30,7 +30,7 @@ var _ = framework.SerialDescribe("[group:pod]", func() {
 	var eventClient *framework.EventClient
 	var subnetClient *framework.SubnetClient
 	var vpcClient *framework.VpcClient
-	var namespaceName, subnetName, podName, vpcName string
+	var namespaceName, subnetName, podName, vpcName, custVPCSubnetName string
 	var subnet *apiv1.Subnet
 	var cidr string
 
@@ -43,6 +43,7 @@ var _ = framework.SerialDescribe("[group:pod]", func() {
 		subnetName = "subnet-" + framework.RandomSuffix()
 		podName = "pod-" + framework.RandomSuffix()
 		cidr = framework.RandomCIDR(f.ClusterIPFamily)
+		custVPCSubnetName = "subnet-" + framework.RandomSuffix()
 
 		ginkgo.By("Creating subnet " + subnetName)
 		subnet = framework.MakeSubnet(subnetName, "", cidr, "", "", "", nil, nil, []string{namespaceName})
@@ -54,6 +55,7 @@ var _ = framework.SerialDescribe("[group:pod]", func() {
 
 		ginkgo.By("Deleting subnet " + subnetName)
 		subnetClient.DeleteSync(subnetName)
+		subnetClient.DeleteSync(custVPCSubnetName)
 
 		ginkgo.By("Deleting VPC " + vpcName)
 		vpcClient.DeleteSync(vpcName)
@@ -78,12 +80,6 @@ var _ = framework.SerialDescribe("[group:pod]", func() {
 		modifyDs.Spec.Template.Spec.Containers[0].Args = newArgs
 		daemonSetClient.PatchSync(modifyDs)
 
-		custVPCSubnetName := "subnet-" + framework.RandomSuffix()
-		ginkgo.DeferCleanup(func() {
-			ginkgo.By("Deleting subnet " + custVPCSubnetName)
-			subnetClient.DeleteSync(custVPCSubnetName)
-		})
-
 		ginkgo.By("Creating VPC " + vpcName)
 		vpcName = "vpc-" + framework.RandomSuffix()
 		customVPC := framework.MakeVpc(vpcName, "", false, false, nil)