[hotfix] [java] Allow setting DataSink parallelism to default value This was missed in FLINK-3589.
Project: http://git-wip-us.apache.org/repos/asf/flink/repo Commit: http://git-wip-us.apache.org/repos/asf/flink/commit/e48a5f19 Tree: http://git-wip-us.apache.org/repos/asf/flink/tree/e48a5f19 Diff: http://git-wip-us.apache.org/repos/asf/flink/diff/e48a5f19 Branch: refs/heads/master Commit: e48a5f19e82ea71ad9c5bf94173299352233ed17 Parents: 0088885 Author: Greg Hogan <c...@greghogan.com> Authored: Thu Apr 21 16:40:39 2016 -0400 Committer: Greg Hogan <c...@greghogan.com> Committed: Thu Apr 21 16:40:39 2016 -0400 ---------------------------------------------------------------------- .../org/apache/flink/api/java/operators/DataSink.java | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/flink/blob/e48a5f19/flink-java/src/main/java/org/apache/flink/api/java/operators/DataSink.java ---------------------------------------------------------------------- diff --git a/flink-java/src/main/java/org/apache/flink/api/java/operators/DataSink.java b/flink-java/src/main/java/org/apache/flink/api/java/operators/DataSink.java index d4b3e3b..2e2d237 100644 --- a/flink-java/src/main/java/org/apache/flink/api/java/operators/DataSink.java +++ b/flink-java/src/main/java/org/apache/flink/api/java/operators/DataSink.java @@ -269,12 +269,13 @@ public class DataSink<T> { * @return This data sink with set parallelism. */ public DataSink<T> setParallelism(int parallelism) { - - if(parallelism < 1) { - throw new IllegalArgumentException("The parallelism of an operator must be at least 1."); + if (parallelism != ExecutionConfig.PARALLELISM_UNKNOWN) { + if (parallelism < 1 && parallelism != ExecutionConfig.PARALLELISM_DEFAULT) { + throw new IllegalArgumentException("The parallelism of an operator must be at least 1."); + } + this.parallelism = parallelism; } - this.parallelism = parallelism; - + return this; } }
