think activity context needs heartbeat to find out workflow was cance…

…lled
PeerDB-io · Jan 4, 2024 · 9ea421f · 9ea421f
1 parent 51760c9
commit 9ea421f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flow/connectors/postgres/cdc.go b/flow/connectors/postgres/cdc.go
@@ -338,6 +338,7 @@ func (p *PostgresCDCSource) consumeStream(
 		var ctx context.Context
 		var cancel context.CancelFunc
 
+		activity.RecordHeartbeat(p.ctx, "consuming stream")
 		if cdcRecordsStorage.IsEmpty() {
 			ctx, cancel = context.WithCancel(p.ctx)
 		} else {