From 9ea421fe22d4405e60fbeca6eeabcdef92a2a142 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philip=20Dub=C3=A9?= <philip@peerdb.io>
Date: Thu, 4 Jan 2024 06:30:18 +0000
Subject: [PATCH] think activity context needs heartbeat to find out workflow
 was cancelled

---
 flow/connectors/postgres/cdc.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flow/connectors/postgres/cdc.go b/flow/connectors/postgres/cdc.go
index fcb3e64174..fcfc6ccf45 100644
--- 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 {