added source label to ephemeral_storage_container_limit_percentage (#97)

Signed-off-by: Michal Minář <michal.minar@id.ethz.ch>

pkg/pod/metrics.go

@@ -76,6 +76,8 @@ func (cr Collector) createMetrics() {
 			"node_name",
 			// Name of container
 			"container",
+			// Source of the limit (either "container" for pod.spec.containers.resources.limits or "node")
+			"source",
 		},
 	)
 
@@ -164,10 +166,11 @@ func (cr Collector) SetMetrics(podName string, podNamespace string, nodeName str
 		if okPodResult {
 			for _, c := range podResult.containers {
 				labels := prometheus.Labels{"pod_namespace": podNamespace,
-					"pod_name": podName, "node_name": nodeName, "container": c.name}
+					"pod_name": podName, "node_name": nodeName, "container": c.name, "source": "node"}
 				if c.limit != 0 {
 					// Use Limit from Container
 					setValue = (usedBytes / c.limit) * 100.0
+					labels["source"] = "container"
 				} else if capacityBytes > 0. {
 					// Default to Node Used Ephemeral Storage
 					setValue = math.Max(capacityBytes-availableBytes, 0.) * 100.0 / capacityBytes

tests/e2e/deployment_test.go

@@ -2,8 +2,6 @@ package e2e
 
 import (
 	"fmt"
-	"github.com/onsi/ginkgo/v2"
-	"github.com/onsi/gomega"
 	"io"
 	"net/http"
 	"os/exec"
@@ -12,6 +10,9 @@ import (
 	"strings"
 	"testing"
 	"time"
+
+	"github.com/onsi/ginkgo/v2"
+	"github.com/onsi/gomega"
 )
 
 var httpClient *http.Client
@@ -120,9 +121,10 @@ func checkPrometheus(checkSlice []string, inverse bool) {
 
 func WatchContainerPercentage() {
 	status := 0
-	re := regexp.MustCompile(`ephemeral_storage_container_limit_percentage{container="grow-test",node_name="minikube".+,pod_namespace="ephemeral-metrics"}\s+(.+)`)
+	re := regexp.MustCompile(`ephemeral_storage_container_limit_percentage{container="grow-test",node_name="minikube".+,pod_namespace="ephemeral-metrics",source="container"}\s+(.+)`)
 	output := requestPrometheusString()
 	match := re.FindAllStringSubmatch(output, -1)
+	gomega.Expect(match).ShouldNot(gomega.BeEmpty())
 	floatValue, _ := strconv.ParseFloat(match[0][1], 64)
 	if floatValue < 100.0 {
 		status = 1