MetricsPacketsHandler supports Prometheus format

1 files changed, 25 insertions, 0 deletions

diff --git a/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java b/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java
index 019fc1e1e78..56efa396297 100644
--- a/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java
+++ b/container-core/src/test/java/com/yahoo/container/jdisc/state/MetricsPacketsHandlerTest.java
@@ -159,6 +159,31 @@ public class MetricsPacketsHandlerTest extends StateHandlerTestBase {
             if (dimensions.has("dim2")) assertDimension(packet, "host", HOST_DIMENSION);
         });
     }
+
+    @Test
+    public void prometheus_metrics() {
+        var context = StateMetricContext.newInstance(Map.of("dim1", "value1"));
+        var snapshot = new MetricSnapshot();
+        snapshot.set(context, "gauge.metric", 0.2);
+        snapshot.add(context, "counter.metric", 5);
+        snapshotProvider.setSnapshot(snapshot);
+        var response = requestAsString("http://localhost/metrics-packets?format=prometheus");
+        var expectedResponse = """
+                # HELP gauge_metric_last
+                # TYPE gauge_metric_last untyped
+                gauge_metric_last{dim1="value1",vespa_service="state-handler-test-base",} 0.2 0
+                # HELP gauge_metric_average
+                # TYPE gauge_metric_average untyped
+                gauge_metric_average{dim1="value1",vespa_service="state-handler-test-base",} 0.2 0
+                # HELP gauge_metric_max
+                # TYPE gauge_metric_max untyped
+                gauge_metric_max{dim1="value1",vespa_service="state-handler-test-base",} 0.2 0
+                # HELP counter_metric_count
+                # TYPE counter_metric_count untyped
+                counter_metric_count{dim1="value1",vespa_service="state-handler-test-base",} 5 0
+                """;
+        assertEquals(expectedResponse, response);
+    }
     
     private List<JsonNode> incrementTimeAndGetJsonPackets() throws Exception {
         advanceToNextSnapshot();