Add the support of GRANT PRIVILEGE

parser/ast.go

@@ -3866,3 +3866,81 @@ func (e *ExplainExpr) String(level int) string {
 	builder.WriteString(e.Statement.String(level))
 	return builder.String()
 }
+
+type PrivilegeExpr struct {
+	PrivilegePos Pos
+	PrivilegeEnd Pos
+	Keywords     []string
+	Params       *ParamExprList
+}
+
+func (p *PrivilegeExpr) Pos() Pos {
+	return p.PrivilegePos
+}
+
+func (p *PrivilegeExpr) End() Pos {
+	return p.PrivilegeEnd
+}
+
+func (p *PrivilegeExpr) String(level int) string {
+	var builder strings.Builder
+	for i, keyword := range p.Keywords {
+		if i > 0 {
+			builder.WriteByte(' ')
+		}
+		builder.WriteString(keyword)
+	}
+	builder.WriteString(p.Params.String(level))
+	return builder.String()
+}
+
+type GrantPrivilegeExpr struct {
+	GrantPos     Pos
+	StatementEnd Pos
+	OnCluster    *OnClusterExpr
+	Privileges   []*PrivilegeExpr
+	On           *TableIdentifier
+	To           []*Ident
+	WithOptions  []string
+}
+
+func (g *GrantPrivilegeExpr) Pos() Pos {
+	return g.GrantPos
+}
+
+func (g *GrantPrivilegeExpr) End() Pos {
+	return g.StatementEnd
+}
+
+func (g *GrantPrivilegeExpr) Type() string {
+	return "GRANT PRIVILEGE"
+}
+
+func (g *GrantPrivilegeExpr) String(level int) string {
+	var builder strings.Builder
+	builder.WriteString("GRANT ")
+	if g.OnCluster != nil {
+		builder.WriteString(NewLine(level))
+		builder.WriteString(g.OnCluster.String(level))
+	}
+	for i, privilege := range g.Privileges {
+		if i > 0 {
+			builder.WriteString(", ")
+		}
+		builder.WriteString(privilege.String(level))
+	}
+	builder.WriteString(" ON ")
+	builder.WriteString(g.On.String(level))
+	builder.WriteString(" TO ")
+	for i, role := range g.To {
+		if i > 0 {
+			builder.WriteString(", ")
+		}
+		builder.WriteString(role.String(level))
+	}
+	for _, option := range g.WithOptions {
+		builder.WriteString(" WITH " + option + " OPTION")
+	}
+
+	return builder.String()
+}

parser/keyword.go

