fix(dataflow): handle nil CurrentBatch in SetBatchLimit

slingdata-io · Dec 1, 2024 · 9fdb5ff · 9fdb5ff
1 parent 578efa8
commit 9fdb5ff
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/dbio/iop/dataflow.go b/core/dbio/iop/dataflow.go
@@ -130,7 +130,9 @@ func (df *Dataflow) SetBatchLimit(limit int64) {
 	defer df.mux.Unlock()
 	for _, ds := range df.Streams {
 		ds.Sp.Config.BatchLimit = limit
-		ds.CurrentBatch.Limit = limit
+		if ds.CurrentBatch != nil {
+			ds.CurrentBatch.Limit = limit
+		}
 	}
 }