Add limit field to WorkloadUtilization and implement resolver (#111)

Author: rbjornstad

internal/graph/gengql/generated.go

@@ -31,6 +31,9 @@ type WorkloadUtilization {
 	"Gets the requested amount of resources for the requested resource type."
 	requested(resourceType: UtilizationResourceType!): Float!
 
+	"Gets the limit of the resources for the requested resource type."
+	limit(resourceType: UtilizationResourceType!): Float
+
 	"Usage between start and end with step size for given resource type."
 	series(input: WorkloadUtilizationSeriesInput!): [UtilizationSample!]!
 }

internal/graph/schema/utilization.graphqls

@@ -53,6 +53,10 @@ func (r *workloadUtilizationResolver) Requested(ctx context.Context, obj *utiliz
 	return utilization.WorkloadResourceRequest(ctx, obj.EnvironmentName, obj.TeamSlug, obj.WorkloadName, resourceType)
 }
 
+func (r *workloadUtilizationResolver) Limit(ctx context.Context, obj *utilization.WorkloadUtilization, resourceType utilization.UtilizationResourceType) (*float64, error) {
+	return utilization.WorkloadResourceLimit(ctx, obj.EnvironmentName, obj.TeamSlug, obj.WorkloadName, resourceType)
+}
+
 func (r *workloadUtilizationResolver) Series(ctx context.Context, obj *utilization.WorkloadUtilization, input utilization.WorkloadUtilizationSeriesInput) ([]*utilization.UtilizationSample, error) {
 	return utilization.WorkloadResourceUsageRange(ctx, obj.EnvironmentName, obj.TeamSlug, obj.WorkloadName, input.ResourceType, input.Start, input.End, input.Step())
 }

internal/graph/utilization.resolvers.go

@@ -14,8 +14,10 @@ import (
 )
 
 const (
+	appCPULimit        = `sum by (container) (kube_pod_container_resource_limits{namespace=%q, container=%q, resource="cpu", unit="core"})`
 	appCPURequest      = `sum by (container) (kube_pod_container_resource_requests{namespace=%q, container=%q, resource="cpu",unit="core"})`
 	appCPUUsage        = `sum by (container) (rate(container_cpu_usage_seconds_total{namespace=%q, container=%q}[5m]))`
+	appMemoryLimit     = `sum by (container) (kube_pod_container_resource_limits{namespace=%q, container=%q, resource="memory", unit="byte"})`
 	appMemoryRequest   = `sum by (container) (kube_pod_container_resource_requests{namespace=%q, container=%q, resource="memory",unit="byte"})`
 	appMemoryUsage     = `sum by (container) (container_memory_working_set_bytes{namespace=%q, container=%q})`
 	teamCPURequest     = `sum by (container, owner_kind) (kube_pod_container_resource_requests{namespace=%q, container!~%q, resource="cpu",unit="core"} * on(pod,namespace) group_left(owner_kind) kube_pod_owner{owner_kind="ReplicaSet"})`
@@ -154,6 +156,26 @@ func WorkloadResourceRequest(ctx context.Context, env string, teamSlug slug.Slug
 	return ensuredVal(v), nil
 }
 
+func WorkloadResourceLimit(ctx context.Context, env string, teamSlug slug.Slug, workloadName string, resourceType UtilizationResourceType) (*float64, error) {
+	q := appMemoryLimit
+	if resourceType == UtilizationResourceTypeCPU {
+		q = appCPULimit
+	}
+
+	c := fromContext(ctx).client
+
+	v, err := c.query(ctx, env, fmt.Sprintf(q, teamSlug, workloadName))
+	if err != nil {
+		return nil, err
+	}
+
+	if len(v) == 0 {
+		return nil, nil
+	}
+
+	return (*float64)(&v[0].Value), nil
+}
+
 func WorkloadResourceUsage(ctx context.Context, env string, teamSlug slug.Slug, workloadName string, resourceType UtilizationResourceType) (float64, error) {
 	q := appMemoryUsage
 	if resourceType == UtilizationResourceTypeCPU {