Move codes to pkg (#46)

pkg/ingress/kube/ingressv1/controller_test.go

@@ -0,0 +1,78 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package ingressv1
+
+import (
+	"testing"
+
+	v1 "k8s.io/api/networking/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"github.com/alibaba/higress/pkg/ingress/kube/common"
+)
+
+func TestShouldProcessIngressUpdate(t *testing.T) {
+	c := controller{
+		options: common.Options{
+			IngressClass: "mse",
+		},
+		ingresses: make(map[string]*v1.Ingress),
+	}
+
+	ingressClass := "mse"
+
+	ingress1 := &v1.Ingress{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: "test-1",
+		},
+		Spec: v1.IngressSpec{
+			IngressClassName: &ingressClass,
+			Rules: []v1.IngressRule{
+				{
+					Host: "test.com",
+					IngressRuleValue: v1.IngressRuleValue{
+						HTTP: &v1.HTTPIngressRuleValue{
+							Paths: []v1.HTTPIngressPath{
+								{
+									Path: "/test",
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	should, _ := c.shouldProcessIngressUpdate(ingress1)
+	if !should {
+		t.Fatal("should be true")
+	}
+
+	ingress2 := *ingress1
+	should, _ = c.shouldProcessIngressUpdate(&ingress2)
+	if should {
+		t.Fatal("should be false")
+	}
+
+	ingress3 := *ingress1
+	ingress3.Annotations = map[string]string{
+		"test": "true",
+	}
+	should, _ = c.shouldProcessIngressUpdate(&ingress3)
+	if !should {
+		t.Fatal("should be true")
+	}
+}

pkg/ingress/kube/ingressv1/status.go

@@ -0,0 +1,133 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package ingressv1
+
+import (
+	"context"
+	"reflect"
+	"sort"
+	"time"
+
+	kubelib "istio.io/istio/pkg/kube"
+	coreV1 "k8s.io/api/core/v1"
+	metaV1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/labels"
+	"k8s.io/client-go/kubernetes"
+	listerv1 "k8s.io/client-go/listers/core/v1"
+	ingresslister "k8s.io/client-go/listers/networking/v1"
+	"k8s.io/client-go/tools/cache"
+
+	"github.com/alibaba/higress/pkg/ingress/kube/common"
+	. "github.com/alibaba/higress/pkg/ingress/log"
+)
+
+// statusSyncer keeps the status IP in each Ingress resource updated
+type statusSyncer struct {
+	client     kubernetes.Interface
+	controller *controller
+
+	watchedNamespace string
+
+	ingressLister      ingresslister.IngressLister
+	ingressClassLister ingresslister.IngressClassLister
+	// search service in the mse vpc
+	serviceLister listerv1.ServiceLister
+}
+
+// newStatusSyncer creates a new instance
+func newStatusSyncer(localKubeClient, client kubelib.Client, controller *controller, namespace string) *statusSyncer {
+	return &statusSyncer{
+		client:             client,
+		controller:         controller,
+		watchedNamespace:   namespace,
+		ingressLister:      client.KubeInformer().Networking().V1().Ingresses().Lister(),
+		ingressClassLister: client.KubeInformer().Networking().V1().IngressClasses().Lister(),
+		// search service in the mse vpc
+		serviceLister: localKubeClient.KubeInformer().Core().V1().Services().Lister(),
+	}
+}
+
+func (s *statusSyncer) run(stopCh <-chan struct{}) {
+	cache.WaitForCacheSync(stopCh, s.controller.HasSynced)
+
+	ticker := time.NewTicker(common.DefaultStatusUpdateInterval)
+	for {
+		select {
+		case <-stopCh:
+			ticker.Stop()
+			return
+		case <-ticker.C:
+			if err := s.runUpdateStatus(); err != nil {
+				IngressLog.Errorf("update status task fail, err %v", err)
+			}
+		}
+	}
+}
+
+func (s *statusSyncer) runUpdateStatus() error {
+	svcList, err := s.serviceLister.Services(s.watchedNamespace).List(common.SvcLabelSelector)
+	if err != nil {
+		return err
+	}
+
+	IngressLog.Debugf("found number %d of svc", len(svcList))
+
+	lbStatusList := common.GetLbStatusList(svcList)
+	if len(lbStatusList) == 0 {
+		return nil
+	}
+
+	return s.updateStatus(lbStatusList)
+}
+
+// updateStatus updates ingress status with the list of IP
+func (s *statusSyncer) updateStatus(status []coreV1.LoadBalancerIngress) error {
+	ingressList, err := s.ingressLister.List(labels.Everything())
+	if err != nil {
+		return err
+	}
+	for _, ingress := range ingressList {
+		shouldTarget, err := s.controller.shouldProcessIngress(ingress)
+		if err != nil {
+			IngressLog.Warnf("error determining whether should target ingress %s/%s within cluster %s for status update: %v",
+				ingress.Namespace, ingress.Name, s.controller.options.ClusterId, err)
+			return err
+		}
+
+		if !shouldTarget {
+			continue
+		}
+
+		curIPs := ingress.Status.LoadBalancer.Ingress
+		sort.SliceStable(curIPs, common.SortLbIngressList(curIPs))
+
+		if reflect.DeepEqual(status, curIPs) {
+			IngressLog.Debugf("skipping update of Ingress %v/%v within cluster %s (no change)",
+				ingress.Namespace, ingress.Name, s.controller.options.ClusterId)
+			continue
+		}
+
+		ingress.Status.LoadBalancer.Ingress = status
+		IngressLog.Infof("Update Ingress %v/%v within cluster %s status",
+			ingress.Namespace, ingress.Name, s.controller.options.ClusterId)
+		_, err = s.client.NetworkingV1().Ingresses(ingress.Namespace).UpdateStatus(context.TODO(), ingress, metaV1.UpdateOptions{})
+		if err != nil {
+			IngressLog.Warnf("error updating ingress %s/%s within cluster %s status: %v",
+				ingress.Namespace, ingress.Name, s.controller.options.ClusterId, err)
+		}
+	}
+
+	return nil
+}