parser: support ALTER TABLE ALTER CHECK, ALTER TABLE DROP CHECK s…

…yntax (pingcap#505)
lyonzhi · Aug 20, 2019 · 21816a4 · 21816a4
1 parent 17f828c
commit 21816a4
Show file tree

Hide file tree

Showing 3 changed files with 4,234 additions and 4,161 deletions.
diff --git a/ast/ddl.go b/ast/ddl.go
@@ -1860,6 +1860,8 @@ const (
 	AlterTableRepairPartition
 	AlterTableImportPartitionTablespace
 	AlterTableDiscardPartitionTablespace
+	AlterTableAlterCheck
+	AlterTableDropCheck
 
 	// TODO: Add more actions
 )
@@ -2293,6 +2295,16 @@ func (n *AlterTableSpec) Restore(ctx *RestoreCtx) error {
 		if !n.WithValidation {
 			ctx.WriteKeyWord(" WITHOUT VALIDATION")
 		}
+	case AlterTableAlterCheck:
+		ctx.WriteKeyWord("ALTER CHECK ")
+		ctx.WriteName(n.Constraint.Name)
+		if n.Constraint.Enforced == false {
+			ctx.WriteKeyWord(" NOT")
+		}
+		ctx.WriteKeyWord(" ENFORCED")
+	case AlterTableDropCheck:
+		ctx.WriteKeyWord("DROP CHECK ")
+		ctx.WriteName(n.Constraint.Name)
 	default:
 		// TODO: not support
 		ctx.WritePlainf(" /* AlterTableType(%d) is not supported */ ", n.Tp)