update max batch size on signal (#910)

PeerDB-io · Dec 27, 2023 · f3d1d8e · f3d1d8e
1 parent a3b2800
commit f3d1d8e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/flow/workflows/cdc_flow.go b/flow/workflows/cdc_flow.go
@@ -291,7 +291,10 @@ func CDCFlowWorkflowWithConfig(
 		c.Receive(ctx, &batchSize)
 		w.logger.Info("received batch size signal: ", batchSize)
 		syncFlowOptions.BatchSize = batchSize
+		cfg.MaxBatchSize = uint32(batchSize)
+		limits.MaxBatchSize = int(batchSize)
 	})
+
 	batchSizeSelector.AddDefault(func() {
 		w.logger.Info("no batch size signal received, batch size remains: ",
 			syncFlowOptions.BatchSize)