Azure · zhiying-lin · Nov 13, 2024 · Nov 14, 2024 · jwtty · Nov 22, 2024
diff --git a/api/v1alpha1/trafficmanagerbackend_types.go b/api/v1alpha1/trafficmanagerbackend_types.go
@@ -10,7 +10,7 @@ const (
 // +kubebuilder:resource:scope=Namespaced,categories={fleet-networking},shortName=tmb
 // +kubebuilder:subresource:status
 // +kubebuilder:printcolumn:JSONPath=`.spec.profile.name`,name="Profile",type=string
-// +kubebuilder:printcolumn:JSONPath=`.spec.endpointRef.name`,name="Backend",type=string
+// +kubebuilder:printcolumn:JSONPath=`.spec.backend.name`,name="Backend",type=string
 // +kubebuilder:printcolumn:JSONPath=`.status.conditions[?(@.type=='Accepted')].status`,name="Is-Accepted",type=string
 // +kubebuilder:printcolumn:JSONPath=`.metadata.creationTimestamp`,name="Age",type=date
 

diff --git a/api/v1alpha1/zz_generated.deepcopy.go b/api/v1alpha1/zz_generated.deepcopy.go
diff --git a/azure_config-new.yaml b/azure_config-new.yaml
@@ -0,0 +1,8 @@
+azureCloudConfig:
+  cloud: "AzurePublicCloud"
+  subscriptionId: "c4528d9e-c99a-48bb-b12d-fde2176a43b8"
+  useManagedIdentityExtension: true
+  userAssignedIdentityID: "7099e851-09b2-4214-ba5f-d4d7de33b8a3"
+  resourceGroup: "zhiyinglin-fleet-networking-e2e-atm-1"
+  location: "westcentralus"
+  vnetName: "my-vnet"
diff --git a/azure_config.yaml b/azure_config.yaml
@@ -0,0 +1,9 @@
+  {
+      "cloud": "AzurePublicCloud",
+      "subscriptionId": "c4528d9e-c99a-48bb-b12d-fde2176a43b8",
+      "useManagedIdentityExtension": true,
+      "userAssignedIdentityID": "7099e851-09b2-4214-ba5f-d4d7de33b8a3",
+      "resourceGroup": "zhiyinglin-fleet-networking-e2e-atm-1",
+      "location": "westcentralus",
+      "vnetName": "my-vnet",
+  }
diff --git a/charts/hub-net-controller-manager/templates/rbac.yaml b/charts/hub-net-controller-manager/templates/rbac.yaml
@@ -137,6 +137,58 @@ rules:
     - get
     - list
     - watch
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerbackends
+  verbs:
+    - create
+    - delete
+    - get
+    - list
+    - patch
+    - update
+    - watch
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerbackends/finalizers
+  verbs:
+    - update
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerbackends/status
+  verbs:
+    - get
+    - patch
+    - update
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerprofiles
+  verbs:
+    - create
+    - delete
+    - get
+    - list
+    - patch
+    - update
+    - watch
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerprofiles/finalizers
+  verbs:
+    - update
+- apiGroups:
+    - networking.fleet.azure.com
+  resources:
+    - trafficmanagerprofiles/status
+  verbs:
+    - get
+    - patch
+    - update
 ---
 kind: ClusterRoleBinding
 apiVersion: rbac.authorization.k8s.io/v1

diff --git a/cmd/hub-net-controller-manager/main.go b/cmd/hub-net-controller-manager/main.go
@@ -10,6 +10,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"go.goms.io/fleet-networking/pkg/controllers/hub/trafficmanagerbackend"
 	"os"
 	"time"
 
@@ -42,6 +43,7 @@ import (
 	"go.goms.io/fleet-networking/pkg/controllers/hub/internalserviceimport"
 	"go.goms.io/fleet-networking/pkg/controllers/hub/membercluster"
 	"go.goms.io/fleet-networking/pkg/controllers/hub/serviceimport"
+	"go.goms.io/fleet-networking/pkg/controllers/hub/trafficmanagerprofile"
 )
 
 var (
@@ -200,13 +202,31 @@ func main() {
 		cloudConfig.SetUserAgent("fleet-hub-net-controller-manager")
 		klog.V(1).InfoS("Cloud config loaded", "cloudConfig", cloudConfig)
 
-		_, _, err = initAzureTrafficManagerClients(cloudConfig) // profilesClient, endpointsClient, err
+		profilesClient, endpointsClient, err := initAzureTrafficManagerClients(cloudConfig) // profilesClient, endpointsClient, err
 		if err != nil {
 			klog.ErrorS(err, "Unable to create Azure Traffic Manager clients")
 			exitWithErrorFunc()
 		}
+		klog.V(1).InfoS("Start to setup TrafficManagerProfile controller")
+		if err := (&trafficmanagerprofile.Reconciler{
+			Client:            mgr.GetClient(),
+			ProfilesClient:    profilesClient,
+			ResourceGroupName: cloudConfig.ResourceGroup,
+		}).SetupWithManager(mgr); err != nil {
+			klog.ErrorS(err, "Unable to create TrafficManagerProfile controller")
+			exitWithErrorFunc()
+		}
 
-		// TODO: start the traffic manager controllers
+		klog.V(1).InfoS("Start to setup TrafficManagerBackend controller")
+		if err := (&trafficmanagerbackend.Reconciler{
+			Client:            mgr.GetClient(),
+			ProfilesClient:    profilesClient,
+			EndpointsClient:   endpointsClient,
+			ResourceGroupName: cloudConfig.ResourceGroup,
+		}).SetupWithManager(ctx, mgr); err != nil {
+			klog.ErrorS(err, "Unable to create TrafficManagerProfile controller")
+			exitWithErrorFunc()
+		}
 	}
 
 	klog.V(1).InfoS("Starting ServiceExportImport controller manager")

diff --git a/config/crd/bases/networking.fleet.azure.com_trafficmanagerbackends.yaml b/config/crd/bases/networking.fleet.azure.com_trafficmanagerbackends.yaml
@@ -22,7 +22,7 @@ spec:
     - jsonPath: .spec.profile.name
       name: Profile
       type: string
-    - jsonPath: .spec.endpointRef.name
+    - jsonPath: .spec.backend.name
       name: Backend
       type: string
     - jsonPath: .status.conditions[?(@.type=='Accepted')].status

diff --git a/config/rbac/role.yaml b/config/rbac/role.yaml
@@ -63,6 +63,7 @@ rules:
   - multiclusterservices
   - serviceexports
   - serviceimports
+  - trafficmanagerbackends
   - trafficmanagerprofiles
   verbs:
   - create
@@ -86,6 +87,7 @@ rules:
   - multiclusterservices/status
   - serviceexports/status
   - serviceimports/status
+  - trafficmanagerbackends/status
   - trafficmanagerprofiles/status
   verbs:
   - get
@@ -96,6 +98,7 @@ rules:
   resources:
   - multiclusterservices/finalizers
   - serviceimports/finalizers
+  - trafficmanagerbackends/finalizers
   - trafficmanagerprofiles/finalizers
   verbs:
   - get

diff --git a/hub_azure_config.yaml b/hub_azure_config.yaml
@@ -0,0 +1,8 @@
+  azureCloudConfig:
+    cloud: "AzurePublicCloud"
+    subscriptionId: "c4528d9e-c99a-48bb-b12d-fde2176a43b8"
+    useManagedIdentityExtension: true
+    userAssignedIdentityID: "aab3e00d-6427-48b0-af5d-ddffb2a284e0"
+    resourceGroup: "zhiyinglin-fleet-networking-e2e-atm-1"
+    location: "westcentralus"
+    vnetName: "my-vnet"
diff --git a/member_1_azure_config.yaml b/member_1_azure_config.yaml
@@ -0,0 +1,8 @@
+  azureCloudConfig:
+    cloud: "AzurePublicCloud"
+    subscriptionId: "c4528d9e-c99a-48bb-b12d-fde2176a43b8"
+    useManagedIdentityExtension: true
+    userAssignedIdentityID: "f882963a-abce-416c-9e44-172f678475f9"
+    resourceGroup: "MC_zhiyinglin-fleet-networking-e2e-atm-1_member-1_westcentralus"
+    location: "westcentralus"
+    vnetName: "my-vnet"
diff --git a/member_2_azure_config.yaml b/member_2_azure_config.yaml
@@ -0,0 +1,8 @@
+  azureCloudConfig:
+    cloud: "AzurePublicCloud"
+    subscriptionId: "c4528d9e-c99a-48bb-b12d-fde2176a43b8"
+    useManagedIdentityExtension: true
+    userAssignedIdentityID: "30d298c0-bf15-402a-b072-8fef4b3653ef"
+    resourceGroup: "MC_zhiyinglin-fleet-networking-e2e-atm-1_member-2_westcentralus"
+    location: "westcentralus"
+    vnetName: "my-vnet"