Add glide.yaml and vendor deps

vendor/k8s.io/kubernetes/pkg/kubectl/cmd/top_pod.go

@@ -0,0 +1,194 @@
+/*
+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 cmd
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"time"
+
+	"k8s.io/kubernetes/pkg/api"
+	coreclient "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset/typed/core/internalversion"
+	"k8s.io/kubernetes/pkg/kubectl/cmd/templates"
+	cmdutil "k8s.io/kubernetes/pkg/kubectl/cmd/util"
+	"k8s.io/kubernetes/pkg/kubectl/metricsutil"
+	"k8s.io/kubernetes/pkg/labels"
+
+	"github.com/golang/glog"
+	"github.com/spf13/cobra"
+)
+
+type TopPodOptions struct {
+	ResourceName    string
+	Namespace       string
+	Selector        string
+	AllNamespaces   bool
+	PrintContainers bool
+	PodClient       coreclient.PodsGetter
+	HeapsterOptions HeapsterTopOptions
+	Client          *metricsutil.HeapsterMetricsClient
+	Printer         *metricsutil.TopCmdPrinter
+}
+
+const metricsCreationDelay = 2 * time.Minute
+
+var (
+	topPodLong = templates.LongDesc(`
+		Display Resource (CPU/Memory/Storage) usage of pods.
+
+		The 'top pod' command allows you to see the resource consumption of pods.
+
+		Due to the metrics pipeline delay, they may be unavailable for a few minutes
+		since pod creation.`)
+
+	topPodExample = templates.Examples(`
+		# Show metrics for all pods in the default namespace
+		kubectl top pod
+
+		# Show metrics for all pods in the given namespace
+		kubectl top pod --namespace=NAMESPACE
+
+		# Show metrics for a given pod and its containers
+		kubectl top pod POD_NAME --containers
+
+		# Show metrics for the pods defined by label name=myLabel
+		kubectl top pod -l name=myLabel`)
+)
+
+func NewCmdTopPod(f cmdutil.Factory, out io.Writer) *cobra.Command {
+	options := &TopPodOptions{}
+
+	cmd := &cobra.Command{
+		Use:     "pod [NAME | -l label]",
+		Short:   "Display Resource (CPU/Memory/Storage) usage of pods",
+		Long:    topPodLong,
+		Example: topPodExample,
+		Run: func(cmd *cobra.Command, args []string) {
+			if err := options.Complete(f, cmd, args, out); err != nil {
+				cmdutil.CheckErr(err)
+			}
+			if err := options.Validate(); err != nil {
+				cmdutil.CheckErr(cmdutil.UsageError(cmd, err.Error()))
+			}
+			if err := options.RunTopPod(); err != nil {
+				cmdutil.CheckErr(err)
+			}
+		},
+		Aliases: []string{"pods"},
+	}
+	cmd.Flags().StringVarP(&options.Selector, "selector", "l", "", "Selector (label query) to filter on")
+	cmd.Flags().BoolVar(&options.PrintContainers, "containers", false, "If present, print usage of containers within a pod.")
+	cmd.Flags().BoolVar(&options.AllNamespaces, "all-namespaces", false, "If present, list the requested object(s) across all namespaces. Namespace in current context is ignored even if specified with --namespace.")
+	options.HeapsterOptions.Bind(cmd.Flags())
+	return cmd
+}
+
+func (o *TopPodOptions) Complete(f cmdutil.Factory, cmd *cobra.Command, args []string, out io.Writer) error {
+	var err error
+	if len(args) == 1 {
+		o.ResourceName = args[0]
+	} else if len(args) > 1 {
+		return cmdutil.UsageError(cmd, cmd.Use)
+	}
+
+	o.Namespace, _, err = f.DefaultNamespace()
+	if err != nil {
+		return err
+	}
+	clientset, err := f.ClientSet()
+	if err != nil {
+		return err
+	}
+	o.PodClient = clientset.Core()
+	o.Client = metricsutil.NewHeapsterMetricsClient(clientset.Core(), o.HeapsterOptions.Namespace, o.HeapsterOptions.Scheme, o.HeapsterOptions.Service, o.HeapsterOptions.Port)
+	o.Printer = metricsutil.NewTopCmdPrinter(out)
+	return nil
+}
+
+func (o *TopPodOptions) Validate() error {
+	if len(o.ResourceName) > 0 && len(o.Selector) > 0 {
+		return errors.New("only one of NAME or --selector can be provided")
+	}
+	return nil
+}
+
+func (o TopPodOptions) RunTopPod() error {
+	var err error
+	selector := labels.Everything()
+	if len(o.Selector) > 0 {
+		selector, err = labels.Parse(o.Selector)
+		if err != nil {
+			return err
+		}
+	}
+	metrics, err := o.Client.GetPodMetrics(o.Namespace, o.ResourceName, o.AllNamespaces, selector)
+	// TODO: Refactor this once Heapster becomes the API server.
+	// First we check why no metrics have been received.
+	if len(metrics) == 0 {
+		// If the API server query is successful but all the pods are newly created,
+		// the metrics are probably not ready yet, so we return the error here in the first place.
+		e := verifyEmptyMetrics(o, selector)
+		if e != nil {
+			return e
+		}
+	}
+	if err != nil {
+		return err
+	}
+	return o.Printer.PrintPodMetrics(metrics, o.PrintContainers, o.AllNamespaces)
+}
+
+func verifyEmptyMetrics(o TopPodOptions, selector labels.Selector) error {
+	if len(o.ResourceName) > 0 {
+		pod, err := o.PodClient.Pods(o.Namespace).Get(o.ResourceName)
+		if err != nil {
+			return err
+		}
+		if err := checkPodAge(pod); err != nil {
+			return err
+		}
+	} else {
+		pods, err := o.PodClient.Pods(o.Namespace).List(api.ListOptions{
+			LabelSelector: selector,
+		})
+		if err != nil {
+			return err
+		}
+		if len(pods.Items) == 0 {
+			return nil
+		}
+		for _, pod := range pods.Items {
+			if err := checkPodAge(&pod); err != nil {
+				return err
+			}
+		}
+	}
+	return errors.New("metrics not available yet")
+}
+
+func checkPodAge(pod *api.Pod) error {
+	age := time.Since(pod.CreationTimestamp.Time)
+	if age > metricsCreationDelay {
+		message := fmt.Sprintf("Metrics not available for pod %s/%s, age: %s", pod.Namespace, pod.Name, age.String())
+		glog.Warningf(message)
+		return errors.New(message)
+	} else {
+		glog.V(2).Infof("Metrics not yet available for pod %s/%s, age: %s", pod.Namespace, pod.Name, age.String())
+		return nil
+	}
+}