Ensure IPAddress has a ClusterName label as CAPI resources

Signed-off-by: Gong Zhang <[email protected]>

Author: zhanggbj

internal/controllers/ipaddressclaim_test.go

@@ -91,7 +91,10 @@ var _ = Describe("IPAddressClaimReconciler", func() {
 		})
 
 		When("the referenced namespaced pool exists", func() {
-			const poolName = "test-pool"
+			const (
+				clusterName = "test-cluster"
+				poolName    = "test-pool"
+			)
 
 			BeforeEach(func() {
 				pool := v1alpha2.InClusterIPPool{
@@ -116,6 +119,7 @@ var _ = Describe("IPAddressClaimReconciler", func() {
 
 			It("should allocate an Address from the Pool", func() {
 				claim := newClaim("test", namespace, "InClusterIPPool", poolName)
+				claim.Labels = map[string]string{clusterv1.ClusterNameLabel: clusterName}
 				expectedIPAddress := ipamv1.IPAddress{
 					ObjectMeta: metav1.ObjectMeta{
 						Name:       "test",
@@ -137,6 +141,7 @@ var _ = Describe("IPAddressClaimReconciler", func() {
 								Name:               poolName,
 							},
 						},
+						Labels: map[string]string{clusterv1.ClusterNameLabel: clusterName},
 					},
 					Spec: ipamv1.IPAddressSpec{
 						ClaimRef: corev1.LocalObjectReference{

pkg/ipamutil/reconciler.go

@@ -198,7 +198,7 @@ func (r *ClaimReconciler) Reconcile(ctx context.Context, req ctrl.Request) (_ ct
 	// The handler will complete it with the ip address.
 	address := NewIPAddress(claim, pool)
 
-	// Patch or create the address, ensuring necessary owner references are set.
+	// Patch or create the address, ensuring necessary owner references and labels are set
 	operationResult, err := controllerutil.CreateOrPatch(ctx, r.Client, &address, func() error {
 		if res, err = handler.EnsureAddress(ctx, &address); err != nil {
 			return err
@@ -208,6 +208,13 @@ func (r *ClaimReconciler) Reconcile(ctx context.Context, req ctrl.Request) (_ ct
 			return errors.Wrap(err, "failed to ensure owner references on address")
 		}
 
+		if val, ok := claim.Labels[clusterv1.ClusterNameLabel]; ok {
+			if address.Labels == nil {
+				address.Labels = make(map[string]string)
+			}
+			address.Labels[clusterv1.ClusterNameLabel] = val
+		}
+
 		_ = controllerutil.AddFinalizer(&address, ProtectAddressFinalizer)
 
 		return nil