format query plan string

ahmed-kamal2004 · ahmed-kamal2004 · commit d3d02ff03658 · 2025-03-05T12:15:35.000+02:00
diff --git a/src/sqlancer/postgres/PostgresProvider.java b/src/sqlancer/postgres/PostgresProvider.java
@@ -360,7 +360,7 @@ public String getQueryPlan(String selectStr, PostgresGlobalState globalState) th
         } catch (SQLException | AssertionError e) {
             queryPlan = "";
         }
-        return queryPlan;
+        return formatQueryPlan(queryPlan);
     }
 
     @Override
@@ -385,4 +385,27 @@ protected boolean addRowsToAllTables(PostgresGlobalState globalState) throws Exc
         return true;
     }
 
+    private String formatQueryPlan(String queryPlan){
+        StringBuilder outQueryPlanFormatted = new StringBuilder();
+        boolean insideBrackets = false;
+
+        for(char ch: queryPlan.toCharArray()){
+            if(ch == '\n' || ch == ' '){
+                continue;
+            }
+            if(ch == '('){
+                insideBrackets = true;
+            }
+            else if(ch == ')'){
+                insideBrackets = false;
+                outQueryPlanFormatted.append(';');
+                continue;
+            }
+            if (!insideBrackets)
+                outQueryPlanFormatted.append(ch);
+        }
+
+        return outQueryPlanFormatted.toString();
+    }
+
 }
