feat(parser): 支持多 SQL (#942)

Author: qianmoQ

core/datacap-parser/src/main/java/io/edurt/datacap/sql/SQLParser.java

@@ -7,9 +7,18 @@
 import org.antlr.v4.runtime.CommonTokenStream;
 import org.antlr.v4.runtime.tree.ParseTree;
 
+import java.util.ArrayList;
+import java.util.List;
+
 public class SQLParser
 {
     public static SQLStatement parse(String sql)
+    {
+        List<SQLStatement> statements = parseMultiple(sql);
+        return statements.isEmpty() ? null : statements.get(0);
+    }
+
+    public static List<SQLStatement> parseMultiple(String sql)
     {
         try {
             // 创建词法分析器和语法分析器
@@ -23,17 +32,32 @@ public static SQLStatement parse(String sql)
             parser.removeErrorListeners();
             parser.addErrorListener(new SQLParserErrorListener());
 
-            // 获取解析树
-            // Get parse tree
-            ParseTree tree = parser.singleStatement();
+            // 获取解析树 - 使用 sqlStatements 而不是 statement
+            // Get parse tree - use sqlStatements instead of statement
+            ParseTree tree = parser.sqlStatements();
 
             // 访问解析树
             // Visit parse tree
             SQLVisitor visitor = new SQLVisitor();
-            return visitor.visit(tree);
+
+            List<SQLStatement> statements = new ArrayList<>();
+
+            // 遍历所有的 singleStatement
+            // Traverse all singleStatement
+            for (int i = 0; i < tree.getChildCount(); i++) {
+                ParseTree child = tree.getChild(i);
+                if (child instanceof SqlBaseParser.SingleStatementContext) {
+                    SQLStatement stmt = visitor.visit(child);
+                    if (stmt != null) {
+                        statements.add(stmt);
+                    }
+                }
+            }
+
+            return statements;
         }
         catch (Exception e) {
             throw new SQLParseException("Failed to parse SQL: " + e.getMessage(), e);
         }
     }
-}
+}

core/datacap-ui/src/views/pages/admin/source/SourceInfo.vue

@@ -287,6 +287,11 @@ export default defineComponent({
           const pluginsResponse = await getPluginsPromise
           if (pluginsResponse.status) {
             this.plugins = pluginsResponse.data
+
+            const hasJsonConvert = this.plugins.filter(value => value.type === 'Convert' && value.name === 'JsonConvert').length > 0
+            if (!hasJsonConvert) {
+              this.testInfo.message = 'JsonConvert plugin is required'
+            }
           }
           this.formState = SourceRequest.of()
         }

test/datacap-test-parser/src/test/java/io/edurt/datacap/test/basic/CreateStatementTest.java

@@ -59,12 +59,52 @@ public void testFlinkSql()
                 "  COL4\n" +
                 "FROM\n" +
                 "  transform_table;";
-        SQLStatement stmt = SQLParser.parse(sql);
 
+        // 解析多个语句
+        // Parse multiple statements
+        List<SQLStatement> statements = SQLParser.parseMultiple(sql);
+
+        System.out.println("Total statements parsed: " + statements.size());
+
+        for (int i = 0; i < statements.size(); i++) {
+            SQLStatement stmt = statements.get(i);
+            System.out.println("Statement " + (i + 1) + ": " + stmt.getClass().getSimpleName());
+
+            if (stmt instanceof CreateTableStatement) {
+                CreateTableStatement createStmt = (CreateTableStatement) stmt;
+                System.out.println("Table name: " + createStmt.getTableName());
+                List<TableOption> options = createStmt.getOptions();
+                System.out.println("Options: " + options);
+                System.out.println("---");
+            }
+        }
+    }
+
+    @Test
+    public void testSingleCreateTable()
+    {
+        String singleSql = "CREATE TABLE source_table (\n" +
+                "  COL1 INT,\n" +
+                "  COL2 TIMESTAMP(0),\n" +
+                "  COL3 VARCHAR(14),\n" +
+                "  COL4 CHAR(1),\n" +
+                "  PRIMARY KEY (COL1) NOT ENFORCED\n" +
+                ") WITH\n" +
+                "  (\n" +
+                "    'connector' = 'mysql',\n" +
+                "    'hostname' = '127.0.0.1',\n" +
+                "    'port' = '3306',\n" +
+                "    'username' = 'root',\n" +
+                "    'password' = 'root',\n" +
+                "    'database' = 'default',\n" +
+                "    'table' = 'example'\n" +
+                "  );";
+
+        SQLStatement stmt = SQLParser.parse(singleSql);
         assertTrue(stmt instanceof CreateTableStatement);
         CreateTableStatement statement = (CreateTableStatement) stmt;
 
         List<TableOption> options = statement.getOptions();
-        System.out.println(options);
+        System.out.println("Single statement options: " + options);
     }
-}
+}

test/datacap-test-parser/src/test/java/io/edurt/datacap/test/basic/SelectStatementTest.java

@@ -1,6 +1,7 @@
 package io.edurt.datacap.test.basic;
 
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
+import io.edurt.datacap.sql.SQLParseException;
 import io.edurt.datacap.sql.SQLParser;
 import io.edurt.datacap.sql.node.Expression;
 import io.edurt.datacap.sql.node.clause.JoinClause;
@@ -146,7 +147,7 @@ public void testSelectWithCaseWhen()
         assertEquals("age_group", selectElements.get(1).getAlias());
     }
 
-    @Test(expected = RuntimeException.class)
+    @Test(expected = SQLParseException.class)
     public void testInvalidSQL()
     {
         String sql = "SELECT * FORM users";