Add support for CATTLE_GLOBAL_LABEL_SELECTOR env var

beeper · Jan 16, 2025 · 239b8be · 239b8be
1 parent a80bf83
commit 239b8be
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/pkg/clustercache/controller.go b/pkg/clustercache/controller.go
@@ -2,6 +2,7 @@ package clustercache
 
 import (
 	"context"
+	"os"
 	"sync"
 	"time"
 
@@ -124,6 +125,17 @@ func (h *clusterCache) addResourceEventHandler(gvk schema2.GroupVersionKind, inf
 	})
 }
 
+func getTweakListOptions() informer.TweakListOptionsFunc {
+	globalLabelSelector := os.Getenv("CATTLE_GLOBAL_LABEL_SELECTOR")
+	if globalLabelSelector == "" {
+		return func(*metav1.ListOptions) {}
+	}
+	return func(opts *metav1.ListOptions) {
+		logrus.Infof("Applying global label selector: %s", globalLabelSelector)
+		opts.LabelSelector = globalLabelSelector
+	}
+}
+
 func (h *clusterCache) OnSchemas(schemas *schema.Collection) error {
 	h.Lock()
 	defer h.Unlock()
@@ -148,7 +160,7 @@ func (h *clusterCache) OnSchemas(schemas *schema.Collection) error {
 		}
 
 		summaryInformer := informer.NewFilteredSummaryInformer(h.summaryClient, gvr, metav1.NamespaceAll, 2*time.Hour,
-			cache.Indexers{cache.NamespaceIndex: cache.MetaNamespaceIndexFunc}, nil)
+			cache.Indexers{cache.NamespaceIndex: cache.MetaNamespaceIndexFunc}, getTweakListOptions())
 		ctx, cancel := context.WithCancel(h.ctx)
 		w := &watcher{
 			ctx:      ctx,