Migrate ingress.class annotation to new IngressClassName field

internal/ingress/annotations/class/main.go

@@ -17,8 +17,6 @@ limitations under the License.
 package class
 
 import (
-	"k8s.io/klog"
-
 	networking "k8s.io/api/networking/v1beta1"
 )
 
@@ -43,10 +41,7 @@ var (
 // the ingress.class annotation, or it's set to the configured in the
 // ingress controller.
 func IsValid(ing *networking.Ingress) bool {
-	ingress, ok := ing.GetAnnotations()[IngressKey]
-	if !ok {
-		klog.V(3).Infof("annotation %v is not present in ingress %v/%v", IngressKey, ing.Namespace, ing.Name)
-	}
+	className := ing.Spec.IngressClassName
 
 	// we have 2 valid combinations
 	// 1 - ingress with default class | blank annotation on ingress
@@ -55,9 +50,17 @@ func IsValid(ing *networking.Ingress) bool {
 	// and 2 invalid combinations
 	// 3 - ingress with default class | fixed annotation on ingress
 	// 4 - ingress with specific class | different annotation on ingress
-	if ingress == "" && IngressClass == DefaultClass {
+	if className != nil {
+		return *className == IngressClass
+	}
+
+	if IngressClass == DefaultClass {
 		return true
 	}
 
-	return ingress == IngressClass
+	if IngressClass == "" {
+		return true
+	}
+
+	return false
 }

internal/ingress/annotations/class/main_test.go

@@ -54,10 +54,10 @@ func TestIsValidClass(t *testing.T) {
 		},
 	}
 
-	data := map[string]string{}
-	ing.SetAnnotations(data)
 	for _, test := range tests {
-		ing.Annotations[IngressKey] = test.ingress
+		if test.ingress != "" {
+			ing.Spec.IngressClassName = &test.ingress
+		}
 
 		IngressClass = test.controller
 		DefaultClass = test.defClass

internal/ingress/controller/controller_test.go

@@ -190,7 +190,8 @@ func TestCheckIngress(t *testing.T) {
 	}
 
 	t.Run("When the ingress class differs from nginx", func(t *testing.T) {
-		ing.ObjectMeta.Annotations["kubernetes.io/ingress.class"] = "different"
+		class := "different"
+		ing.Spec.IngressClassName = &class
 		nginx.command = testNginxTestCommand{
 			t:   t,
 			err: fmt.Errorf("test error"),
@@ -201,7 +202,8 @@ func TestCheckIngress(t *testing.T) {
 	})
 
 	t.Run("when the class is the nginx one", func(t *testing.T) {
-		ing.ObjectMeta.Annotations["kubernetes.io/ingress.class"] = "nginx"
+		class := "nginx"
+		ing.Spec.IngressClassName = &class
 		nginx.command = testNginxTestCommand{
 			t:        t,
 			err:      nil,

internal/ingress/controller/store/store.go

@@ -949,12 +949,22 @@ func toIngress(obj interface{}) (*networkingv1beta1.Ingress, bool) {
 			return nil, false
 		}
 
+		ing.Spec.IngressClassName = extractClassName(ing)
 		return ing, true
 	}
 
 	if ing, ok := obj.(*networkingv1beta1.Ingress); ok {
+		ing.Spec.IngressClassName = extractClassName(ing)
 		return ing, true
 	}
 
 	return nil, false
 }
+
+func extractClassName(ing *networkingv1beta1.Ingress) *string {
+	if c, ok := ing.Annotations[class.IngressKey]; ok {
+		return &c
+	}
+
+	return nil
+}