diff --git a/pkg/kubelet/kubelet.go b/pkg/kubelet/kubelet.go index f93edf0e80e..59699d66570 100644 --- a/pkg/kubelet/kubelet.go +++ b/pkg/kubelet/kubelet.go @@ -207,7 +207,11 @@ func NewMainKubelet( fieldSelector := fields.Set{client.ObjectNameField: nodeName}.AsSelector() listWatch := &cache.ListWatch{ ListFunc: func() (runtime.Object, error) { - return kubeClient.Nodes().List(labels.Everything(), fieldSelector) + obj, err := kubeClient.Nodes().Get(nodeName) + if err != nil { + return nil, err + } + return &api.NodeList{Items: []api.Node{*obj}}, nil }, WatchFunc: func(resourceVersion string) (watch.Interface, error) { return kubeClient.Nodes().Watch(labels.Everything(), fieldSelector, resourceVersion) @@ -714,17 +718,7 @@ func (kl *Kubelet) GetNode() (*api.Node, error) { if kl.standaloneMode { return nil, errors.New("no node entry for kubelet in standalone mode") } - l, err := kl.nodeLister.List() - if err != nil { - return nil, errors.New("cannot list nodes") - } - nodeName := kl.nodeName - for _, n := range l.Items { - if n.Name == nodeName { - return &n, nil - } - } - return nil, fmt.Errorf("node %v not found", nodeName) + return kl.nodeLister.GetNodeInfo(kl.nodeName) } // Starts garbage collection threads. diff --git a/pkg/kubelet/kubelet_test.go b/pkg/kubelet/kubelet_test.go index 87fba50f571..52c5699eed7 100644 --- a/pkg/kubelet/kubelet_test.go +++ b/pkg/kubelet/kubelet_test.go @@ -18,7 +18,6 @@ package kubelet import ( "bytes" - "errors" "fmt" "io" "io/ioutil" @@ -966,7 +965,12 @@ type testNodeLister struct { } func (ls testNodeLister) GetNodeInfo(id string) (*api.Node, error) { - return nil, errors.New("not implemented") + for _, node := range ls.nodes { + if node.Name == id { + return &node, nil + } + } + return nil, fmt.Errorf("Node with name: %s does not exist", id) } func (ls testNodeLister) List() (api.NodeList, error) {