From 2cc7be2921db3b20991cefd496e923dd8e923fc8 Mon Sep 17 00:00:00 2001 From: songjuchao Date: Mon, 18 Nov 2019 23:02:15 +0800 Subject: [PATCH] add node LabelSelector field, just for https://github.com/robscott/kube-capacity/issues/19 --- pkg/capacity/capacity.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/pkg/capacity/capacity.go b/pkg/capacity/capacity.go index a064a8bc..3df37779 100644 --- a/pkg/capacity/capacity.go +++ b/pkg/capacity/capacity.go @@ -47,7 +47,7 @@ func FetchAndPrint(showContainers, showPods, showUtil bool, podLabels, nodeLabel } pmList = getPodMetrics(mClientset) - nmList = getNodeMetrics(mClientset) + nmList = getNodeMetrics(mClientset, nodeLabels) } cm := buildClusterMetric(podList, pmList, nodeList, nmList) @@ -129,8 +129,10 @@ func getPodMetrics(mClientset *metrics.Clientset) *v1beta1.PodMetricsList { return pmList } -func getNodeMetrics(mClientset *metrics.Clientset) *v1beta1.NodeMetricsList { - nmList, err := mClientset.MetricsV1beta1().NodeMetricses().List(metav1.ListOptions{}) +func getNodeMetrics(mClientset *metrics.Clientset, nodeLabels string) *v1beta1.NodeMetricsList { + nmList, err := mClientset.MetricsV1beta1().NodeMetricses().List(metav1.ListOptions{ + LabelSelector: nodeLabels, + }) if err != nil { fmt.Printf("Error getting Node Metrics: %v\n", err) fmt.Println("For this to work, metrics-server needs to be running in your cluster")