fixing minor review comment

PeerDB-io · Nov 27, 2023 · 0d24790 · 0d24790
1 parent af46026
commit 0d24790
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/flow/connectors/postgres/cdc.go b/flow/connectors/postgres/cdc.go
@@ -319,12 +319,12 @@ func (p *PostgresCDCSource) consumeStream(
 							TableName:  tableName,
 							PkeyColVal: compositePKeyString,
 						}
-						_, ok := tablePKeyLastSeen[tablePkeyVal]
+						recIndex, ok := tablePKeyLastSeen[tablePkeyVal]
 						if !ok {
 							addRecord(rec)
 							tablePKeyLastSeen[tablePkeyVal] = len(localRecords) - 1
 						} else {
-							oldRec := localRecords[tablePKeyLastSeen[tablePkeyVal]]
+							oldRec := localRecords[recIndex]
 							// iterate through unchanged toast cols and set them in new record
 							updatedCols := r.NewItems.UpdateIfNotExists(oldRec.GetItems())
 							for _, col := range updatedCols {
@@ -362,9 +362,9 @@ func (p *PostgresCDCSource) consumeStream(
 						TableName:  tableName,
 						PkeyColVal: compositePKeyString,
 					}
-					_, ok := tablePKeyLastSeen[tablePkeyVal]
+					recIndex, ok := tablePKeyLastSeen[tablePkeyVal]
 					if ok {
-						latestRecord := localRecords[tablePKeyLastSeen[tablePkeyVal]]
+						latestRecord := localRecords[recIndex]
 						deleteRecord := rec.(*model.DeleteRecord)
 						deleteRecord.Items = latestRecord.GetItems()
 						updateRecord, ok := latestRecord.(*model.UpdateRecord)