feat: implement PostgresStatsStore querying continuous aggregates

2026-03-16 18:22:44 +01:00
parent 9fd02c4edb
commit 527e2cf017
2 changed files with 248 additions and 0 deletions
--- a/cameleer3-server-app/src/test/java/com/cameleer3/server/app/storage/PostgresStatsStoreIT.java
+++ b/cameleer3-server-app/src/test/java/com/cameleer3/server/app/storage/PostgresStatsStoreIT.java
@@ -0,0 +1,61 @@
+package com.cameleer3.server.app.storage;
+
+import com.cameleer3.server.app.AbstractPostgresIT;
+import com.cameleer3.server.core.search.ExecutionStats;
+import com.cameleer3.server.core.search.StatsTimeseries;
+import com.cameleer3.server.core.storage.ExecutionStore;
+import com.cameleer3.server.core.storage.ExecutionStore.ExecutionRecord;
+import com.cameleer3.server.core.storage.StatsStore;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.jdbc.core.JdbcTemplate;
+
+import java.time.Instant;
+import java.time.temporal.ChronoUnit;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class PostgresStatsStoreIT extends AbstractPostgresIT {
+
+    @Autowired StatsStore statsStore;
+    @Autowired ExecutionStore executionStore;
+    @Autowired JdbcTemplate jdbc;
+
+    @Test
+    void statsReturnsCountsForTimeWindow() {
+        Instant now = Instant.now().truncatedTo(ChronoUnit.SECONDS);
+        insertExecution("stats-1", "route-a", "app-1", "COMPLETED", now, 100L);
+        insertExecution("stats-2", "route-a", "app-1", "FAILED", now.plusSeconds(10), 200L);
+        insertExecution("stats-3", "route-b", "app-1", "COMPLETED", now.plusSeconds(20), 50L);
+
+        // Force continuous aggregate refresh
+        jdbc.execute("CALL refresh_continuous_aggregate('stats_1m_all', null, null)");
+
+        ExecutionStats stats = statsStore.stats(now.minusSeconds(60), now.plusSeconds(60));
+        assertEquals(3, stats.totalCount());
+        assertEquals(1, stats.failedCount());
+    }
+
+    @Test
+    void timeseriesReturnsBuckets() {
+        Instant now = Instant.now().truncatedTo(ChronoUnit.MINUTES);
+        for (int i = 0; i < 10; i++) {
+            insertExecution("ts-" + i, "route-a", "app-1", "COMPLETED",
+                    now.plusSeconds(i * 30), 100L + i);
+        }
+
+        jdbc.execute("CALL refresh_continuous_aggregate('stats_1m_all', null, null)");
+
+        StatsTimeseries ts = statsStore.timeseries(now.minusMinutes(1), now.plusMinutes(10), 5);
+        assertNotNull(ts);
+        assertFalse(ts.buckets().isEmpty());
+    }
+
+    private void insertExecution(String id, String routeId, String groupName,
+                                  String status, Instant startTime, long durationMs) {
+        executionStore.upsert(new ExecutionRecord(
+                id, routeId, "agent-1", groupName, status, null, null,
+                startTime, startTime.plusMillis(durationMs), durationMs,
+                status.equals("FAILED") ? "error" : null, null, null));
+    }
+}