add support for repair table stmt (#36)

arana-db · May 24, 2023 · 14fbd14 · 14fbd14
1 parent 6c840d5
commit 14fbd14
Show file tree

Hide file tree

Showing 5 changed files with 4,680 additions and 4,610 deletions.
diff --git a/ast/ddl.go b/ast/ddl.go
@@ -44,6 +44,7 @@ var (
 	_ DDLNode = &RenameTableStmt{}
 	_ DDLNode = &TruncateTableStmt{}
 	_ DDLNode = &RepairTableStmt{}
+	_ DDLNode = &AdminRepairTableStmt{}
 	_ DDLNode = &OptimizeTableStmt{}
 	_ DDLNode = &CheckTableStmt{}
 
@@ -1987,20 +1988,20 @@ func (c *CheckTableStmt) Accept(v Visitor) (Node, bool) {
 	return v.Leave(c)
 }
 
-// RepairTableStmt is a statement to repair tableInfo.
-type RepairTableStmt struct {
+// AdminRepairTableStmt is a statement to repair tableInfo.
+type AdminRepairTableStmt struct {
 	ddlNode
 	Table      *TableName
 	CreateStmt *CreateTableStmt
 }
 
 // Accept implements Node Accept interface.
-func (n *RepairTableStmt) Accept(v Visitor) (Node, bool) {
+func (n *AdminRepairTableStmt) Accept(v Visitor) (Node, bool) {
 	newNode, skipChildren := v.Enter(n)
 	if skipChildren {
 		return v.Leave(newNode)
 	}
-	n = newNode.(*RepairTableStmt)
+	n = newNode.(*AdminRepairTableStmt)
 	node, ok := n.Table.Accept(v)
 	if !ok {
 		return n, false
@@ -2015,7 +2016,7 @@ func (n *RepairTableStmt) Accept(v Visitor) (Node, bool) {
 }
 
 // Restore implements Node interface.
-func (n *RepairTableStmt) Restore(ctx *format.RestoreCtx) error {
+func (n *AdminRepairTableStmt) Restore(ctx *format.RestoreCtx) error {
 	ctx.WriteKeyWord("ADMIN REPAIR TABLE ")
 	if err := n.Table.Restore(ctx); err != nil {
 		return errors.Annotatef(err, "An error occurred while restore RepairTableStmt.table : [%v]", n.Table)
@@ -2027,6 +2028,48 @@ func (n *RepairTableStmt) Restore(ctx *format.RestoreCtx) error {
 	return nil
 }
 
+// RepairTableStmt is a statement to repair tableInfo.
+// See https://dev.mysql.com/doc/refman/8.0/en/repair-table.html
+type RepairTableStmt struct {
+	ddlNode
+	Tables []*TableName
+	Quick  bool
+}
+
+// Accept implements Node Accept interface.
+func (n *RepairTableStmt) Accept(v Visitor) (Node, bool) {
+	newNode, skipChildren := v.Enter(n)
+	if skipChildren {
+		return v.Leave(newNode)
+	}
+	n = newNode.(*RepairTableStmt)
+	for i := range n.Tables {
+		node, ok := n.Tables[i].Accept(v)
+		if !ok {
+			return n, false
+		}
+		n.Tables[i] = node.(*TableName)
+	}
+	return v.Leave(n)
+}
+
+// Restore implements Node interface.
+func (n *RepairTableStmt) Restore(ctx *format.RestoreCtx) error {
+	ctx.WriteKeyWord("REPAIR TABLE ")
+	for i, v := range n.Tables {
+		if i != 0 {
+			ctx.WritePlain(", ")
+		}
+		if err := v.Restore(ctx); err != nil {
+			return errors.Annotatef(err, "An error occurred while restore RepairTableStmt.Tables[%d]", i)
+		}
+	}
+	if n.Quick {
+		ctx.WritePlain(" QUICK")
+	}
+	return nil
+}
+
 // PlacementOptionType is the type for PlacementOption
 type PlacementOptionType int
 

diff --git a/ast/ddl_test.go b/ast/ddl_test.go
@@ -43,6 +43,7 @@ func TestDDLVisitorCover(t *testing.T) {
 		{&CheckTableStmt{Tables: []*TableName{{}, {}}}, 0, 0},
 		{&RenameTableStmt{TableToTables: []*TableToTable{}}, 0, 0},
 		{&TruncateTableStmt{Table: &TableName{}}, 0, 0},
+		{&RepairTableStmt{Tables: []*TableName{}}, 0, 0},
 
 		// TODO: cover children
 		{&AlterTableStmt{Table: &TableName{}, Specs: []*AlterTableSpec{alterTableSpec}}, 0, 0},