*: suport alter table set flash replica syntax (pingcap#564)

ast/ddl.go

@@ -1680,7 +1680,8 @@ const (
  AlterTablePartition
  AlterTableEnableKeys
  AlterTableDisableKeys
-
+ // AlterTableSetTiFlashReplica uses to set the table TiFlash replica.
+ AlterTableSetTiFlashReplica
  // TODO: Add more actions
 )
 
@@ -1760,11 +1761,30 @@ type AlterTableSpec struct {
  PartitionNames []model.CIStr
  PartDefinitions []*PartitionDefinition
  Num uint64
+ TiFlashReplica *TiFlashReplicaSpec
+}
+
+type TiFlashReplicaSpec struct {
+ Count uint64
+ Labels []string
 }
 
 // Restore implements Node interface.
 func (n *AlterTableSpec) Restore(ctx *RestoreCtx) error {
  switch n.Tp {
+ case AlterTableSetTiFlashReplica:
+ ctx.WriteKeyWord("SET TIFLASH REPLICA ")
+ ctx.WritePlainf("%d", n.TiFlashReplica.Count)
+ if len(n.TiFlashReplica.Labels) == 0 {
+ break
+ }
+ ctx.WriteKeyWord(" LOCATION LABELS ")
+ for i, v := range n.TiFlashReplica.Labels {
+ if i > 0 {
+ ctx.WritePlain(", ")
+ }
+ ctx.WriteString(v)
+ }
  case AlterTableOption:
  switch {
  case len(n.Options) == 2 &&

misc.go

@@ -340,6 +340,7 @@ var tokenMap = map[string]int{
  "KEY_BLOCK_SIZE": keyBlockSize,
  "KEYS": keys,
  "KILL": kill,
+ "LABELS": labels,
  "LAST": last,
  "LEADING": leading,
  "LEFT": left,
@@ -354,6 +355,7 @@ var tokenMap = map[string]int{
  "LOCAL": local,
  "LOCALTIME": localTime,
  "LOCALTIMESTAMP": localTs,
+ "LOCATION": location,
  "LOCK": lock,
  "LONG": long,
  "LONGBLOB": longblobType,
@@ -460,6 +462,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,9 @@ const (
  ActionModifySchemaCharsetAndCollate ActionType = 26
  ActionLockTable ActionType = 27
  ActionUnlockTable ActionType = 28
+ ActionRepairTable ActionType = 29
+ ActionSetTiFlashReplica ActionType = 30
+ ActionUpdateTiFlashReplicaStatus ActionType = 31
  ActionAddPrimaryKey ActionType = 32
  ActionDropPrimaryKey ActionType = 33
 )
@@ -99,6 +102,9 @@ var actionMap = map[ActionType]string{
  ActionDropPrimaryKey: "drop primary key",
  ActionLockTable: "lock table",
  ActionUnlockTable: "unlock table",
+ ActionRepairTable: "repair table",
+ ActionSetTiFlashReplica: "set tiflash replica",
+ ActionUpdateTiFlashReplicaStatus: "update tiflash replica status",
 }
 
 // String return current ddl action in string

model/model.go

@@ -248,6 +248,9 @@ type TableInfo struct {
 
  // Version means the version of the table info.
  Version uint16 `json:"version"`
+
+ // TiFlashReplica means the TiFlash replica info.
+ TiFlashReplica *TiFlashReplicaInfo `json:"tiflash_replica"`
 }
 
 // TableLockInfo provides meta data describing a table lock.
@@ -335,6 +338,13 @@ func (t TableLockType) String() string {
  return ""
 }
 
+// TiFlashReplicaInfo means the flash replica info.
+type TiFlashReplicaInfo struct {
+ Count uint64
+ LocationLabels []string
+ Available bool
+}
+
 // GetPartitionInfo returns the partition information.
 func (t *TableInfo) GetPartitionInfo() *PartitionInfo {
  if t.Partition != nil && t.Partition.Enable {