adding a needed label on the Namespace when using UDN primary network

cloud-bulldozer · Jan 29, 2025 · 2d866e7 · 2d866e7
1 parent 747be21
commit 2d866e7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/cmd/k8s-netperf/k8s-netperf.go b/cmd/k8s-netperf/k8s-netperf.go
@@ -151,7 +151,7 @@ var rootCmd = &cobra.Command{
 		}
 
 		// Build the namespace and create the sa account
-		err = k8s.BuildInfra(client)
+		err = k8s.BuildInfra(client, udn)
 		if err != nil {
 			log.Error(err)
 			os.Exit(1)

diff --git a/pkg/k8s/kubernetes.go b/pkg/k8s/kubernetes.go
@@ -87,13 +87,18 @@ const k8sNetperfImage = "quay.io/cloud-bulldozer/k8s-netperf:latest"
 const udnName = "udn-l2-primary"
 
 // BuildInfra will create the infra for the SUT
-func BuildInfra(client *kubernetes.Clientset) error {
+func BuildInfra(client *kubernetes.Clientset, udn bool) error {
 	_, err := client.CoreV1().Namespaces().Get(context.TODO(), namespace, metav1.GetOptions{})
 	if err == nil {
 		log.Infof("♻️ Namespace already exists, reusing it")
 	} else {
 		log.Infof("🔨 Creating namespace: %s", namespace)
-		_, err := client.CoreV1().Namespaces().Create(context.TODO(), &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: namespace}}, metav1.CreateOptions{})
+		if udn {
+			_, err = client.CoreV1().Namespaces().Create(context.TODO(), &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: namespace,
+				Labels: map[string]string{"k8s.ovn.org/primary-user-defined-network": ""}}}, metav1.CreateOptions{})
+		} else {
+			_, err = client.CoreV1().Namespaces().Create(context.TODO(), &corev1.Namespace{ObjectMeta: metav1.ObjectMeta{Name: namespace}}, metav1.CreateOptions{})
+		}
 		if err != nil {
 			return fmt.Errorf("😥 Unable to create namespace: %v", err)
 		}