*: suport alter table set flash replica syntax

ast/ddl.go

@@ -1921,6 +1921,8 @@ const (
 	AlterTableIndexInvisible
 	// TODO: Add more actions
 	AlterTableOrderByColumns
+	// AlterTableSetFlashReplica uses to set the table TiFlash replica.
+	AlterTableSetFlashReplica
 )
 
 // LockType is the type for AlterTableSpec.
@@ -2015,6 +2017,12 @@ type AlterTableSpec struct {
 	WithValidation  bool
 	Num             uint64
 	Visibility      IndexVisibility
+	FlashReplica    *FlashReplicaSpec
+}
+
+type FlashReplicaSpec struct {
+	Count  uint64
+	Labels []string
 }
 
 // AlterOrderItem represents an item in order by at alter table stmt.
@@ -2038,6 +2046,19 @@ func (n *AlterOrderItem) Restore(ctx *RestoreCtx) error {
 // Restore implements Node interface.
 func (n *AlterTableSpec) Restore(ctx *RestoreCtx) error {
 	switch n.Tp {
+	case AlterTableSetFlashReplica:
+		ctx.WriteKeyWord("SET FLASH REPLICA ")
+		ctx.WritePlainf("%d", n.FlashReplica.Count)
+		if len(n.FlashReplica.Labels) == 0 {
+			break
+		}
+		ctx.WriteKeyWord(" LOCATION LABELS ")
+		for i, v := range n.FlashReplica.Labels {
+			if i > 0 {
+				ctx.WritePlain(", ")
+			}
+			ctx.WriteString(v)
+		}
 	case AlterTableOption:
 		switch {
 		case len(n.Options) == 2 &&

misc.go

@@ -282,6 +282,7 @@ var tokenMap = map[string]int{
 	"FIRST":                    first,
 	"FIXED":                    fixed,
 	"FLOAT":                    floatType,
+	"FLASH":                    flash,
 	"FLUSH":                    flush,
 	"FOLLOWING":                following,
 	"FOR":                      forKwd,
@@ -350,6 +351,7 @@ var tokenMap = map[string]int{
 	"KEY_BLOCK_SIZE":           keyBlockSize,
 	"KEYS":                     keys,
 	"KILL":                     kill,
+	"LABELS":                   labels,
 	"LAST":                     last,
 	"LEADING":                  leading,
 	"LEFT":                     left,
@@ -364,6 +366,7 @@ var tokenMap = map[string]int{
 	"LOCAL":                    local,
 	"LOCALTIME":                localTime,
 	"LOCALTIMESTAMP":           localTs,
+	"LOCATION":                 location,
 	"LOCK":                     lock,
 	"LONG":                     long,
 	"LONGBLOB":                 longblobType,
@@ -477,6 +480,7 @@ var tokenMap = map[string]int{
 	"REPEATABLE":               repeatable,
 	"REPLACE":                  replace,
 	"RESPECT":                  respect,
+	"REPLICA":                  replica,
 	"REPLICATION":              replication,
 	"REQUIRE":                  require,
 	"RESTRICT":                 restrict,

model/ddl.go

@@ -58,6 +58,8 @@ const (
 	ActionModifySchemaCharsetAndCollate ActionType = 26
 	ActionLockTable                     ActionType = 27
 	ActionUnlockTable                   ActionType = 28
+	ActionSetFlashReplica               ActionType = 29
+	ActionUpdateFlashReplicaStatus      ActionType = 30
 )
 
 // AddIndexStr is a string related to the operation of "add index".
@@ -92,6 +94,8 @@ var actionMap = map[ActionType]string{
 	ActionModifySchemaCharsetAndCollate: "modify schema charset and collate",
 	ActionLockTable:                     "lock table",
 	ActionUnlockTable:                   "unlock table",
+	ActionSetFlashReplica:               "set flash replica",
+	ActionUpdateFlashReplicaStatus:      "update flash replica status",
 }
 
 // String return current ddl action in string

model/model.go

@@ -236,6 +236,9 @@ type TableInfo struct {
 
 	// Version means the version of the table info.
 	Version uint16 `json:"version"`
+
+	// FlashReplica means the flash replica info.
+	FlashReplica *FlashReplicaInfo `json:"flash_replica"`
 }
 
 // TableLockInfo provides meta data describing a table lock.
@@ -323,6 +326,14 @@ func (t TableLockType) String() string {
 	return ""
 }
 
+// FlashReplicaInfo means the flash replica info.
+type FlashReplicaInfo struct {
+	Count            uint64
+	LocationLabels   []string
+	RegionCount      uint64
+	FlashRegionCount uint64
+}
+
 // GetPartitionInfo returns the partition information.
 func (t *TableInfo) GetPartitionInfo() *PartitionInfo {
 	if t.Partition != nil && t.Partition.Enable {