parser: fix compile error with mod function

Author: Defined2014

pkg/parser/ast/expressions.go

@@ -85,6 +85,9 @@ type BetweenExpr struct {
 
 // Restore implements Node interface.
 func (n *BetweenExpr) Restore(ctx *format.RestoreCtx) error {
+	if ctx.Flags.HasRestoreBracketAroundBetweenExpr() {
+		ctx.WritePlain("(")
+	}
 	if err := n.Expr.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred while restore BetweenExpr.Expr")
 	}
@@ -100,6 +103,9 @@ func (n *BetweenExpr) Restore(ctx *format.RestoreCtx) error {
 	if err := n.Right.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred while restore BetweenExpr.Right ")
 	}
+	if ctx.Flags.HasRestoreBracketAroundBetweenExpr() {
+		ctx.WritePlain(")")
+	}
 	return nil
 }
 
@@ -175,6 +181,8 @@ func (n *BinaryOperationExpr) Restore(ctx *format.RestoreCtx) error {
 	if ctx.Flags.HasRestoreBracketAroundBinaryOperation() {
 		ctx.WritePlain("(")
 	}
+	originalFlags := ctx.Flags
+	ctx.Flags |= format.RestoreBracketAroundBetweenExpr
 	if err := n.L.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred when restore BinaryOperationExpr.L")
 	}
@@ -184,6 +192,7 @@ func (n *BinaryOperationExpr) Restore(ctx *format.RestoreCtx) error {
 	if err := n.R.Restore(ctx); err != nil {
 		return errors.Annotate(err, "An error occurred when restore BinaryOperationExpr.R")
 	}
+	ctx.Flags = originalFlags
 	if ctx.Flags.HasRestoreBracketAroundBinaryOperation() {
 		ctx.WritePlain(")")
 	}

pkg/parser/format/format.go

@@ -239,6 +239,8 @@ const (
 	RestoreWithoutSchemaName
 	RestoreWithoutTableName
 	RestoreForNonPrepPlanCache
+
+	RestoreBracketAroundBetweenExpr
 )
 
 const (
@@ -323,6 +325,12 @@ func (rf RestoreFlags) HasStringWithoutDefaultCharset() bool {
 	return rf.has(RestoreStringWithoutDefaultCharset)
 }
 
+// HasRestoreBracketAroundBetweenExpr returns a boolean indicating
+// whether `rf` has `RestoreBracketAroundBetweenExpr` flag.
+func (rf RestoreFlags) HasRestoreBracketAroundBetweenExpr() bool {
+	return rf.has(RestoreBracketAroundBetweenExpr)
+}
+
 // HasStringWithoutCharset returns a boolean indicating whether `rf` has `RestoreStringWithoutCharset` flag.
 func (rf RestoreFlags) HasStringWithoutCharset() bool {
 	return rf.has(RestoreStringWithoutCharset)