Add AST Visualization

manticore-projects · manticore-projects · commit 520e1e658ba8 · 2022-08-18T18:34:52.000+07:00
Show the Statement's Java Objects in a tree hierarchy
diff --git a/src/test/java/net/sf/jsqlparser/test/AssortedFeatureTests.java b/src/test/java/net/sf/jsqlparser/test/AssortedFeatureTests.java
@@ -0,0 +1,60 @@
+/*-
+ * #%L
+ * JSQLParser library
+ * %%
+ * Copyright (C) 2004 - 2022 JSQLParser
+ * %%
+ * Dual licensed under GNU LGPL 2.1 or Apache License 2.0
+ * #L%
+ */
+package net.sf.jsqlparser.test;
+
+import net.sf.jsqlparser.JSQLParserException;
+import net.sf.jsqlparser.expression.LongValue;
+import net.sf.jsqlparser.expression.StringValue;
+import net.sf.jsqlparser.parser.CCJSqlParserUtil;
+import net.sf.jsqlparser.statement.Statement;
+import net.sf.jsqlparser.util.deparser.ExpressionDeParser;
+import net.sf.jsqlparser.util.deparser.SelectDeParser;
+import net.sf.jsqlparser.util.deparser.StatementDeParser;
+import org.junit.jupiter.api.Test;
+
+public class AssortedFeatureTests {
+
+    static class ReplaceColumnAndLongValues extends ExpressionDeParser {
+
+        @Override
+        public void visit(StringValue stringValue) {
+            this.getBuffer().append("?");
+        }
+
+        @Override
+        public void visit(LongValue longValue) {
+            this.getBuffer().append("?");
+        }
+    }
+
+    public static String cleanStatement(String sql) throws JSQLParserException {
+        StringBuilder buffer = new StringBuilder();
+        ExpressionDeParser expr = new ReplaceColumnAndLongValues();
+
+        SelectDeParser selectDeparser = new SelectDeParser(expr, buffer);
+        expr.setSelectVisitor(selectDeparser);
+        expr.setBuffer(buffer);
+
+        StatementDeParser stmtDeparser = new StatementDeParser(expr, selectDeparser, buffer);
+
+        Statement stmt = CCJSqlParserUtil.parse(sql);
+
+        stmt.accept(stmtDeparser);
+        return stmtDeparser.getBuffer().toString();
+    }
+
+    @Test
+    public void testIssue1608() throws JSQLParserException {
+        System.out.println(cleanStatement("SELECT 'abc', 5 FROM mytable WHERE col='test'"));
+        System.out.println(cleanStatement("UPDATE table1 A SET A.columna = 'XXX' WHERE A.cod_table = 'YYY'"));
+        System.out.println(cleanStatement("INSERT INTO example (num, name, address, tel) VALUES (1, 'name', 'test ', '1234-1234')"));
+        System.out.println(cleanStatement("DELETE FROM table1 where col=5 and col2=4"));
+    }
+}
