77 lines
2.5 KiB
Go
77 lines
2.5 KiB
Go
// Package router — NSWatcher for multi-tenant mode.
|
|
//
|
|
// Listens for Namespaces labeled fission.io/managed=true and calls
|
|
// HTTPTriggerSet.AddNamespace so the router picks up HTTPTriggers and
|
|
// Functions in tenant namespaces without a restart.
|
|
package router
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"go.uber.org/zap"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
k8sInformers "k8s.io/client-go/informers"
|
|
"k8s.io/client-go/kubernetes"
|
|
k8sCache "k8s.io/client-go/tools/cache"
|
|
|
|
"github.com/fission/fission/pkg/utils"
|
|
"github.com/fission/fission/pkg/utils/manager"
|
|
)
|
|
|
|
// StartNSWatcher registers a Kubernetes Namespace Informer for the router.
|
|
// Whenever a Namespace with label fission.io/managed=true appears (or is relabeled),
|
|
// the router immediately subscribes to HTTPTriggers and Functions in that namespace.
|
|
func StartNSWatcher(
|
|
ctx context.Context,
|
|
logger *zap.Logger,
|
|
kubeClient kubernetes.Interface,
|
|
ts *HTTPTriggerSet,
|
|
mgr manager.Interface,
|
|
) {
|
|
nsManager, err := utils.NewWatcherNamespaceManager(ctx, utils.DefaultNSResolver().Snapshot(), utils.NamespaceSourceEnv, time.Now().UTC(), NewNamespaceSubscriber(ts, mgr))
|
|
if err != nil {
|
|
logger.Error("router.NSWatcher: BootstrapAndDispatch failed", zap.Error(err))
|
|
}
|
|
|
|
factory := k8sInformers.NewSharedInformerFactoryWithOptions(
|
|
kubeClient,
|
|
30*time.Minute,
|
|
k8sInformers.WithTweakListOptions(func(opts *metav1.ListOptions) {
|
|
opts.LabelSelector = utils.ManagedNamespaceLabelSelector()
|
|
}),
|
|
)
|
|
|
|
nsInformer := factory.Core().V1().Namespaces().Informer()
|
|
|
|
_, _ = nsInformer.AddEventHandler(k8sCache.ResourceEventHandlerFuncs{
|
|
AddFunc: func(obj interface{}) {
|
|
nsObj, ok := obj.(*corev1.Namespace)
|
|
if ok {
|
|
utils.HandleWatcherNamespaceAdd(ctx, logger, "router.NSWatcher", nsManager, nsObj)
|
|
}
|
|
},
|
|
UpdateFunc: func(oldObj, newObj interface{}) {
|
|
oldNSObj, _ := oldObj.(*corev1.Namespace)
|
|
nsObj, ok := newObj.(*corev1.Namespace)
|
|
if ok {
|
|
utils.HandleWatcherNamespaceUpdate(ctx, logger, "router.NSWatcher", nsManager, oldNSObj, nsObj)
|
|
}
|
|
},
|
|
DeleteFunc: func(obj interface{}) {
|
|
utils.HandleWatcherNamespaceDelete(logger, "router.NSWatcher", nsManager, obj)
|
|
},
|
|
})
|
|
|
|
mgr.Add(ctx, func(ctx context.Context) {
|
|
logger.Info("router.NSWatcher: started",
|
|
zap.String("label", utils.ManagedNamespaceLabelSelector()))
|
|
factory.Start(ctx.Done())
|
|
factory.WaitForCacheSync(ctx.Done())
|
|
logger.Info("router.NSWatcher: cache synced — watching for new namespaces")
|
|
<-ctx.Done()
|
|
logger.Info("router.NSWatcher: stopped")
|
|
})
|
|
}
|