mirror of
https://github.com/alibaba/higress.git
synced 2026-02-25 13:10:50 +08:00
130 lines
3.6 KiB
Go
130 lines
3.6 KiB
Go
// 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 controller
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"istio.io/istio/pkg/cluster"
|
|
"istio.io/istio/pkg/kube/controllers"
|
|
kerrors "k8s.io/apimachinery/pkg/api/errors"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"k8s.io/client-go/tools/cache"
|
|
|
|
"github.com/alibaba/higress/pkg/ingress/kube/util"
|
|
. "github.com/alibaba/higress/pkg/ingress/log"
|
|
)
|
|
|
|
type Controller[lister any] interface {
|
|
AddEventHandler(addOrUpdate func(util.ClusterNamespacedName), delete ...func(util.ClusterNamespacedName))
|
|
|
|
Run(stop <-chan struct{})
|
|
|
|
HasSynced() bool
|
|
|
|
Lister() lister
|
|
|
|
Get(types.NamespacedName) (controllers.Object, error)
|
|
|
|
Informer() cache.SharedIndexInformer
|
|
}
|
|
|
|
type GetObjectFunc[lister any] func(lister, types.NamespacedName) (controllers.Object, error)
|
|
|
|
type CommonController[lister any] struct {
|
|
typeName string
|
|
queue controllers.Queue
|
|
informer cache.SharedIndexInformer
|
|
lister lister
|
|
updateHandler func(util.ClusterNamespacedName)
|
|
removeHandler func(util.ClusterNamespacedName)
|
|
getFunc GetObjectFunc[lister]
|
|
clusterId cluster.ID
|
|
}
|
|
|
|
func NewCommonController[lister any](typeName string, listerObj lister, informer cache.SharedIndexInformer,
|
|
getFunc GetObjectFunc[lister], clusterId cluster.ID) Controller[lister] {
|
|
c := &CommonController[lister]{
|
|
typeName: typeName,
|
|
lister: listerObj,
|
|
informer: informer,
|
|
clusterId: clusterId,
|
|
getFunc: getFunc,
|
|
}
|
|
c.queue = controllers.NewQueue(typeName,
|
|
controllers.WithReconciler(c.onEvent),
|
|
controllers.WithMaxAttempts(5))
|
|
_, _ = c.informer.AddEventHandler(controllers.ObjectHandler(c.queue.AddObject))
|
|
return c
|
|
}
|
|
|
|
func (c *CommonController[lister]) Lister() lister {
|
|
return c.lister
|
|
}
|
|
|
|
func (c *CommonController[lister]) Informer() cache.SharedIndexInformer {
|
|
return c.informer
|
|
}
|
|
|
|
func (c *CommonController[lister]) AddEventHandler(addOrUpdate func(util.ClusterNamespacedName), delete ...func(util.ClusterNamespacedName)) {
|
|
c.updateHandler = addOrUpdate
|
|
if len(delete) > 0 {
|
|
c.removeHandler = delete[0]
|
|
}
|
|
}
|
|
|
|
func (c *CommonController[lister]) Run(stop <-chan struct{}) {
|
|
if !cache.WaitForCacheSync(stop, c.informer.HasSynced) {
|
|
IngressLog.Errorf("Failed to sync %s controller cache", c.typeName)
|
|
return
|
|
}
|
|
c.queue.Run(stop)
|
|
}
|
|
|
|
func (c *CommonController[lister]) onEvent(namespacedName types.NamespacedName) error {
|
|
if c.getFunc == nil {
|
|
return errors.New("getFunc is nil")
|
|
}
|
|
obj := util.ClusterNamespacedName{
|
|
NamespacedName: types.NamespacedName{
|
|
Namespace: namespacedName.Namespace,
|
|
Name: namespacedName.Name,
|
|
},
|
|
ClusterId: c.clusterId,
|
|
}
|
|
_, err := c.getFunc(c.lister, namespacedName)
|
|
if err != nil {
|
|
if kerrors.IsNotFound(err) {
|
|
if c.removeHandler == nil {
|
|
return nil
|
|
}
|
|
c.removeHandler(obj)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
|
|
c.updateHandler(obj)
|
|
return nil
|
|
}
|
|
|
|
func (c *CommonController[lister]) Get(namespacedName types.NamespacedName) (controllers.Object, error) {
|
|
return c.getFunc(c.lister, namespacedName)
|
|
}
|
|
|
|
func (c *CommonController[lister]) HasSynced() bool {
|
|
return c.queue.HasSynced()
|
|
}
|