@@ -2,6 +2,7 @@ package parser
 
 const (
 	KeywordAdd          = "ADD"
+	KeywordAdmin        = "ADMIN"
 	KeywordAfter        = "AFTER"
 	KeywordAlias        = "ALIAS"
 	KeywordAll          = "ALL"
@@ -29,8 +30,10 @@ const (
 	KeywordCodec        = "CODEC"
 	KeywordCollate      = "COLLATE"
 	KeywordColumn       = "COLUMN"
+	KeywordColumns      = "COLUMNS"
 	KeywordComment      = "COMMENT"
 	KeywordCompiled     = "COMPILED"
+	KeywordConfig       = "CONFIG"
 	KeywordConstraint   = "CONSTRAINT"
 	KeywordCreate       = "CREATE"
 	KeywordCross        = "CROSS"
@@ -78,7 +81,9 @@ const (
 	KeywordFrom         = "FROM"
 	KeywordFull         = "FULL"
 	KeywordFunction     = "FUNCTION"
+	KeywordFunctions    = "FUNCTIONS"
 	KeywordGlobal       = "GLOBAL"
+	KeywordGrant        = "GRANT"
 	KeywordGranularity  = "GRANULARITY"
 	KeywordGroup        = "GROUP"
 	KeywordHaving       = "HAVING"
@@ -120,6 +125,7 @@ const (
 	KeywordModify       = "MODIFY"
 	KeywordMonth        = "MONTH"
 	KeywordMove         = "MOVE"
+	KeywordMoves        = "MOVES"
 	KeywordMutation     = "MUTATION"
 	KeywordNan_sql      = "NAN_SQL"
 	KeywordNo           = "NO"
@@ -129,27 +135,34 @@ const (
 	KeywordOffset       = "OFFSET"
 	KeywordOn           = "ON"
 	KeywordOptimize     = "OPTIMIZE"
+	KeywordOption       = "OPTION"
 	KeywordOr           = "OR"
 	KeywordOrder        = "ORDER"
 	KeywordOuter        = "OUTER"
 	KeywordOutfile      = "OUTFILE"
 	KeywordOver         = "OVER"
 	KeywordPartition    = "PARTITION"
 	KeywordPipeline     = "PIPELINE"
+	KeywordPolicy       = "POLICY"
 	KeywordPopulate     = "POPULATE"
 	KeywordPreceding    = "PRECEDING"
 	KeywordPrewhere     = "PREWHERE"
 	KeywordPrimary      = "PRIMARY"
 	KeywordProjection   = "PROJECTION"
 	KeywordQuarter      = "QUARTER"
 	KeywordQuery        = "QUERY"
+	KeywordQueues       = "QUEUES"
+	KeywordQuota        = "QUOTA"
 	KeywordRange        = "RANGE"
+	KeywordRefresh      = "REFRESH"
 	KeywordReload       = "RELOAD"
 	KeywordRemove       = "REMOVE"
 	KeywordRename       = "RENAME"
 	KeywordReplace      = "REPLACE"
 	KeywordReplica      = "REPLICA"
 	KeywordReplicated   = "REPLICATED"
+	KeywordReplication  = "REPLICATION"
+	KeywordRestart      = "RESTART"
 	KeywordRight        = "RIGHT"
 	KeywordRole         = "ROLE"
 	KeywordRollup       = "ROLLUP"
@@ -163,6 +176,7 @@ const (
 	KeywordSet          = "SET"
 	KeywordSettings     = "SETTINGS"
 	KeywordShow         = "SHOW"
+	KeywordShutdown     = "SHUTDOWN"
 	KeywordSource       = "SOURCE"
 	KeywordStart        = "START"
 	KeywordStop         = "STOP"
@@ -208,6 +222,7 @@ const (
 
 var keywords = NewSet(
 	KeywordAdd,
+	KeywordAdmin,
 	KeywordAfter,
 	KeywordAlias,
 	KeywordAll,
@@ -235,8 +250,10 @@ var keywords = NewSet(
 	KeywordCodec,
 	KeywordCollate,
 	KeywordColumn,
+	KeywordColumns,
 	KeywordComment,
 	KeywordCompiled,
+	KeywordConfig,
 	KeywordConstraint,
 	KeywordCreate,
 	KeywordCross,
@@ -284,7 +301,9 @@ var keywords = NewSet(
 	KeywordFrom,
 	KeywordFull,
 	KeywordFunction,
+	KeywordFunctions,
 	KeywordGlobal,
+	KeywordGrant,
 	KeywordGranularity,
 	KeywordGroup,
 	KeywordHaving,
@@ -326,6 +345,7 @@ var keywords = NewSet(
 	KeywordModify,
 	KeywordMonth,
 	KeywordMove,
+	KeywordMoves,
 	KeywordMutation,
 	KeywordNan_sql,
 	KeywordNo,
@@ -335,27 +355,33 @@ var keywords = NewSet(
 	KeywordOffset,
 	KeywordOn,
 	KeywordOptimize,
+	KeywordOption,
 	KeywordOr,
 	KeywordOrder,
 	KeywordOuter,
 	KeywordOutfile,
 	KeywordOver,
 	KeywordPartition,
 	KeywordPipeline,
+	KeywordPolicy,
 	KeywordPopulate,
 	KeywordPreceding,
 	KeywordPrewhere,
 	KeywordPrimary,
 	KeywordProjection,
 	KeywordQuarter,
 	KeywordQuery,
+	KeywordQueues,
+	KeywordQuota,
 	KeywordRange,
 	KeywordReload,
 	KeywordRemove,
 	KeywordRename,
 	KeywordReplace,
 	KeywordReplica,
 	KeywordReplicated,
+	KeywordReplication,
+	KeywordRestart,
 	KeywordRight,
 	KeywordRole,
 	KeywordRollup,
@@ -369,6 +395,7 @@ var keywords = NewSet(
 	KeywordSet,
 	KeywordSettings,
 	KeywordShow,
+	KeywordShutdown,
 	KeywordSource,
 	KeywordStart,
 	KeywordStop,