refactor truncate partition

Author: mj-db

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

@@ -72,8 +72,6 @@ public void setPartitions(List<String> partitions) {
 
     private boolean useBrackets = false;
 
-    private String truncatePartitionName = null;
-
     private boolean useIfNotExists = false;
 
     public Index getOldIndex() {
@@ -442,19 +440,6 @@ public void setUk(boolean uk) {
         this.uk = uk;
     }
 
-    public String getTruncatePartitionName() {
-        return truncatePartitionName;
-    }
-
-    public void setTruncatePartitionName(String truncatePartitionName) {
-        this.truncatePartitionName = truncatePartitionName;
-    }
-
-    public AlterExpression withTruncatePartitionName(String truncatePartitionName) {
-        this.truncatePartitionName = truncatePartitionName;
-        return this;
-    }
-
     public boolean isUseIfNotExists() {
         return useIfNotExists;
     }
@@ -508,8 +493,8 @@ public String toString() {
             // Oracle Multi Column Drop
             b.append("DROP (").append(PlainSelect.getStringList(pkColumns)).append(')');
         } else if (operation == AlterOperation.TRUNCATE_PARTITION
-                && truncatePartitionName != null) {
-            b.append("TRUNCATE PARTITION ").append(truncatePartitionName);
+                && partitions != null) {
+            b.append("TRUNCATE PARTITION ").append(PlainSelect.getStringList(partitions));
         } else {
             if (operation == AlterOperation.COMMENT_WITH_EQUAL_SIGN) {
                 b.append("COMMENT =").append(" ");

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

@@ -7436,7 +7436,10 @@ AlterExpression AlterExpression():
            }
           )
       |
-      LOOKAHEAD(2) <K_TRUNCATE> <K_PARTITION> { alterExp.setOperation(AlterOperation.TRUNCATE_PARTITION); } truncatePartitionName = RelObjectName() { alterExp.setTruncatePartitionName(truncatePartitionName); }
+      LOOKAHEAD(2) <K_TRUNCATE> <K_PARTITION> { alterExp.setOperation(AlterOperation.TRUNCATE_PARTITION); }
+       partitions=PartitionNamesList() {
+           alterExp.setPartitions(partitions);
+       }
       |
       tokens = captureRest() {
                     alterExp.setOperation(AlterOperation.UNSPECIFIC);

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

@@ -1252,4 +1252,27 @@ public void testIssue2106AlterTableDropPartition() throws JSQLParserException {
         assertEquals("p202009", partitionNames.get(2));
         assertEquals("p202010", partitionNames.get(3));
     }
+
+    @Test
+    public void testIssue2106AlterTableTruncatePartition() throws JSQLParserException {
+        String sql =
+                "ALTER TABLE dkpg_payments TRUNCATE PARTITION p201701, p201707, p201801, p201807";
+        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.TRUNCATE_PARTITION, partitionExp.getOperation());
+        List<String> partitionNames = partitionExp.getPartitions();
+        assertNotNull(partitionNames);
+        assertEquals(4, partitionNames.size());
+
+        assertEquals("p201701", partitionNames.get(0));
+        assertEquals("p201707", partitionNames.get(1));
+        assertEquals("p201801", partitionNames.get(2));
+        assertEquals("p201807", partitionNames.get(3));
+    }
 }