Add option to show only pods exceeding requests (#9)

cmd/cmd.go

@@ -22,6 +22,7 @@ var (
 	showNodes          = false
 	verbose            = false
 	aggregation        = "POD"
+	onlyWarnings       = false
 )
 
 func init() {
@@ -46,6 +47,13 @@ func init() {
 		verbose,
 		"show full resource names",
 	)
+	rootCmd.PersistentFlags().BoolVarP(
+		&onlyWarnings,
+		"warnings",
+		"w",
+		onlyWarnings,
+		"only show resources using excessive resources",
+	)
 	rootCmd.PersistentFlags().StringVarP(
 		&aggregation,
 		"by",
@@ -75,6 +83,7 @@ var rootCmd = &cobra.Command{
 			Verbose:            verbose,
 			ShowNodes:          showNodes,
 			ColoredOutput:      color,
+			OnlyWarnings:       onlyWarnings,
 		}
 		return client.Run(args)
 	},

pkg/model/args.go

@@ -13,6 +13,7 @@ type Args struct {
 	Verbose            bool
 	ShowNodes          bool
 	ColoredOutput      bool
+	OnlyWarnings       bool
 }
 
 type Aggregation int

pkg/model/podresource.go

@@ -47,6 +47,13 @@ func (p *PodResource) Memory() *Resource {
 	return res
 }
 
+func (r *Resource) ExceedsRequest() bool {
+	if r.Request == 0 {
+		return false
+	}
+	return r.Usage > r.Request
+}
+
 func MergePodResources(resources ...map[string]*PodResource) (map[string]*PodResource, error) {
 	merged := map[string]*PodResource{}
 	for _, resource := range resources {

pkg/writer/writer.go

@@ -22,6 +22,9 @@ func Write(response map[string]*model.PodResource, args *model.Args) error {
 
 	var allRows []*ResourceRow
 	for _, pod := range resources {
+		if args.OnlyWarnings && !(pod.Cpu().ExceedsRequest() || pod.Memory().ExceedsRequest()) {
+			continue
+		}
 		allRows = append(allRows, PodToRows(pod)...)
 	}