Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,10 @@
/** A set of requirements such as distribution and ordering reported to Spark during writes. */
public class SparkWriteRequirements {

private static final long NO_ADVISORY_PARTITION_SIZE = 0;
public static final SparkWriteRequirements EMPTY =
new SparkWriteRequirements(Distributions.unspecified(), new SortOrder[0], 0);
new SparkWriteRequirements(
Distributions.unspecified(), new SortOrder[0], NO_ADVISORY_PARTITION_SIZE);

private final Distribution distribution;
private final SortOrder[] ordering;
Expand All @@ -37,7 +39,11 @@ public class SparkWriteRequirements {
Distribution distribution, SortOrder[] ordering, long advisoryPartitionSize) {
this.distribution = distribution;
this.ordering = ordering;
this.advisoryPartitionSize = advisoryPartitionSize;
// Spark prohibits requesting a particular advisory partition size without distribution
this.advisoryPartitionSize =
distribution instanceof UnspecifiedDistribution
? NO_ADVISORY_PARTITION_SIZE
: advisoryPartitionSize;
}

public Distribution distribution() {
Expand All @@ -53,7 +59,6 @@ public boolean hasOrdering() {
}

public long advisoryPartitionSize() {
// Spark prohibits requesting a particular advisory partition size without distribution
return distribution instanceof UnspecifiedDistribution ? 0 : advisoryPartitionSize;
return advisoryPartitionSize;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,10 @@
/** A set of requirements such as distribution and ordering reported to Spark during writes. */
public class SparkWriteRequirements {

private static final long NO_ADVISORY_PARTITION_SIZE = 0;
public static final SparkWriteRequirements EMPTY =
new SparkWriteRequirements(Distributions.unspecified(), new SortOrder[0], 0);
new SparkWriteRequirements(
Distributions.unspecified(), new SortOrder[0], NO_ADVISORY_PARTITION_SIZE);

private final Distribution distribution;
private final SortOrder[] ordering;
Expand All @@ -37,7 +39,11 @@ public class SparkWriteRequirements {
Distribution distribution, SortOrder[] ordering, long advisoryPartitionSize) {
this.distribution = distribution;
this.ordering = ordering;
this.advisoryPartitionSize = advisoryPartitionSize;
// Spark prohibits requesting a particular advisory partition size without distribution
this.advisoryPartitionSize =
distribution instanceof UnspecifiedDistribution
? NO_ADVISORY_PARTITION_SIZE
: advisoryPartitionSize;
}

public Distribution distribution() {
Expand All @@ -53,7 +59,6 @@ public boolean hasOrdering() {
}

public long advisoryPartitionSize() {
// Spark prohibits requesting a particular advisory partition size without distribution
return distribution instanceof UnspecifiedDistribution ? 0 : advisoryPartitionSize;
return advisoryPartitionSize;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,10 @@
/** A set of requirements such as distribution and ordering reported to Spark during writes. */
public class SparkWriteRequirements {

private static final long NO_ADVISORY_PARTITION_SIZE = 0;
public static final SparkWriteRequirements EMPTY =
new SparkWriteRequirements(Distributions.unspecified(), new SortOrder[0], 0);
new SparkWriteRequirements(
Distributions.unspecified(), new SortOrder[0], NO_ADVISORY_PARTITION_SIZE);

private final Distribution distribution;
private final SortOrder[] ordering;
Expand All @@ -37,7 +39,11 @@ public class SparkWriteRequirements {
Distribution distribution, SortOrder[] ordering, long advisoryPartitionSize) {
this.distribution = distribution;
this.ordering = ordering;
this.advisoryPartitionSize = advisoryPartitionSize;
// Spark prohibits requesting a particular advisory partition size without distribution
this.advisoryPartitionSize =
distribution instanceof UnspecifiedDistribution
? NO_ADVISORY_PARTITION_SIZE
: advisoryPartitionSize;
}

public Distribution distribution() {
Expand All @@ -53,7 +59,6 @@ public boolean hasOrdering() {
}

public long advisoryPartitionSize() {
// Spark prohibits requesting a particular advisory partition size without distribution
return distribution instanceof UnspecifiedDistribution ? 0 : advisoryPartitionSize;
return advisoryPartitionSize;
}
}
Loading