backend: k8scache: add stopwatcher to remove the context from watch r…

backend/cmd/headlamp.go

@@ -47,6 +47,7 @@ import (
 	auth "github.com/kubernetes-sigs/headlamp/backend/pkg/auth"
 	"github.com/kubernetes-sigs/headlamp/backend/pkg/cache"
 	cfg "github.com/kubernetes-sigs/headlamp/backend/pkg/config"
+	"github.com/kubernetes-sigs/headlamp/backend/pkg/k8cache"
 	"github.com/kubernetes-sigs/headlamp/backend/pkg/serviceproxy"
 
 	headlampcfg "github.com/kubernetes-sigs/headlamp/backend/pkg/headlampconfig"
@@ -444,6 +445,8 @@ func createHeadlampHandler(config *HeadlampConfig) http.Handler {
 			pluginEventChan,
 			config.cache,
 		)
+		// Set the function to stop kubeconfig context watchers
+		kubeconfig.StopContextWatcher = k8cache.StopWatcher
 		// in-cluster mode is unlikely to want reloading kubeconfig.
 		go kubeconfig.LoadAndWatchFiles(config.KubeConfigStore, kubeConfigPath, kubeconfig.KubeConfig, skipFunc)
 	}

backend/pkg/k8cache/cacheInvalidation.go

@@ -161,6 +161,17 @@ func CheckForChanges(
 	go runWatcher(ctx, k8scache, contextKey, kContext)
 }
 
+// StopWatcher stops and cleans up the watcher for a given contextKey.
+func StopWatcher(contextKey string) {
+	logger.Log(logger.LevelInfo, nil, nil, "stopping watcher for context: "+contextKey)
+
+	if cancelAny, ok := contextCancel.Load(contextKey); ok {
+		cancelAny.(context.CancelFunc)()
+		contextCancel.Delete(contextKey)
+		watcherRegistry.Delete(contextKey)
+	}
+}
+
 // runWatcher is a long-lived goroutine that sets up and runs Kubernetes informers.
 // It watches for resource changes and invalidates corresponding cache entries.
 // This function will only exit when its context is cancelled.

backend/pkg/kubeconfig/watcher.go

@@ -13,6 +13,8 @@ import (
 
 const watchInterval = 10 * time.Second
 
+var StopContextWatcher func(contextName string)
+
 // LoadAndWatchFiles loads kubeconfig files and watches them for changes.
 func LoadAndWatchFiles(kubeConfigStore ContextStore, paths string, source int, ignoreFunc shouldBeSkippedFunc) {
 	// create ticker
@@ -138,6 +140,10 @@ func syncContexts(kubeConfigStore ContextStore, paths string, source int, ignore
 		}
 
 		if !found {
+			if StopContextWatcher != nil {
+				StopContextWatcher(existingCtx.Name)
+			}
+
 			err := kubeConfigStore.RemoveContext(existingCtx.Name)
 			if err != nil {
 				logger.Log(logger.LevelError, nil, err, "error removing context")