SqlRecorder: added dumpQueryBuffer, better line breaks for multi-SQL

Evolveum · Jul 15, 2022 · 886364c · 886364c
1 parent 5b969a2
commit 886364c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/repo/repo-sqale/src/test/java/com/evolveum/midpoint/repo/sqale/SqaleRepoBaseTest.java b/repo/repo-sqale/src/test/java/com/evolveum/midpoint/repo/sqale/SqaleRepoBaseTest.java
@@ -427,7 +427,7 @@ boolean record = !queryRecorder.isRecording();
         } finally {
             if (record) {
                 queryRecorder.stopRecording();
-                display(queryRecorder.getQueryBuffer().toString());
+                display(queryRecorder.dumpQueryBuffer());
             }
         }
     }
@@ -524,7 +524,7 @@ boolean record = !queryRecorder.isRecording();
         } finally {
             if (record) {
                 queryRecorder.stopRecording();
-                display(queryRecorder.getQueryBuffer().toString());
+                display(queryRecorder.dumpQueryBuffer());
             }
         }
     }
@@ -679,7 +679,7 @@ protected void withQueryRecorded(CheckedRunnable block) {
             throw new TestException(e);
         } finally {
             queryRecorder.stopRecording();
-            display(queryRecorder.getQueryBuffer().toString());
+            display(queryRecorder.dumpQueryBuffer());
         }
     }
 }
diff --git a/repo/repo-sqlbase/src/main/java/com/evolveum/midpoint/repo/sqlbase/querydsl/SqlRecorder.java b/repo/repo-sqlbase/src/main/java/com/evolveum/midpoint/repo/sqlbase/querydsl/SqlRecorder.java
@@ -102,6 +102,12 @@ public synchronized Queue<QueryEntry> getQueryBuffer() {
         return new ArrayDeque<>(queryBuffer);
     }
 
+    public synchronized String dumpQueryBuffer() {
+        StringBuilder sb = new StringBuilder("QUERIES:\n");
+        queryBuffer.forEach(q -> sb.append(q).append('\n'));
+        return sb.toString();
+    }
+
     public synchronized void clearBuffer() {
         queryBuffer.clear();
     }