feat mysql alter force,engine,algorithm,lock (#2121)

minleejae · mj-db · web-flow · commit 18c1a2c63d97 · 2024-12-09T16:09:37.000+07:00
Co-authored-by: mj-db &lt;mj.db@kakaocorp.com&gt;
diff --git a/src/main/java/net/sf/jsqlparser/statement/alter/AlterExpression.java b/src/main/java/net/sf/jsqlparser/statement/alter/AlterExpression.java
@@ -61,6 +61,8 @@ public class AlterExpression implements Serializable {
     private String characterSet;
     private String collation;
     private String lockOption;
+    private String algorithmOption;
+    private String engineOption;
     private String commentText;
     private String tableOption;
 
@@ -438,6 +440,22 @@ public void setLockOption(String lockOption) {
         this.lockOption = lockOption;
     }
 
+    public String getAlgorithmOption() {
+        return algorithmOption;
+    }
+
+    public void setAlgorithmOption(String algorithmOption) {
+        this.algorithmOption = algorithmOption;
+    }
+
+    public String getEngineOption() {
+        return engineOption;
+    }
+
+    public void setEngineOption(String engineOption) {
+        this.engineOption = engineOption;
+    }
+
     public boolean getUseEqual() {
         return useEqual;
     }
@@ -478,6 +496,24 @@ public String toString() {
             b.append(optionalSpecifier);
         } else if (operation == AlterOperation.SET_TABLE_OPTION) {
             b.append(tableOption);
+        } else if (operation == AlterOperation.ENGINE) {
+            b.append("ENGINE ");
+            if (useEqual) {
+                b.append("= ");
+            }
+            b.append(engineOption);
+        } else if (operation == AlterOperation.ALGORITHM) {
+            b.append("ALGORITHM ");
+            if (useEqual) {
+                b.append("= ");
+            }
+            b.append(algorithmOption);
+        } else if (operation == AlterOperation.LOCK) {
+            b.append("LOCK ");
+            if (useEqual) {
+                b.append("= ");
+            }
+            b.append(lockOption);
         } else if (getOldIndex() != null) {
             b.append("RENAME");
             switch (operation) {
diff --git a/src/main/java/net/sf/jsqlparser/statement/alter/AlterOperation.java b/src/main/java/net/sf/jsqlparser/statement/alter/AlterOperation.java
@@ -10,7 +10,7 @@
 package net.sf.jsqlparser.statement.alter;
 
 public enum AlterOperation {
-    ADD, ALTER, DROP, DROP_PRIMARY_KEY, DROP_UNIQUE, DROP_FOREIGN_KEY, MODIFY, CHANGE, CONVERT, ALGORITHM, RENAME, RENAME_TABLE, RENAME_INDEX, RENAME_KEY, RENAME_CONSTRAINT, COMMENT, COMMENT_WITH_EQUAL_SIGN, UNSPECIFIC, ADD_PARTITION, DROP_PARTITION, TRUNCATE_PARTITION, SET_TABLE_OPTION, LOCK;
+    ADD, ALTER, DROP, DROP_PRIMARY_KEY, DROP_UNIQUE, DROP_FOREIGN_KEY, MODIFY, CHANGE, CONVERT, ALGORITHM, RENAME, RENAME_TABLE, RENAME_INDEX, RENAME_KEY, RENAME_CONSTRAINT, COMMENT, COMMENT_WITH_EQUAL_SIGN, UNSPECIFIC, ADD_PARTITION, DROP_PARTITION, TRUNCATE_PARTITION, SET_TABLE_OPTION, ENGINE, FORCE, LOCK;
 
     public static AlterOperation from(String operation) {
         return Enum.valueOf(AlterOperation.class, operation.toUpperCase());
diff --git a/src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt b/src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt
@@ -7167,7 +7167,6 @@ AlterExpression AlterExpression():
         (
             (
                 <K_ADD> { alterExp.setOperation(AlterOperation.ADD);
-                        System.out.println("test");
                 }
                 |
                 <K_ALTER> { alterExp.setOperation(AlterOperation.ALTER); }
@@ -7438,12 +7437,16 @@ AlterExpression AlterExpression():
             )
         )
         |
+        (
+        <K_FORCE>{ alterExp.setOperation(AlterOperation.FORCE); }
+        )
+        |
         (
          <K_ALGORITHM> {
             alterExp.setOperation(AlterOperation.ALGORITHM);
          }
          ["=" { alterExp.setUseEqual(true);} ]
-         sk3 = RelObjectName() {alterExp.addParameters(sk3); }
+         sk3 = RelObjectName() {alterExp.setAlgorithmOption(sk3); }
         )
         |
         (
@@ -7453,6 +7456,11 @@ AlterExpression AlterExpression():
         ["=" { alterExp.setUseEqual(true);} ]
         sk3 = RelObjectName() {alterExp.setLockOption(sk3); }
         )
+        |
+        (<K_ENGINE> {alterExp.setOperation(AlterOperation.ENGINE);}
+        ["=" { alterExp.setUseEqual(true);} ]
+        sk3 = RelObjectName() {alterExp.setEngineOption(sk3); }
+        )
       |
       LOOKAHEAD(2)  <K_RENAME> { alterExp.setOperation(AlterOperation.RENAME); } [ <K_COLUMN> {  alterExp.hasColumn(true);} ]
                     ( tk=<S_IDENTIFIER> | tk=<S_QUOTED_IDENTIFIER> ) { alterExp.setColOldName(tk.image); }
@@ -7516,17 +7524,6 @@ AlterExpression AlterExpression():
             }
       )
       |
-      (<K_ENGINE> {alterExp.setOperation(AlterOperation.SET_TABLE_OPTION);}
-                    ["=" { alterExp.setUseEqual(true);} ]
-                    tk=<S_IDENTIFIER> {
-                        if (alterExp.getUseEqual()) {
-                            alterExp.setTableOption("ENGINE = " + tk.image);
-                        } else {
-                            alterExp.setTableOption("ENGINE " + tk.image);
-                        }
-                  }
-            )
-      |
       LOOKAHEAD(2)
            (<K_RENAME> ((<K_INDEX> {alterExp.setOperation(AlterOperation.RENAME_INDEX);}
                     | <K_KEY> {alterExp.setOperation(AlterOperation.RENAME_KEY);})
diff --git a/src/test/java/net/sf/jsqlparser/statement/alter/AlterTest.java b/src/test/java/net/sf/jsqlparser/statement/alter/AlterTest.java
@@ -1354,8 +1354,48 @@ public void testIssue2114AlterTableEngine() throws JSQLParserException {
         assertEquals(1, alterExpressions.size());
 
         AlterExpression engineExp = alterExpressions.get(0);
-        assertEquals(AlterOperation.SET_TABLE_OPTION, engineExp.getOperation());
-        assertEquals(engineExp.getTableOption(), "ENGINE = InnoDB");
+        assertEquals(AlterOperation.ENGINE, engineExp.getOperation());
+        assertEquals(engineExp.getEngineOption(), "InnoDB");
         assertSqlCanBeParsedAndDeparsed(sql);
     }
+
+    @Test
+    public void testIssue2118AlterTableForceAndEngine() throws JSQLParserException {
+        String sql1 = "ALTER TABLE my_table FORCE";
+        Statement stmt1 = CCJSqlParserUtil.parse(sql1);
+        assertTrue(stmt1 instanceof Alter);
+        Alter alter1 = (Alter) stmt1;
+        List<AlterExpression> alterExpressions1 = alter1.getAlterExpressions();
+        assertNotNull(alterExpressions1);
+        assertEquals(1, alterExpressions1.size());
+
+        AlterExpression forceExp = alterExpressions1.get(0);
+        assertEquals(AlterOperation.FORCE, forceExp.getOperation());
+        assertSqlCanBeParsedAndDeparsed(sql1);
+
+        String sql2 = "ALTER TABLE tbl_name FORCE, ENGINE=InnoDB, ALGORITHM=INPLACE, LOCK=NONE";
+        Statement stmt2 = CCJSqlParserUtil.parse(sql2);
+        assertTrue(stmt2 instanceof Alter);
+        Alter alter2 = (Alter) stmt2;
+        List<AlterExpression> alterExpressions2 = alter2.getAlterExpressions();
+        assertNotNull(alterExpressions2);
+        assertEquals(4, alterExpressions2.size());
+
+        AlterExpression forceExp2 = alterExpressions2.get(0);
+        assertEquals(AlterOperation.FORCE, forceExp2.getOperation());
+
+        AlterExpression engineExp = alterExpressions2.get(1);
+        assertEquals(AlterOperation.ENGINE, engineExp.getOperation());
+        assertEquals(engineExp.getEngineOption(), "InnoDB");
+
+        AlterExpression algorithmExp = alterExpressions2.get(2);
+        assertEquals(AlterOperation.ALGORITHM, algorithmExp.getOperation());
+        assertEquals("INPLACE", algorithmExp.getAlgorithmOption());
+
+        AlterExpression lockExp = alterExpressions2.get(3);
+        assertEquals(AlterOperation.LOCK, lockExp.getOperation());
+        assertEquals("NONE", lockExp.getLockOption());
+
+        assertSqlCanBeParsedAndDeparsed(sql2);
+    }
 }

Original file line number	Diff line number	Diff line change
`@@ -7167,7 +7167,6 @@ AlterExpression AlterExpression():`
`7167`	`7167`	`(`
`7168`	`7168`	`(`
`7169`	`7169`	`<K_ADD> { alterExp.setOperation(AlterOperation.ADD);`
`7170`		`- System.out.println("test");`
`7171`	`7170`	`}`
`7172`	`7171`	`\|`
`7173`	`7172`	`<K_ALTER> { alterExp.setOperation(AlterOperation.ALTER); }`
`@@ -7438,12 +7437,16 @@ AlterExpression AlterExpression():`
`7438`	`7437`	`)`
`7439`	`7438`	`)`
`7440`	`7439`	`\|`
	`7440`	`+ (`
	`7441`	`+ <K_FORCE>{ alterExp.setOperation(AlterOperation.FORCE); }`
	`7442`	`+ )`
	`7443`	`+ \|`
`7441`	`7444`	`(`
`7442`	`7445`	`<K_ALGORITHM> {`
`7443`	`7446`	`alterExp.setOperation(AlterOperation.ALGORITHM);`
`7444`	`7447`	`}`
`7445`	`7448`	`["=" { alterExp.setUseEqual(true);} ]`
`7446`		`- sk3 = RelObjectName() {alterExp.addParameters(sk3); }`
	`7449`	`+ sk3 = RelObjectName() {alterExp.setAlgorithmOption(sk3); }`
`7447`	`7450`	`)`
`7448`	`7451`	`\|`
`7449`	`7452`	`(`
`@@ -7453,6 +7456,11 @@ AlterExpression AlterExpression():`
`7453`	`7456`	`["=" { alterExp.setUseEqual(true);} ]`
`7454`	`7457`	`sk3 = RelObjectName() {alterExp.setLockOption(sk3); }`
`7455`	`7458`	`)`
	`7459`	`+ \|`
	`7460`	`+ (<K_ENGINE> {alterExp.setOperation(AlterOperation.ENGINE);}`
	`7461`	`+ ["=" { alterExp.setUseEqual(true);} ]`
	`7462`	`+ sk3 = RelObjectName() {alterExp.setEngineOption(sk3); }`
	`7463`	`+ )`
`7456`	`7464`	`\|`
`7457`	`7465`	`LOOKAHEAD(2) <K_RENAME> { alterExp.setOperation(AlterOperation.RENAME); } [ <K_COLUMN> { alterExp.hasColumn(true);} ]`
`7458`	`7466`	`( tk=<S_IDENTIFIER> \| tk=<S_QUOTED_IDENTIFIER> ) { alterExp.setColOldName(tk.image); }`
`@@ -7516,17 +7524,6 @@ AlterExpression AlterExpression():`
`7516`	`7524`	`}`
`7517`	`7525`	`)`
`7518`	`7526`	`\|`
`7519`		`- (<K_ENGINE> {alterExp.setOperation(AlterOperation.SET_TABLE_OPTION);}`
`7520`		`- ["=" { alterExp.setUseEqual(true);} ]`
`7521`		`- tk=<S_IDENTIFIER> {`
`7522`		`- if (alterExp.getUseEqual()) {`
`7523`		`- alterExp.setTableOption("ENGINE = " + tk.image);`
`7524`		`- } else {`
`7525`		`- alterExp.setTableOption("ENGINE " + tk.image);`
`7526`		`- }`
`7527`		`- }`
`7528`		`- )`
`7529`		`- \|`
`7530`	`7527`	`LOOKAHEAD(2)`
`7531`	`7528`	`(<K_RENAME> ((<K_INDEX> {alterExp.setOperation(AlterOperation.RENAME_INDEX);}`
`7532`	`7529`	`\| <K_KEY> {alterExp.setOperation(AlterOperation.RENAME_KEY);})`