Set lower bound on desiredBundleSizeBytes to 0

apache · Oct 3, 2023 · 4cf0310 · 4cf0310
1 parent f06b1f8
commit 4cf0310
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...o/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigtable/BigtableIO.java b/...o/google-cloud-platform/src/main/java/org/apache/beam/sdk/io/gcp/bigtable/BigtableIO.java
@@ -1326,7 +1326,7 @@ public List<BigtableSource> split(long desiredBundleSizeBytes, PipelineOptions o
  long maximumNumberOfSplits = 4000;
  long sizeEstimate = getEstimatedSizeBytes(options);
  desiredBundleSizeBytes =
- Math.max(sizeEstimate / maximumNumberOfSplits, desiredBundleSizeBytes);
+ Math.max(sizeEstimate / maximumNumberOfSplits, Math.max(1, desiredBundleSizeBytes));
 
  // Delegate to testable helper.
  List<BigtableSource> splits =