#141 Writing calls per second.

philterd · Nov 16, 2024 · 89658ec · 89658ec
1 parent 731fd25
commit 89658ec
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/phileas-benchmark/src/test/java/ai/philterd/phileas/benchmarks/tests/BenchmarksTest.java b/phileas-benchmark/src/test/java/ai/philterd/phileas/benchmarks/tests/BenchmarksTest.java
@@ -33,7 +33,6 @@ static boolean benchmarksEnabled() {
     public void runBenchmarks() throws Exception {
 
         final String branch = getGitBranch();
-        final String runId = UUID.randomUUID().toString();
 
         final Collection<Result> results = new LinkedList<>();
 
@@ -82,6 +81,7 @@ public void runBenchmarks() throws Exception {
 
                 // calls_per_second
                 final Result result = new Result();
+                result.setRunId(UUID.randomUUID().toString());
                 result.setCallsPerSecond(calls);
                 result.setDocument(document);
                 result.setWorkloadMillis(workload_millis);
@@ -118,8 +118,21 @@ INSERT INTO benchmarks(document, workload_mills, redactor, timestamp, phileas_ve
                     statement.setLong(4, result.getTimestamp());
                     statement.setString(5, result.getPhileasVersion());
                     statement.setString(6, result.getBranch());
-                    statement.setString(7, runId);
+                    statement.setString(7, result.getRunId());
                     statement.executeUpdate();
+                }
+
+                for(final Integer i : result.getCallsPerSecond().keySet()) {
+
+                    try (PreparedStatement statement = connection.prepareStatement("""
+                    INSERT INTO calls_per_second(length, calls_per_second, run_id)
+                    VALUES (?, ?, ?)
+                  """)) {
+                        statement.setInt(1, i);
+                        statement.setLong(2, result.getCallsPerSecond().get(i));
+                        statement.setString(3, result.getRunId());
+                        statement.executeUpdate();
+                    }
 
                 }