diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/io/WriteFiles.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/io/WriteFiles.java index 2fd77a1ad0c6..ae0746294077 100644 --- a/sdks/java/core/src/main/java/org/apache/beam/sdk/io/WriteFiles.java +++ b/sdks/java/core/src/main/java/org/apache/beam/sdk/io/WriteFiles.java @@ -345,18 +345,14 @@ public WriteFiles withNumShards( * * * - *

Note that value provided here cannot exceed the default value ({@link - * #DEFAULT_MAX_NUM_WRITERS_PER_BUNDLE}). + *

The default value is {@link #DEFAULT_MAX_NUM_WRITERS_PER_BUNDLE}. */ public WriteFiles withMaxNumWritersPerBundle( int maxNumWritersPerBundle) { checkArgument( getMaxNumWritersPerBundle() != -1, "Cannot use withMaxNumWritersPerBundle() after withNoSpilling() has been set."); - checkArgument( - maxNumWritersPerBundle > 0 && maxNumWritersPerBundle <= DEFAULT_MAX_NUM_WRITERS_PER_BUNDLE, - "maxNumWritersPerBundle must be greater than 0 and less than or equal to %s", - DEFAULT_MAX_NUM_WRITERS_PER_BUNDLE); + checkArgument(maxNumWritersPerBundle > 0, "maxNumWritersPerBundle must be greater than 0"); return toBuilder().setMaxNumWritersPerBundle(maxNumWritersPerBundle).build(); } diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/io/WriteFilesTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/io/WriteFilesTest.java index 09f3f04d5b09..c8d55a8724e7 100644 --- a/sdks/java/core/src/test/java/org/apache/beam/sdk/io/WriteFilesTest.java +++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/io/WriteFilesTest.java @@ -686,6 +686,8 @@ public void testBuildWrite() { assertTrue(write.withEvictWritersWhenFull().getEvictWritersWhenFull()); assertTrue(write.withEvictWritersWhenFull(true).getEvictWritersWhenFull()); assertFalse(write.withEvictWritersWhenFull(false).getEvictWritersWhenFull()); + assertEquals(50, write.withMaxNumWritersPerBundle(50).getMaxNumWritersPerBundle()); + assertThrows(IllegalArgumentException.class, () -> write.withMaxNumWritersPerBundle(0)); } @Test