Don't heartbeat forever you muppet

inngest · Oct 12, 2024 · d287ef9 · d287ef9
1 parent ae3fafe
commit d287ef9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/replicator/pgreplicator/pg.go b/pkg/replicator/pgreplicator/pg.go
@@ -253,6 +253,10 @@ func (p *pg) Pull(ctx context.Context, cc chan *changeset.Changeset) error {
 
 		t := time.NewTicker(p.heartbeatTime)
 		for range t.C {
+			if ctx.Err() != nil {
+				return
+			}
+
 			// Send a hearbeat every minute
 			p.queryLock.Lock()
 			_, err := p.queryConn.Exec(ctx, "SELECT pg_logical_emit_message(false, 'heartbeat', now()::varchar);")