feat: sync some useful syntax from antlr/grammars-v4

Author: liuxy0551

src/grammar/mysql/MySqlParser.g4

@@ -310,7 +310,7 @@ createTablespaceNdb
     ;
 
 createTrigger
-    : KW_CREATE ownerStatement? KW_TRIGGER ifNotExists? trigger_name=fullId triggerTime=(
+    : KW_CREATE ownerStatement? ifNotExists? KW_TRIGGER ifNotExists? trigger_name=fullId triggerTime=(
         KW_BEFORE
         | KW_AFTER
     ) triggerEvent=(KW_INSERT | KW_UPDATE | KW_DELETE) KW_ON tableName KW_FOR KW_EACH KW_ROW (
@@ -907,19 +907,6 @@ replaceStatement
     )? (('(' columnNames ')')? replaceStatementValuesOrSelectOrTable | setAssignmentList)
     ;
 
-// selectStatement
-//     : querySpecification lockClause?                   # simpleSelect
-//     | querySpecificationNointo lockClause? intoClause? # simpleSelect
-//     | queryExpression lockClause?                      # parenthesisSelect
-//     | querySpecificationNointo unionStatement+ (
-//         KW_UNION unionType=(KW_ALL | KW_DISTINCT)? (querySpecification | queryExpression)
-//     )? orderByClause? limitClause? lockClause? # unionSelect
-//     | queryExpressionNointo unionParenthesis+ (
-//         KW_UNION unionType=(KW_ALL | KW_DISTINCT)? queryExpression
-//     )? orderByClause? limitClause? lockClause?         # unionParenthesisSelect
-//     | querySpecificationNointo (',' lateralStatement)+ # withLateralStatement
-//     ;
-
 // TODO: Simplify the rules to fit SLL(*) Mode
 selectStatement
     : querySpecification unionStatement* (
@@ -1127,6 +1114,7 @@ queryExpression
  */
 querySpecification
     : KW_SELECT selectSpec* selectElements intoClause? fromClause groupByClause? havingClause? windowClause? orderByClause? limitClause? intoClause?
+        unionStatement?
     ;
 
 unionStatement
@@ -3421,6 +3409,7 @@ keywordsCanBeId
     | KW_RESOURCE_GROUP_USER
     | KW_RESUME
     | KW_RETURNED_SQLSTATE
+    | KW_RETURNING
     | KW_RETURNS
     | KW_REUSE
     | KW_ROLE

test/parser/mysql/suggestion/tokenSuggestion.test.ts

@@ -60,6 +60,7 @@ describe('MySQL Token Suggestion', () => {
             'ALGORITHM',
             'OR',
             'TRIGGER',
+            'IF',
             'TABLESPACE',
             'UNDO',
             'TABLE',

test/parser/mysql/syntax/fixtures/alterTable.sql

@@ -102,3 +102,6 @@ ALTER TABLE t1 ALTER COLUMN c1 DROP DEFAULT, DROP CONSTRAINT c2_positive, ALTER
 ALTER TABLE t1 ALTER COLUMN c1 DROP DEFAULT, DROP CONSTRAINT c2_positive, ALTER CHECK c2_positive NOT ENFORCED REBUILD PARTITION p0 REBUILD PARTITION ALL;
 ALTER TABLE t1 ALTER COLUMN c1 DROP DEFAULT, DROP CONSTRAINT c2_positive, ALTER CHECK c2_positive NOT ENFORCED REPAIR PARTITION p0 REPAIR PARTITION ALL;
 ALTER TABLE t1 ALTER COLUMN c1 SET VISIBLE, DROP CONSTRAINT c2_positive, ALTER CHECK c2_positive NOT ENFORCED REMOVE PARTITIONING UPGRADE PARTITIONING;
+
+
+ALTER TABLE `order` ADD cancelled TINYINT(1) DEFAULT 0 NOT NULL, ADD delivered TINYINT(1) DEFAULT 0 NOT NULL, ADD returning TINYINT(1) DEFAULT 0 NOT NULL;

test/parser/mysql/syntax/fixtures/createTrigger.sql

@@ -20,3 +20,5 @@ CREATE TRIGGER my_trigger AFTER DELETE ON your_table FOR EACH ROW SET NEW.column
 
 CREATE TRIGGER my_trigger AFTER INSERT ON your_table FOR EACH ROW FOLLOWS other_trigger_name SET NEW.column_name = some_value;
 CREATE TRIGGER my_trigger AFTER INSERT ON your_table FOR EACH ROW PRECEDES other_trigger_name SET NEW.column_name = some_value;
+
+CREATE TRIGGER IF NOT EXISTS my_trigger AFTER INSERT ON your_table FOR EACH ROW PRECEDES other_trigger_name SET NEW.column_name = some_value;

test/parser/mysql/syntax/fixtures/select.sql

@@ -70,3 +70,11 @@ SELECT id, data INTO @x, @y FROM test.t1 LIMIT 1;
 SELECT * FROM t1 AS t INTO @x,@y,@z;
 SELECT * FROM t1 AS t INTO OUTFILE '/tmp/select-values.txt';
 SELECT a,b,a+b INTO OUTFILE '/tmp/result.txt' FIELDS TERMINATED BY ',' OPTIONALLY ENCLOSED BY '"' LINES TERMINATED BY '\n' FROM test_table;
+
+SELECT ID ,SUM(COL_1) AS SUM_COL_1 FROM (
+    (SELECT ID ,COL_1 FROM TEST_AUTO_INC
+    UNION ALL
+    SELECT ID ,COL_1 FROM TEST_AUTO_INC TAI)
+    UNION ALL
+    (SELECT ID ,COL_1 FROM TEST_AUTO_INC TAI)
+)SS GROUP BY 1 ORDER BY 1;