*: refine split region syntax add split partition table region syntax. (

pingcap#561)
lyonzhi · Oct 12, 2019 · 464e332 · 464e332
1 parent 8b33331
commit 464e332
Show file tree

Hide file tree

Showing 4 changed files with 7,309 additions and 7,201 deletions.
diff --git a/ast/dml.go b/ast/dml.go
@@ -2427,8 +2427,11 @@ func (n *FrameBound) Accept(v Visitor) (Node, bool) {
 type SplitRegionStmt struct {
 	dmlNode
 
-	Table     *TableName
-	IndexName model.CIStr
+	Table          *TableName
+	IndexName      model.CIStr
+	PartitionNames []model.CIStr
+
+	SplitSyntaxOpt *SplitSyntaxOption
 
 	SplitOpt *SplitOption
 }
@@ -2440,11 +2443,39 @@ type SplitOption struct {
 	ValueLists [][]ExprNode
 }
 
+type SplitSyntaxOption struct {
+	HasRegionFor bool
+	HasPartition bool
+}
+
 func (n *SplitRegionStmt) Restore(ctx *RestoreCtx) error {
-	ctx.WriteKeyWord("SPLIT TABLE ")
+	ctx.WriteKeyWord("SPLIT ")
+	if n.SplitSyntaxOpt != nil {
+		if n.SplitSyntaxOpt.HasRegionFor {
+			ctx.WriteKeyWord("REGION FOR ")
+		}
+		if n.SplitSyntaxOpt.HasPartition {
+			ctx.WriteKeyWord("PARTITION ")
+
+		}
+	}
+	ctx.WriteKeyWord("TABLE ")
+
 	if err := n.Table.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred while restore SplitIndexRegionStmt.Table")
 	}
+	if len(n.PartitionNames) > 0 {
+		ctx.WriteKeyWord(" PARTITION")
+		ctx.WritePlain("(")
+		for i, v := range n.PartitionNames {
+			if i != 0 {
+				ctx.WritePlain(", ")
+			}
+			ctx.WriteName(v.String())
+		}
+		ctx.WritePlain(")")
+	}
+
 	if len(n.IndexName.L) > 0 {
 		ctx.WriteKeyWord(" INDEX ")
 		ctx.WriteName(n.IndexName.String())

diff --git a/misc.go b/misc.go
@@ -469,6 +469,7 @@ var tokenMap = map[string]int{
 	"REFERENCES":               references,
 	"REGEXP":                   regexpKwd,
 	"REGIONS":                  regions,
+	"REGION":                   region,
 	"RELOAD":                   reload,
 	"REMOVE":                   remove,
 	"RENAME":                   rename,