From 21c8f05ca431a767d9235ccfd3f63755808ca283 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philip=20Dub=C3=A9?= <philip@peerdb.io>
Date: Tue, 4 Jun 2024 17:12:24 +0000
Subject: [PATCH] cdc replay from catalog

---
 flow/connectors/postgres/cdc.go | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/flow/connectors/postgres/cdc.go b/flow/connectors/postgres/cdc.go
index 92f6827182..acdeb105b0 100644
--- a/flow/connectors/postgres/cdc.go
+++ b/flow/connectors/postgres/cdc.go
@@ -580,8 +580,34 @@ func (rp *cdcRecordProcessor[Items]) processMessage(
 		rp.records.UpdateLatestCheckpoint(int64(msg.CommitLSN))
 		p.commitLock = nil
 	case *pglogrepl.StreamCommitMessageV2:
-		// TODO first replay streams from catalog
-		// TODO select streams from v2cdc where flow_name = $1 and xid = $2 order by lsn
+		// TODO track first stream bit so we can skip reading catalog when transaction is in single batch
+		rows, err := p.CatalogPool.Query(ctx,
+			"select stream from v2cdc where flow_name = $1 and xid = $2 order by lsn",
+			p.FlowJobName, msg.Xid)
+		if err != nil {
+			return err
+		}
+		for rows.Next() {
+			var stream [][]byte
+			if err := rows.Scan(&stream); err != nil {
+				return err
+			}
+
+			for _, m := range stream {
+				mxld, err := pglogrepl.ParseXLogData(m)
+				if err != nil {
+					return err
+				}
+				logicalMsg, err = pglogrepl.ParseV2(mxld.WALData, p.inStream)
+				if err != nil {
+					return err
+				}
+				if err := rp.processMessage(ctx, p, mxld.WALStart, logicalMsg, currentClientXlogPos); err != nil {
+					return err
+				}
+			}
+		}
+
 		txbufs := p.txBuffer[msg.Xid]
 		for _, txbuf := range txbufs {
 			for _, m := range txbuf.Streams {