parser: Add ADD_COLUMNAR_REPLICA_ON_DEMAND (pingcap#61159)

ref pingcap#54245

Author: breezewish

pkg/parser/ast/ddl.go

@@ -736,15 +736,16 @@ const (
 type IndexOption struct {
 	node
 
-	KeyBlockSize        uint64
-	Tp                  IndexType
-	Comment             string
-	ParserName          CIStr
-	Visibility          IndexVisibility
-	PrimaryKeyTp        PrimaryKeyType
-	Global              bool
-	SplitOpt            *SplitOption `json:"-"` // SplitOption contains expr nodes, which cannot marshal for DDL job arguments.
-	SecondaryEngineAttr string
+	KeyBlockSize               uint64
+	Tp                         IndexType
+	Comment                    string
+	ParserName                 CIStr
+	Visibility                 IndexVisibility
+	PrimaryKeyTp               PrimaryKeyType
+	Global                     bool
+	SplitOpt                   *SplitOption `json:"-"` // SplitOption contains expr nodes, which cannot marshal for DDL job arguments.
+	SecondaryEngineAttr        string
+	AddColumnarReplicaOnDemand int
 }
 
 // IsEmpty is true if only default options are given
@@ -767,7 +768,16 @@ func (n *IndexOption) IsEmpty() bool {
 // Restore implements Node interface.
 func (n *IndexOption) Restore(ctx *format.RestoreCtx) error {
 	hasPrevOption := false
+
+	if n.AddColumnarReplicaOnDemand > 0 {
+		ctx.WriteKeyWord("ADD_COLUMNAR_REPLICA_ON_DEMAND")
+		hasPrevOption = true
+	}
+
 	if n.PrimaryKeyTp != PrimaryKeyTypeDefault {
+		if hasPrevOption {
+			ctx.WritePlain(" ")
+		}
 		_ = ctx.WriteWithSpecialComments(tidb.FeatureIDClusteredIndex, func() error {
 			ctx.WriteKeyWord(n.PrimaryKeyTp.String())
 			return nil

pkg/parser/keywords.go

@@ -36,7 +36,7 @@ func TestKeywords(t *testing.T) {
 }
 
 func TestKeywordsLength(t *testing.T) {
-	require.Equal(t, 662, len(parser.Keywords))
+	require.Equal(t, 663, len(parser.Keywords))
 
 	reservedNr := 0
 	for _, kw := range parser.Keywords {