Add APPLY PATCHES and CLEAR COLUMN IN PARTITION support

- Add AlterApplyPatches command type for ALTER TABLE ... APPLY PATCHES IN PARTITION
- Add IN PARTITION parsing for CLEAR COLUMN command
- Fix Partition output wrapper in explain for CLEAR COLUMN

Author: claude

ast/ast.go

@@ -644,6 +644,7 @@ const (
 	AlterMovePartition      AlterCommandType = "MOVE_PARTITION"
 	AlterFreezePartition    AlterCommandType = "FREEZE_PARTITION"
 	AlterFreeze             AlterCommandType = "FREEZE"
+	AlterApplyPatches       AlterCommandType = "APPLY_PATCHES"
 	AlterDeleteWhere        AlterCommandType = "DELETE_WHERE"
 	AlterUpdate             AlterCommandType = "UPDATE"
 	AlterAddProjection      AlterCommandType = "ADD_PROJECTION"

internal/explain/statements.go

@@ -1198,7 +1198,8 @@ func explainAlterCommand(sb *strings.Builder, cmd *ast.AlterCommand, indent stri
 			fmt.Fprintf(sb, "%s Identifier %s\n", indent, cmd.ColumnName)
 		}
 		if cmd.Partition != nil {
-			Node(sb, cmd.Partition, depth+1)
+			fmt.Fprintf(sb, "%s Partition (children 1)\n", indent)
+			Node(sb, cmd.Partition, depth+2)
 		}
 	case ast.AlterCommentColumn:
 		if cmd.ColumnName != "" {
@@ -1233,7 +1234,7 @@ func explainAlterCommand(sb *strings.Builder, cmd *ast.AlterCommand, indent stri
 	case ast.AlterModifySetting:
 		fmt.Fprintf(sb, "%s Set\n", indent)
 	case ast.AlterDropPartition, ast.AlterDetachPartition, ast.AlterAttachPartition,
-		ast.AlterReplacePartition, ast.AlterFetchPartition, ast.AlterMovePartition, ast.AlterFreezePartition:
+		ast.AlterReplacePartition, ast.AlterFetchPartition, ast.AlterMovePartition, ast.AlterFreezePartition, ast.AlterApplyPatches:
 		if cmd.Partition != nil {
 			// PARTITION ALL is shown as Partition_ID (empty) in EXPLAIN AST
 			if ident, ok := cmd.Partition.(*ast.Identifier); ok && strings.ToUpper(ident.Name()) == "ALL" {
@@ -1430,7 +1431,7 @@ func countAlterCommandChildren(cmd *ast.AlterCommand) int {
 	case ast.AlterModifySetting:
 		children = 1
 	case ast.AlterDropPartition, ast.AlterDetachPartition, ast.AlterAttachPartition,
-		ast.AlterReplacePartition, ast.AlterFetchPartition, ast.AlterMovePartition, ast.AlterFreezePartition:
+		ast.AlterReplacePartition, ast.AlterFetchPartition, ast.AlterMovePartition, ast.AlterFreezePartition, ast.AlterApplyPatches:
 		if cmd.Partition != nil {
 			children++
 		}

parser/parser.go

@@ -4569,10 +4569,18 @@ func (p *Parser) parseAlterCommand() *ast.AlterCommand {
 			} else if p.currentIs(token.COLUMN) {
 				cmd.Type = ast.AlterClearColumn
 				p.nextToken()
-				if p.currentIs(token.IDENT) {
+				if p.currentIs(token.IDENT) || p.current.Token.IsKeyword() {
 					cmd.ColumnName = p.current.Value
 					p.nextToken()
 				}
+				// Parse IN PARTITION
+				if p.currentIs(token.IN) {
+					p.nextToken() // skip IN
+					if p.currentIs(token.PARTITION) {
+						p.nextToken() // skip PARTITION
+						cmd.Partition = p.parseExpression(LOWEST)
+					}
+				}
 			} else if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PROJECTION" {
 				cmd.Type = ast.AlterClearProjection
 				p.nextToken()
@@ -4760,6 +4768,20 @@ func (p *Parser) parseAlterCommand() *ast.AlterCommand {
 				}
 			}
 		}
+	case token.APPLY:
+		// APPLY PATCHES IN PARTITION expr
+		p.nextToken() // skip APPLY
+		if p.currentIs(token.IDENT) && strings.ToUpper(p.current.Value) == "PATCHES" {
+			p.nextToken() // skip PATCHES
+			cmd.Type = ast.AlterApplyPatches
+			if p.currentIs(token.IN) {
+				p.nextToken() // skip IN
+				if p.currentIs(token.PARTITION) {
+					p.nextToken() // skip PARTITION
+					cmd.Partition = p.parseExpression(LOWEST)
+				}
+			}
+		}
 	case token.DELETE:
 		// DELETE WHERE condition - mutation to delete rows
 		cmd.Type = ast.AlterDeleteWhere

parser/testdata/00446_clear_column_in_partition_zookeeper_long/metadata.json

@@ -1,14 +1 @@
-{
-  "explain_todo": {
-    "stmt11": true,
-    "stmt24": true,
-    "stmt31": true,
-    "stmt34": true,
-    "stmt37": true,
-    "stmt38": true,
-    "stmt43": true,
-    "stmt44": true,
-    "stmt48": true,
-    "stmt8": true
-  }
-}
+{}

parser/testdata/01114_clear_column_compact_parts/metadata.json

@@ -1,6 +1 @@
-{
-  "explain_todo": {
-    "stmt4": true,
-    "stmt6": true
-  }
-}
+{}

parser/testdata/01387_clear_column_default_depends/metadata.json

@@ -1,5 +1 @@
-{
-  "explain_todo": {
-    "stmt28": true
-  }
-}
+{}

parser/testdata/01388_clear_all_columns/metadata.json

@@ -1,5 +1 @@
-{
-  "explain_todo": {
-    "stmt9": true
-  }
-}
+{}

parser/testdata/01416_clear_column_pk/metadata.json

@@ -1,6 +1 @@
-{
-  "explain_todo": {
-    "stmt4": true,
-    "stmt5": true
-  }
-}
+{}

parser/testdata/03100_lwu_22_detach_attach_patches/metadata.json

@@ -1,13 +1,9 @@
 {
   "explain_todo": {
-    "stmt21": true,
     "stmt23": true,
-    "stmt24": true,
     "stmt25": true,
     "stmt26": true,
-    "stmt27": true,
     "stmt28": true,
-    "stmt30": true,
     "stmt33": true,
     "stmt37": true
   }