feat mysql alter drop partition

Author: mj-db

src/main/java/net/sf/jsqlparser/statement/alter/AlterExpression.java

@@ -49,6 +49,15 @@ public class AlterExpression implements Serializable {
     private boolean uk;
     private boolean useEqual;
 
+    public List<String> getPartitions() {
+        return partitions;
+    }
+
+    public void setPartitions(List<String> partitions) {
+        this.partitions = partitions;
+    }
+
+    private List<String> partitions;
     private List<PartitionDefinition> partitionDefinitions;
     private List<ConstraintState> constraints;
     private List<String> parameters;

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, TRUNCATE_PARTITION, 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, LOCK;
 
     public static AlterOperation from(String operation) {
         return Enum.valueOf(AlterOperation.class, operation.toUpperCase());

src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

@@ -7060,6 +7060,26 @@ List<PartitionDefinition> PartitionDefinitions():
     }
 }
 
+
+List<String> PartitionNamesList() :
+{
+    Token tk;
+    List<String> partitionNames = new ArrayList<String>();
+}
+{
+    tk = <S_IDENTIFIER> {
+        partitionNames.add(tk.image);
+    }
+    (
+        "," tk = <S_IDENTIFIER> {
+            partitionNames.add(tk.image);
+        }
+    )*
+    {
+        return partitionNames;
+    }
+}
+
 /**
 * This production needs refactoring to multiple smaller productions. The target class should
 * be splitted as well.
@@ -7081,6 +7101,7 @@ AlterExpression AlterExpression():
     AlterExpression.ColumnDropNotNull alterExpressionColumnDropNotNull = null;
     AlterExpression.ColumnDropDefault alterExpressionColumnDropDefault = null;
     ReferentialAction.Action action = null;
+    List<String> partitions = null;
     List<PartitionDefinition> partitionDefinition = null;
     String truncatePartitionName = null;
 
@@ -7126,7 +7147,6 @@ AlterExpression AlterExpression():
                 |
                 (
                     <K_PARTITION> {
-                        System.out.println("test _ Partition");
                         alterExp.setOperation(AlterOperation.ADD_PARTITION);
                     }
                     partitionDefinition=PartitionDefinitions() {
@@ -7296,6 +7316,15 @@ AlterExpression AlterExpression():
       |
         <K_DROP> { alterExp.setOperation(AlterOperation.DROP); }
         (
+            (
+                <K_PARTITION> {
+                    alterExp.setOperation(AlterOperation.DROP_PARTITION);
+                }
+                partitions=PartitionNamesList() {
+                    alterExp.setPartitions(partitions);
+                }
+            )
+            |
             (
                 (
                     // we use the PK Columns Field instead of the Column Field

src/test/java/net/sf/jsqlparser/statement/alter/AlterTest.java

@@ -1229,4 +1229,27 @@ public void testIssue2106AlterTableAddPartitionCodeTransaction() throws JSQLPars
         assertEquals(Collections.singletonList("'20210901'"), partitions.get(0).getValues());
         assertEquals("InnoDB", partitions.get(0).getStorageEngine());
     }
+
+    @Test
+    public void testIssue2106AlterTableDropPartition() throws JSQLParserException {
+        String sql =
+                "ALTER TABLE dkpg_payment_details DROP PARTITION p202007, p202008, p202009, p202010";
+        Statement stmt = CCJSqlParserUtil.parse(sql);
+        assertTrue(stmt instanceof Alter);
+        Alter alter = (Alter) stmt;
+        List<AlterExpression> alterExpressions = alter.getAlterExpressions();
+        assertNotNull(alterExpressions);
+        assertEquals(1, alterExpressions.size());
+
+        AlterExpression partitionExp = alterExpressions.get(0);
+        assertEquals(AlterOperation.DROP_PARTITION, partitionExp.getOperation());
+        List<String> partitionNames = partitionExp.getPartitions();
+        assertNotNull(partitionNames);
+        assertEquals(4, partitionNames.size());
+
+        assertEquals("p202007", partitionNames.get(0));
+        assertEquals("p202008", partitionNames.get(1));
+        assertEquals("p202009", partitionNames.get(2));
+        assertEquals("p202010", partitionNames.get(3));
+    }
 }