update: 添加分区表与非分区表转换语法支持 (#434)

hanchuanchuan · Apr 5, 2022 · 37c624b · 37c624b
1 parent 767bf4f
commit 37c624b
Show file tree

Hide file tree

Showing 11 changed files with 7,106 additions and 6,117 deletions.
diff --git a/ast/ddl.go b/ast/ddl.go
diff --git a/ast/misc.go b/ast/misc.go
@@ -1322,6 +1322,20 @@ func (n *DropBindingStmt) Accept(v Visitor) (Node, bool) {
 	return v.Leave(n)
 }
 
+// Extended statistics types.
+const (
+	StatsTypeCardinality uint8 = iota
+	StatsTypeDependency
+	StatsTypeCorrelation
+)
+
+// StatisticsSpec is the specification for ADD /DROP STATISTICS.
+type StatisticsSpec struct {
+	StatsName string
+	StatsType uint8
+	Columns   []*ColumnName
+}
+
 // DoStmt is the struct for DO statement.
 type DoStmt struct {
 	stmtNode

diff --git a/format/format.go b/format/format.go
@@ -218,6 +218,12 @@ const (
 	RestoreNameBackQuotes
 
 	RestoreSpacesAroundBinaryOperation
+	RestoreBracketAroundBinaryOperation
+
+	RestoreStringWithoutCharset
+	RestoreStringWithoutDefaultCharset
+
+	RestoreTiDBSpecialComment
 )
 
 const (
@@ -278,6 +284,22 @@ func (rf RestoreFlags) HasSpacesAroundBinaryOperationFlag() bool {
 	return rf.has(RestoreSpacesAroundBinaryOperation)
 }
 
+func (rf RestoreFlags) HasRestoreBracketAroundBinaryOperation() bool {
+	return rf.has(RestoreBracketAroundBinaryOperation)
+}
+
+func (rf RestoreFlags) HasStringWithoutDefaultCharset() bool {
+	return rf.has(RestoreStringWithoutDefaultCharset)
+}
+
+func (rf RestoreFlags) HasStringWithoutCharset() bool {
+	return rf.has(RestoreStringWithoutCharset)
+}
+
+func (rf RestoreFlags) HasTiDBSpecialCommentFlag() bool {
+	return rf.has(RestoreTiDBSpecialComment)
+}
+
 // RestoreCtx is `Restore` context to hold flags and writer.
 type RestoreCtx struct {
 	Flags     RestoreFlags
@@ -302,6 +324,20 @@ func (ctx *RestoreCtx) WriteKeyWord(keyWord string) {
 	fmt.Fprint(ctx.In, keyWord)
 }
 
+func (ctx *RestoreCtx) WriteWithSpecialComments(featureID string, fn func()) {
+	if !ctx.Flags.HasTiDBSpecialCommentFlag() {
+		fn()
+		return
+	}
+	ctx.WritePlain("/*T!")
+	if len(featureID) != 0 {
+		ctx.WritePlainf("[%s]", featureID)
+	}
+	ctx.WritePlain(" ")
+	fn()
+	ctx.WritePlain(" */")
+}
+
 // WriteString writes the string into writer
 // `str` may be wrapped in quotes and escaped according to RestoreFlags.
 func (ctx *RestoreCtx) WriteString(str string) {

diff --git a/go.mod b/go.mod
@@ -31,7 +31,7 @@ require (
 	github.com/dustin/go-humanize v1.0.0 // indirect
 	github.com/eknkc/amber v0.0.0-20171010120322-cdade1c07385 // indirect
 	github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5 // indirect
-	github.com/etcd-io/gofail v0.0.0-20180808172546-51ce9a71510a // indirect
+	github.com/etcd-io/gofail v0.0.0-20180808172546-51ce9a71510a
 	github.com/ghodss/yaml v1.0.0 // indirect
 	github.com/go-sql-driver/mysql v1.4.1
 	github.com/gofrs/uuid v3.2.0+incompatible

diff --git a/parser/misc.go b/parser/misc.go
@@ -149,6 +149,7 @@ var tokenMap = map[string]int{
 	"AS":               as,
 	"ASC":              asc,
 	"ASCII":            ascii,
+	"ATTRIBUTES":       attributes,
 	"AUTO_INCREMENT":   autoIncrement,
 	"AUTO_RANDOM":      autoRandom,
 	"AUTO_RANDOM_BASE": autoRandomBase,
@@ -396,6 +397,7 @@ var tokenMap = map[string]int{
 	"PACK_KEYS":                packKeys,
 	"PARSER":                   parser,
 	"PARTITION":                partition,
+	"PARTITIONING":             partitioning,
 	"PARTITIONS":               partitions,
 	"PASSWORD":                 password,
 	"PLUGINS":                  plugins,
@@ -422,7 +424,9 @@ var tokenMap = map[string]int{
 	"REFERENCES":               references,
 	"REGEXP":                   regexpKwd,
 	"RELOAD":                   reload,
+	"REMOVE":                   remove,
 	"RENAME":                   rename,
+	"REORGANIZE":               reorganize,
 	"REPAIR":                   repair,
 	"REPEAT":                   repeat,
 	"REPEATABLE":               repeatable,