ut

Author: iamlinjunhong

pkg/sql/compile/types.go

@@ -90,8 +90,6 @@ func (m magicType) String() string {
 		return "DropDatabase"
 	case DropTable:
 		return "DropTable"
-	case DropPitr:
-		return "DropPitr"
 	case DropIndex:
 		return "DropIndex"
 	case TruncateTable:

pkg/sql/parsers/dialect/mysql/mysql_sql_test.go

@@ -16,6 +16,7 @@ package mysql
 
 import (
 	"context"
+	"github.com/stretchr/testify/require"
 	"testing"
 
 	"github.com/matrixorigin/matrixone/pkg/sql/parsers/dialect"
@@ -81,6 +82,39 @@ func TestOriginSQL(t *testing.T) {
 	}
 }
 
+var (
+	pitrSQLs = []struct {
+		input  string
+		output string
+	}{
+		{
+			input:  "create pitr pitr_d for database db1 range 1 'd' internal",
+			output: "create pitr pitr_d for database db1 range 1  d internal",
+		}, {
+			input:  "drop pitr pitr_d internal",
+			output: "drop pitr pitr_d internal",
+		},
+	}
+)
+
+func TestPitrInternal(t *testing.T) {
+	for _, pitrSQL := range pitrSQLs {
+		if pitrSQL.output == "" {
+			pitrSQL.output = pitrSQL.input
+		}
+		ast, err := ParseOne(context.TODO(), pitrSQL.input, 1)
+		if err != nil {
+			t.Errorf("Parse(%q) err: %v", pitrSQL.input, err)
+			return
+		}
+		out := tree.String(ast, dialect.MYSQL)
+		if pitrSQL.output != out {
+			t.Errorf("Parsing failed. \nExpected/Got:\n%s\n%s", pitrSQL.output, out)
+		}
+		require.Equal(t, ast.StmtKind().ExecLocation(), tree.EXEC_IN_ENGINE)
+	}
+}
+
 var (
 	validSQL = []struct {
 		input  string

pkg/sql/parsers/tree/pitr.go

@@ -88,6 +88,9 @@ func (node *DropPitr) Format(ctx *FmtCtx) {
 		ctx.WriteString("if exists ")
 	}
 	node.Name.Format(ctx)
+	if node.Internal {
+		ctx.WriteString(" internal")
+	}
 }
 
 func (node *DropPitr) reset() { *node = DropPitr{} }
@@ -151,6 +154,9 @@ func (node *CreatePitr) Format(ctx *FmtCtx) {
 	ctx.WriteString(fmt.Sprintf("%v ", node.PitrValue))
 	ctx.WriteString(" ")
 	ctx.WriteString(node.PitrUnit)
+	if node.Internal {
+		ctx.WriteString(" internal")
+	}
 }
 
 func (node *CreatePitr) GetStatementType() string { return "Create PITR" }