cron workflows must return canceled error explicitly (#1363)

PeerDB-io · Feb 23, 2024 · 4bafd25 · 4bafd25
1 parent 803380c
commit 4bafd25
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flow/workflows/scheduled_flows.go b/flow/workflows/scheduled_flows.go
@@ -10,6 +10,9 @@ import (
 
 // RecordSlotSizeWorkflow monitors replication slot size
 func RecordSlotSizeWorkflow(ctx workflow.Context) error {
+	if ctx.Err() != nil {
+		return ctx.Err()
+	}
 	ctx = workflow.WithActivityOptions(ctx, workflow.ActivityOptions{
 		StartToCloseTimeout: time.Hour,
 	})
@@ -19,6 +22,9 @@ func RecordSlotSizeWorkflow(ctx workflow.Context) error {
 
 // HeartbeatFlowWorkflow sends WAL heartbeats
 func HeartbeatFlowWorkflow(ctx workflow.Context) error {
+	if ctx.Err() != nil {
+		return ctx.Err()
+	}
 	ctx = workflow.WithActivityOptions(ctx, workflow.ActivityOptions{
 		StartToCloseTimeout: time.Hour,
 	})