add the new stmt: drop stats partition (#1177)

pingcap · Mar 1, 2021 · 267e02a · 267e02a
1 parent cc1bbd2
commit 267e02a
Show file tree

Hide file tree

Showing 4 changed files with 4,913 additions and 4,883 deletions.
diff --git a/ast/stats.go b/ast/stats.go
@@ -174,7 +174,8 @@ func (n *AnalyzeTableStmt) Accept(v Visitor) (Node, bool) {
 type DropStatsStmt struct {
 	stmtNode
 
-	Table *TableName
+	Table          *TableName
+	PartitionNames []model.CIStr
 }
 
 // Restore implements Node interface.
@@ -184,6 +185,15 @@ func (n *DropStatsStmt) Restore(ctx *format.RestoreCtx) error {
 		return errors.Annotate(err, "An error occurred while add table")
 	}
 
+	if len(n.PartitionNames) != 0 {
+		ctx.WriteKeyWord(" PARTITION ")
+	}
+	for i, partition := range n.PartitionNames {
+		if i != 0 {
+			ctx.WritePlain(",")
+		}
+		ctx.WriteName(partition.O)
+	}
 	return nil
 }