add field selector for node store ListWatch if node argument given

Author: txqiangguo

internal/store/node.go

@@ -18,6 +18,8 @@ package store
 
 import (
 	"context"
+	"k8s.io/apimachinery/pkg/fields"
+	"k8s.io/klog/v2"
 	"strings"
 
 	basemetrics "k8s.io/component-base/metrics"
@@ -520,12 +522,23 @@ func wrapNodeFunc(f func(*v1.Node) *metric.Family) func(interface{}) *metric.Fam
 	}
 }
 
-func createNodeListWatch(kubeClient clientset.Interface, _ string, _ string) cache.ListerWatcher {
+func createNodeListWatch(kubeClient clientset.Interface, _ string, fieldSelector string) cache.ListerWatcher {
+	// if given node name, it then lists and watches specified node by its name instead of all nodes.
+	if fieldSelector != "" {
+		selector, _ := fields.ParseSelector(fieldSelector)
+		nodeName, ok := selector.RequiresExactMatch("spec.nodeName")
+		if ok {
+			fieldSelector = fields.OneTermEqualSelector("metadata.name", nodeName).String()
+			klog.InfoS("Transform fieldSelector for node store", "fieldSelector", fieldSelector)
+		}
+	}
 	return &cache.ListWatch{
 		ListFunc: func(opts metav1.ListOptions) (runtime.Object, error) {
+			opts.FieldSelector = fieldSelector
 			return kubeClient.CoreV1().Nodes().List(context.TODO(), opts)
 		},
 		WatchFunc: func(opts metav1.ListOptions) (watch.Interface, error) {
+			opts.FieldSelector = fieldSelector
 			return kubeClient.CoreV1().Nodes().Watch(context.TODO(), opts)
 		},
 	}

pkg/options/types.go

@@ -145,7 +145,7 @@ func (n *NodeType) GetNodeFieldSelector() string {
 		klog.InfoS("Using node type is nil")
 		return EmptyFieldSelector()
 	}
-	pattern := "[^a-zA-Z0-9_,-]+"
+	pattern := "[^a-zA-Z0-9_,-\\.]+"
 	re := regexp.MustCompile(pattern)
 	result := re.ReplaceAllString(n.String(), "")
 	klog.InfoS("Using node type", "node", result)