vendor files

vendor/k8s.io/kubernetes/pkg/kubectl/metricsutil/BUILD

@@ -0,0 +1,43 @@
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "metrics_client.go",
+        "metrics_printer.go",
+    ],
+    importpath = "k8s.io/kubernetes/pkg/kubectl/metricsutil",
+    visibility = [
+        "//build/visible_to:pkg_kubectl_metricsutil_CONSUMERS",
+    ],
+    deps = [
+        "//pkg/api/legacyscheme:go_default_library",
+        "//pkg/apis/core/validation:go_default_library",
+        "//pkg/printers:go_default_library",
+        "//vendor/k8s.io/api/core/v1:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/api/resource:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/apis/meta/v1:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/labels:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/runtime/schema:go_default_library",
+        "//vendor/k8s.io/client-go/kubernetes/typed/core/v1:go_default_library",
+        "//vendor/k8s.io/metrics/pkg/apis/metrics/v1alpha1:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = [
+        "//build/visible_to:pkg_kubectl_metricsutil_CONSUMERS",
+    ],
+)

vendor/k8s.io/kubernetes/pkg/kubectl/metricsutil/metrics_client.go

@@ -0,0 +1,167 @@
+/*
+Copyright 2016 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package metricsutil
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/labels"
+	"k8s.io/apimachinery/pkg/runtime/schema"
+	corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
+	"k8s.io/kubernetes/pkg/apis/core/validation"
+	metricsapi "k8s.io/metrics/pkg/apis/metrics/v1alpha1"
+)
+
+const (
+	DefaultHeapsterNamespace = "kube-system"
+	DefaultHeapsterScheme    = "http"
+	DefaultHeapsterService   = "heapster"
+	DefaultHeapsterPort      = "" // use the first exposed port on the service
+)
+
+var (
+	prefix       = "/apis"
+	groupVersion = fmt.Sprintf("%s/%s", metricsGv.Group, metricsGv.Version)
+	metricsRoot  = fmt.Sprintf("%s/%s", prefix, groupVersion)
+
+	// TODO: get this from metrics api once it's finished
+	metricsGv = schema.GroupVersion{Group: "metrics", Version: "v1alpha1"}
+)
+
+type HeapsterMetricsClient struct {
+	SVCClient         corev1.ServicesGetter
+	HeapsterNamespace string
+	HeapsterScheme    string
+	HeapsterService   string
+	HeapsterPort      string
+}
+
+func NewHeapsterMetricsClient(svcClient corev1.ServicesGetter, namespace, scheme, service, port string) *HeapsterMetricsClient {
+	return &HeapsterMetricsClient{
+		SVCClient:         svcClient,
+		HeapsterNamespace: namespace,
+		HeapsterScheme:    scheme,
+		HeapsterService:   service,
+		HeapsterPort:      port,
+	}
+}
+
+func DefaultHeapsterMetricsClient(svcClient corev1.ServicesGetter) *HeapsterMetricsClient {
+	return NewHeapsterMetricsClient(svcClient, DefaultHeapsterNamespace, DefaultHeapsterScheme, DefaultHeapsterService, DefaultHeapsterPort)
+}
+
+func podMetricsUrl(namespace string, name string) (string, error) {
+	if namespace == metav1.NamespaceAll {
+		return fmt.Sprintf("%s/pods", metricsRoot), nil
+	}
+	errs := validation.ValidateNamespaceName(namespace, false)
+	if len(errs) > 0 {
+		message := fmt.Sprintf("invalid namespace: %s - %v", namespace, errs)
+		return "", errors.New(message)
+	}
+	if len(name) > 0 {
+		errs = validation.ValidatePodName(name, false)
+		if len(errs) > 0 {
+			message := fmt.Sprintf("invalid pod name: %s - %v", name, errs)
+			return "", errors.New(message)
+		}
+	}
+	return fmt.Sprintf("%s/namespaces/%s/pods/%s", metricsRoot, namespace, name), nil
+}
+
+func nodeMetricsUrl(name string) (string, error) {
+	if len(name) > 0 {
+		errs := validation.ValidateNodeName(name, false)
+		if len(errs) > 0 {
+			message := fmt.Sprintf("invalid node name: %s - %v", name, errs)
+			return "", errors.New(message)
+		}
+	}
+	return fmt.Sprintf("%s/nodes/%s", metricsRoot, name), nil
+}
+
+func (cli *HeapsterMetricsClient) GetNodeMetrics(nodeName string, selector string) ([]metricsapi.NodeMetrics, error) {
+	params := map[string]string{"labelSelector": selector}
+	path, err := nodeMetricsUrl(nodeName)
+	if err != nil {
+		return []metricsapi.NodeMetrics{}, err
+	}
+	resultRaw, err := GetHeapsterMetrics(cli, path, params)
+	if err != nil {
+		return []metricsapi.NodeMetrics{}, err
+	}
+	metrics := make([]metricsapi.NodeMetrics, 0)
+	if len(nodeName) == 0 {
+		metricsList := metricsapi.NodeMetricsList{}
+		err = json.Unmarshal(resultRaw, &metricsList)
+		if err != nil {
+			return []metricsapi.NodeMetrics{}, fmt.Errorf("failed to unmarshall heapster response: %v", err)
+		}
+		metrics = append(metrics, metricsList.Items...)
+	} else {
+		var singleMetric metricsapi.NodeMetrics
+		err = json.Unmarshal(resultRaw, &singleMetric)
+		if err != nil {
+			return []metricsapi.NodeMetrics{}, fmt.Errorf("failed to unmarshall heapster response: %v", err)
+		}
+		metrics = append(metrics, singleMetric)
+	}
+	return metrics, nil
+}
+
+func (cli *HeapsterMetricsClient) GetPodMetrics(namespace string, podName string, allNamespaces bool, selector labels.Selector) ([]metricsapi.PodMetrics, error) {
+	if allNamespaces {
+		namespace = metav1.NamespaceAll
+	}
+	path, err := podMetricsUrl(namespace, podName)
+	if err != nil {
+		return []metricsapi.PodMetrics{}, err
+	}
+
+	params := map[string]string{"labelSelector": selector.String()}
+	allMetrics := make([]metricsapi.PodMetrics, 0)
+
+	resultRaw, err := GetHeapsterMetrics(cli, path, params)
+	if err != nil {
+		return []metricsapi.PodMetrics{}, err
+	}
+	if len(podName) == 0 {
+		metrics := metricsapi.PodMetricsList{}
+		err = json.Unmarshal(resultRaw, &metrics)
+		if err != nil {
+			return []metricsapi.PodMetrics{}, fmt.Errorf("failed to unmarshall heapster response: %v", err)
+		}
+		allMetrics = append(allMetrics, metrics.Items...)
+	} else {
+		var singleMetric metricsapi.PodMetrics
+		err = json.Unmarshal(resultRaw, &singleMetric)
+		if err != nil {
+			return []metricsapi.PodMetrics{}, fmt.Errorf("failed to unmarshall heapster response: %v", err)
+		}
+		allMetrics = append(allMetrics, singleMetric)
+	}
+	return allMetrics, nil
+}
+
+func GetHeapsterMetrics(cli *HeapsterMetricsClient, path string, params map[string]string) ([]byte, error) {
+	return cli.SVCClient.Services(cli.HeapsterNamespace).
+		ProxyGet(cli.HeapsterScheme, cli.HeapsterService, cli.HeapsterPort, path, params).
+		DoRaw()
+}

vendor/k8s.io/kubernetes/pkg/kubectl/metricsutil/metrics_printer.go

@@ -0,0 +1,199 @@
+/*
+Copyright 2016 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package metricsutil
+
+import (
+	"fmt"
+	"io"
+	"sort"
+
+	"k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/resource"
+	"k8s.io/kubernetes/pkg/api/legacyscheme"
+	"k8s.io/kubernetes/pkg/printers"
+	metricsapi "k8s.io/metrics/pkg/apis/metrics/v1alpha1"
+)
+
+var (
+	MeasuredResources = []v1.ResourceName{
+		v1.ResourceCPU,
+		v1.ResourceMemory,
+	}
+	NodeColumns     = []string{"NAME", "CPU(cores)", "CPU%", "MEMORY(bytes)", "MEMORY%"}
+	PodColumns      = []string{"NAME", "CPU(cores)", "MEMORY(bytes)"}
+	NamespaceColumn = "NAMESPACE"
+	PodColumn       = "POD"
+)
+
+type ResourceMetricsInfo struct {
+	Name      string
+	Metrics   v1.ResourceList
+	Available v1.ResourceList
+}
+
+type TopCmdPrinter struct {
+	out io.Writer
+}
+
+func NewTopCmdPrinter(out io.Writer) *TopCmdPrinter {
+	return &TopCmdPrinter{out: out}
+}
+
+func (printer *TopCmdPrinter) PrintNodeMetrics(metrics []metricsapi.NodeMetrics, availableResources map[string]v1.ResourceList) error {
+	if len(metrics) == 0 {
+		return nil
+	}
+	w := printers.GetNewTabWriter(printer.out)
+	defer w.Flush()
+
+	sort.Slice(metrics, func(i, j int) bool {
+		return metrics[i].Name < metrics[j].Name
+	})
+
+	printColumnNames(w, NodeColumns)
+	var usage v1.ResourceList
+	for _, m := range metrics {
+		err := legacyscheme.Scheme.Convert(&m.Usage, &usage, nil)
+		if err != nil {
+			return err
+		}
+		printMetricsLine(w, &ResourceMetricsInfo{
+			Name:      m.Name,
+			Metrics:   usage,
+			Available: availableResources[m.Name],
+		})
+	}
+	return nil
+}
+
+func (printer *TopCmdPrinter) PrintPodMetrics(metrics []metricsapi.PodMetrics, printContainers bool, withNamespace bool) error {
+	if len(metrics) == 0 {
+		return nil
+	}
+	w := printers.GetNewTabWriter(printer.out)
+	defer w.Flush()
+
+	if withNamespace {
+		printValue(w, NamespaceColumn)
+	}
+	if printContainers {
+		printValue(w, PodColumn)
+	}
+
+	sort.Slice(metrics, func(i, j int) bool {
+		if withNamespace && metrics[i].Namespace != metrics[j].Namespace {
+			return metrics[i].Namespace < metrics[j].Namespace
+		}
+		return metrics[i].Name < metrics[j].Name
+	})
+
+	printColumnNames(w, PodColumns)
+	for _, m := range metrics {
+		err := printSinglePodMetrics(w, &m, printContainers, withNamespace)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func printColumnNames(out io.Writer, names []string) {
+	for _, name := range names {
+		printValue(out, name)
+	}
+	fmt.Fprint(out, "\n")
+}
+
+func printSinglePodMetrics(out io.Writer, m *metricsapi.PodMetrics, printContainersOnly bool, withNamespace bool) error {
+	containers := make(map[string]v1.ResourceList)
+	podMetrics := make(v1.ResourceList)
+	for _, res := range MeasuredResources {
+		podMetrics[res], _ = resource.ParseQuantity("0")
+	}
+
+	for _, c := range m.Containers {
+		var usage v1.ResourceList
+		err := legacyscheme.Scheme.Convert(&c.Usage, &usage, nil)
+		if err != nil {
+			return err
+		}
+		containers[c.Name] = usage
+		if !printContainersOnly {
+			for _, res := range MeasuredResources {
+				quantity := podMetrics[res]
+				quantity.Add(usage[res])
+				podMetrics[res] = quantity
+			}
+		}
+	}
+	if printContainersOnly {
+		for contName := range containers {
+			if withNamespace {
+				printValue(out, m.Namespace)
+			}
+			printValue(out, m.Name)
+			printMetricsLine(out, &ResourceMetricsInfo{
+				Name:      contName,
+				Metrics:   containers[contName],
+				Available: v1.ResourceList{},
+			})
+		}
+	} else {
+		if withNamespace {
+			printValue(out, m.Namespace)
+		}
+		printMetricsLine(out, &ResourceMetricsInfo{
+			Name:      m.Name,
+			Metrics:   podMetrics,
+			Available: v1.ResourceList{},
+		})
+	}
+	return nil
+}
+
+func printMetricsLine(out io.Writer, metrics *ResourceMetricsInfo) {
+	printValue(out, metrics.Name)
+	printAllResourceUsages(out, metrics)
+	fmt.Fprint(out, "\n")
+}
+
+func printValue(out io.Writer, value interface{}) {
+	fmt.Fprintf(out, "%v\t", value)
+}
+
+func printAllResourceUsages(out io.Writer, metrics *ResourceMetricsInfo) {
+	for _, res := range MeasuredResources {
+		quantity := metrics.Metrics[res]
+		printSingleResourceUsage(out, res, quantity)
+		fmt.Fprint(out, "\t")
+		if available, found := metrics.Available[res]; found {
+			fraction := float64(quantity.MilliValue()) / float64(available.MilliValue()) * 100
+			fmt.Fprintf(out, "%d%%\t", int64(fraction))
+		}
+	}
+}
+
+func printSingleResourceUsage(out io.Writer, resourceType v1.ResourceName, quantity resource.Quantity) {
+	switch resourceType {
+	case v1.ResourceCPU:
+		fmt.Fprintf(out, "%vm", quantity.MilliValue())
+	case v1.ResourceMemory:
+		fmt.Fprintf(out, "%vMi", quantity.Value()/(1024*1024))
+	default:
+		fmt.Fprintf(out, "%v", quantity.Value())
+	}
+}