Update encoder.go

pingcap · Feb 18, 2025 · 99300a1 · 99300a1
1 parent 6cd203d
commit 99300a1
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/pkg/sink/codec/canal/encoder.go b/pkg/sink/codec/canal/encoder.go
@@ -601,14 +601,17 @@ type JSONTxnEventEncoder struct {
 	txnCommitTs uint64
 	txnSchema   *string
 	txnTable    *string
+
+	columnSelector columnselector.Selector
 }
 
 // NewJSONTxnEventEncoderBuilder creates a new JSONTxnEventEncoder
 func NewJSONTxnEventEncoderBuilder(config *common.Config) common.TxnEventEncoder {
 	return &JSONTxnEventEncoder{
-		valueBuf:   &bytes.Buffer{},
-		terminator: []byte(config.Terminator),
-		config:     config,
+		valueBuf:       &bytes.Buffer{},
+		terminator:     []byte(config.Terminator),
+		columnSelector: columnselector.NewDefaultColumnSelector(),
+		config:         config,
 	}
 }
 
@@ -620,9 +623,10 @@ func (j *JSONTxnEventEncoder) AppendTxnEvent(event *commonEvent.DMLEvent) error
 			break
 		}
 		value, err := newJSONMessageForDML(&commonEvent.RowEvent{
-			TableInfo: event.TableInfo,
-			CommitTs:  event.CommitTs,
-			Event:     row,
+			TableInfo:      event.TableInfo,
+			CommitTs:       event.CommitTs,
+			Event:          row,
+			ColumnSelector: j.columnSelector,
 		}, j.config, false, "")
 		if err != nil {
 			return err