bulk advise on all pods

Author: jasonhawkharris

internal/scout/advise/advise.go

@@ -4,6 +4,13 @@ import "github.com/sourcegraph/src-cli/internal/scout"
 
 type Option = func(config *scout.Config)
 
+const (
+	OVER_100 = "\t%s %s: Your %s usage is over 100%% (%.2f%%). Add more %s."
+	OVER_80  = "\t%s %s: Your %s usage is over 80%% (%.2f%%). Consider raising limits."
+	OVER_40  = "\t%s %s: Your %s usage is under 80%% (%.2f%%). Keep %s allocation the same."
+	UNDER_40 = "\t%s %s: Your %s usage is under 40%% (%.2f%%). Consider lowering limits."
+)
+
 func WithNamespace(namespace string) Option {
 	return func(config *scout.Config) {
 		config.Namespace = namespace

internal/scout/advise/k8s.go

@@ -4,7 +4,10 @@ import (
 	"context"
 	"fmt"
 
+	"github.com/sourcegraph/sourcegraph/lib/errors"
 	"github.com/sourcegraph/src-cli/internal/scout"
+	"github.com/sourcegraph/src-cli/internal/scout/kube"
+	v1 "k8s.io/api/core/v1"
 	"k8s.io/client-go/kubernetes"
 	"k8s.io/client-go/rest"
 	metricsv "k8s.io/metrics/pkg/client/clientset/versioned"
@@ -33,23 +36,129 @@ func K8s(
 		opt(cfg)
 	}
 
-	/* pods, err := kube.GetPods(ctx, cfg)
+	pods, err := kube.GetPods(ctx, cfg)
 	if err != nil {
 		return errors.Wrap(err, "could not get list of pods")
-	} */
+	}
 
-	/* if cfg.Pod != "" {
+	if cfg.Pod != "" {
 		pod, err := kube.GetPod(cfg.Pod, pods)
 		if err != nil {
 			return errors.Wrap(err, "could not get pod")
 		}
 
-	} */
-	fmt.Println("advise.K8s needs code")
+		err = Advise(ctx, cfg, pod)
+		if err != nil {
+			return errors.Wrap(err, "could not advise")
+		}
+		return nil
+	}
+
+	for _, pod := range pods {
+		fmt.Println(scout.EmojiFingerPointRight, pod.Name)
+		err = Advise(ctx, cfg, pod)
+		if err != nil {
+			return errors.Wrap(err, "could not advise")
+		}
+	}
+
 	return nil
 }
 
-/* func Advise(ctx context.Context, cfg *scout.Config, pod v1.Pod) error {
-    fmt.Println(pod.Name)
-    return nil
-} */
+func Advise(ctx context.Context, cfg *scout.Config, pod v1.Pod) error {
+	var advice []string
+	usageMetrics, err := getUsageMetrics(ctx, cfg, pod)
+	if err != nil {
+		return errors.Wrap(err, "could not get usage metrics")
+	}
+
+	for _, metrics := range usageMetrics {
+		cpuAdvice := checkUsage(metrics.CpuUsage, "CPU", metrics.ContainerName, pod.Name)
+		advice = append(advice, cpuAdvice)
+
+		memoryAdvice := checkUsage(metrics.MemoryUsage, "memory", metrics.ContainerName, pod.Name)
+		advice = append(advice, memoryAdvice)
+
+		if metrics.Storage != nil {
+			storageAdvice := checkUsage(metrics.StorageUsage, "storage", metrics.ContainerName, pod.Name)
+			advice = append(advice, storageAdvice)
+		}
+
+		for _, msg := range advice {
+			fmt.Println(msg)
+		}
+	}
+
+	return nil
+}
+
+func getUsageMetrics(ctx context.Context, cfg *scout.Config, pod v1.Pod) ([]scout.UsageStats, error) {
+	var usages []scout.UsageStats
+	var usage scout.UsageStats
+	podMetrics, err := kube.GetPodMetrics(ctx, cfg, pod)
+	if err != nil {
+		return usages, errors.Wrap(err, "while attempting to fetch pod metrics")
+	}
+
+	containerMetrics := &scout.ContainerMetrics{
+		PodName: cfg.Pod,
+		Limits:  map[string]scout.Resources{},
+	}
+
+	if err = kube.GetLimits(ctx, cfg, &pod, containerMetrics); err != nil {
+		return usages, errors.Wrap(err, "failed to get get container metrics")
+	}
+
+	for _, container := range podMetrics.Containers {
+		usage, err = kube.GetUsage(ctx, cfg, *containerMetrics, pod, container)
+		if err != nil {
+			return usages, errors.Wrapf(err, "could not compile usages data for row: %s\n", container.Name)
+		}
+		usages = append(usages, usage)
+	}
+
+	return usages, nil
+}
+
+func checkUsage(usage float64, resourceType, container, pod string) string {
+	var message string
+
+	switch {
+	case usage >= 100:
+		message = fmt.Sprintf(
+			OVER_100,
+			scout.FlashingLightEmoji,
+			container,
+			resourceType,
+			usage,
+			resourceType,
+		)
+	case usage >= 80 && usage < 100:
+		message = fmt.Sprintf(
+			OVER_80,
+			scout.WarningSign,
+			container,
+			resourceType,
+			usage,
+		)
+	case usage >= 40 && usage < 80:
+		message = fmt.Sprintf(
+			OVER_40,
+			scout.SuccessEmoji,
+			container,
+			resourceType,
+			usage,
+			resourceType,
+		)
+	default:
+		message = fmt.Sprintf(
+			UNDER_40,
+			scout.WarningSign,
+			container,
+			usage,
+			resourceType,
+		)
+	}
+
+	return message
+}

internal/scout/constants.go

@@ -1,5 +1,11 @@
 package scout
 
 const (
-	ABillion float64 = 1000000000
+	ABillion              float64 = 1_000_000_000
+	EmojiFingerPointRight         = "👉"
+	FailureEmoji                  = "🛑"
+	FlashingLightEmoji            = "🚨"
+	HourglassEmoji                = "⌛"
+	SuccessEmoji                  = "✅"
+	WarningSign                   = "⚠️ " // why does this need an extra space to align?!?!
 )