add SEM SQL command for all statements

Signed-off-by: Yang Keao <[email protected]>

Author: YangKeao

pkg/parser/ast/BUILD.bazel

@@ -13,6 +13,7 @@ go_library(
         "misc.go",
         "model.go",
         "procedure.go",
+        "sem.go",
         "stats.go",
         "util.go",
     ],
@@ -46,6 +47,7 @@ go_test(
         "misc_test.go",
         "model_test.go",
         "procedure_test.go",
+        "sem_test.go",
         "util_test.go",
     ],
     embed = [":ast"],

pkg/parser/ast/ast.go

@@ -106,6 +106,12 @@ type FuncNode interface {
 type StmtNode interface {
 	Node
 	statement()
+	// SEMCommand generates a string that represents the command type of the statement.
+	// It's only used for Security Enforcement Mode (SEM) for now. If it's going to be
+	// re-used for other purposes, we may need to rename and give it a clearer definition.
+	//
+	// The function of this method is similar to `GetStmtLabel`, but it returns more detail.
+	SEMCommand() string
 }
 
 // DDLNode represents DDL statement node.
@@ -158,8 +164,6 @@ func GetStmtLabel(stmtNode StmtNode) string {
 		return "AnalyzeTable"
 	case *BeginStmt:
 		return "Begin"
-	case *ChangeStmt:
-		return "Change"
 	case *CommitStmt:
 		return "Commit"
 	case *CompactTableStmt:

pkg/parser/ast/dml.go

@@ -3054,6 +3054,8 @@ const (
 	ShowDistributions
 	ShowPlanForSQL
 	ShowDistributionJobs
+	// showTpCount is the count of all kinds of `SHOW` statements.
+	showTpCount
 )
 
 const (

pkg/parser/ast/misc.go

@@ -61,6 +61,7 @@ var (
 	_ StmtNode = &CompactTableStmt{}
 	_ StmtNode = &SetResourceGroupStmt{}
 	_ StmtNode = &TrafficStmt{}
+	_ StmtNode = &RecommendIndexStmt{}
 
 	_ Node = &PrivElem{}
 	_ Node = &VariableAssignment{}
@@ -1391,41 +1392,6 @@ func (n *SetPwdStmt) Accept(v Visitor) (Node, bool) {
 	return v.Leave(n)
 }
 
-type ChangeStmt struct {
-	stmtNode
-
-	NodeType string
-	State    string
-	NodeID   string
-}
-
-// Restore implements Node interface.
-func (n *ChangeStmt) Restore(ctx *format.RestoreCtx) error {
-	ctx.WriteKeyWord("CHANGE ")
-	ctx.WriteKeyWord(n.NodeType)
-	ctx.WriteKeyWord(" TO NODE_STATE ")
-	ctx.WritePlain("=")
-	ctx.WriteString(n.State)
-	ctx.WriteKeyWord(" FOR NODE_ID ")
-	ctx.WriteString(n.NodeID)
-	return nil
-}
-
-// SecureText implements SensitiveStatement interface.
-func (n *ChangeStmt) SecureText() string {
-	return fmt.Sprintf("change %s to node_state='%s' for node_id '%s'", strings.ToLower(n.NodeType), n.State, n.NodeID)
-}
-
-// Accept implements Node Accept interface.
-func (n *ChangeStmt) Accept(v Visitor) (Node, bool) {
-	newNode, skipChildren := v.Enter(n)
-	if skipChildren {
-		return v.Leave(newNode)
-	}
-	n = newNode.(*ChangeStmt)
-	return v.Leave(n)
-}
-
 // SetRoleStmtType is the type for FLUSH statement.
 type SetRoleStmtType int
 
@@ -2566,6 +2532,8 @@ const (
 	AdminUnsetBDRRole
 	AdminAlterDDLJob
 	AdminWorkloadRepoCreate
+	// adminTpCount is the total number of admin statement types.
+	adminTpCount
 )
 
 // HandleRange represents a range where handle value >= Begin and < End.
@@ -3500,6 +3468,8 @@ const (
 	BRIEKindShowJob
 	BRIEKindShowQuery
 	BRIEKindShowBackupMeta
+	// brieKindCount is the total number of BRIE kinds.
+	brieKindCount
 	// common BRIE options
 	BRIEOptionRateLimit BRIEOptionType = iota + 1
 	BRIEOptionConcurrency

pkg/parser/ast/procedure.go

@@ -46,6 +46,7 @@ var (
 
 	_ ErrNode = &ProcedureErrorCon{}
 	_ ErrNode = &ProcedureErrorVal{}
+	_ ErrNode = &ProcedureErrorState{}
 )
 
 // procedure param type.