Migrate TaintManager to use watch for listing pods instead of expensive

listing pods call.

pkg/controller/nodelifecycle/node_lifecycle_controller.go

@@ -126,7 +126,8 @@ const (
 
 const (
 	// The amount of time the nodecontroller should sleep between retrying node health updates
-	retrySleepTime = 20 * time.Millisecond
+	retrySleepTime   = 20 * time.Millisecond
+	nodeNameKeyIndex = "spec.nodeName"
 )
 
 // labelReconcileInfo lists Node labels to reconcile, and how to reconcile them.
@@ -364,11 +365,40 @@ func NewNodeLifecycleController(
 	nc.podInformerSynced = podInformer.Informer().HasSynced
 
 	if nc.runTaintManager {
+		podInformer.Informer().AddIndexers(cache.Indexers{
+			nodeNameKeyIndex: func(obj interface{}) ([]string, error) {
+				pod, ok := obj.(*v1.Pod)
+				if !ok {
+					return []string{}, nil
+				}
+				if len(pod.Spec.NodeName) == 0 {
+					return []string{}, nil
+				}
+				return []string{pod.Spec.NodeName}, nil
+			},
+		})
+
+		podIndexer := podInformer.Informer().GetIndexer()
 		podLister := podInformer.Lister()
 		podGetter := func(name, namespace string) (*v1.Pod, error) { return podLister.Pods(namespace).Get(name) }
+		podByNodeNameLister := func(nodeName string) ([]v1.Pod, error) {
+			objs, err := podIndexer.ByIndex(nodeNameKeyIndex, nodeName)
+			if err != nil {
+				return nil, err
+			}
+			pods := make([]v1.Pod, 0, len(objs))
+			for _, obj := range objs {
+				pod, ok := obj.(*v1.Pod)
+				if !ok {
+					continue
+				}
+				pods = append(pods, *pod)
+			}
+			return pods, nil
+		}
 		nodeLister := nodeInformer.Lister()
 		nodeGetter := func(name string) (*v1.Node, error) { return nodeLister.Get(name) }
-		nc.taintManager = scheduler.NewNoExecuteTaintManager(kubeClient, podGetter, nodeGetter)
+		nc.taintManager = scheduler.NewNoExecuteTaintManager(kubeClient, podGetter, nodeGetter, podByNodeNameLister)
 		nodeInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
 			AddFunc: nodeutil.CreateAddNodeHandler(func(node *v1.Node) error {
 				nc.taintManager.NodeUpdated(nil, node)