Never ask for a bundle size of 0

apache · Oct 3, 2023 · f06b1f8 · f06b1f8
1 parent 33d52a8
commit f06b1f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...direct-java/src/main/java/org/apache/beam/runners/direct/BoundedReadEvaluatorFactory.java b/...direct-java/src/main/java/org/apache/beam/runners/direct/BoundedReadEvaluatorFactory.java
@@ -213,7 +213,7 @@ public Collection<CommittedBundle<BoundedSourceShard<T>>> getInitialInputs(
  throws Exception {
  BoundedSource<T> source = ReadTranslation.boundedSourceFromTransform(transform);
  long estimatedBytes = source.getEstimatedSizeBytes(options);
- long bytesPerBundle = estimatedBytes / targetParallelism;
+ long bytesPerBundle = Math.max(1L, estimatedBytes / targetParallelism);
  List<? extends BoundedSource<T>> bundles = source.split(bytesPerBundle, options);
  ImmutableList.Builder<CommittedBundle<BoundedSourceShard<T>>> shards =
  ImmutableList.builder();