Skip to content

Commit

Permalink
[SQL] SPJ: Add documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
szehon-ho committed May 25, 2024
1 parent 3cb30c2 commit bc9bca2
Showing 1 changed file with 97 additions and 0 deletions.
97 changes: 97 additions & 0 deletions docs/sql-performance-tuning.md
Original file line number Diff line number Diff line change
Expand Up @@ -428,3 +428,100 @@ You can control the details of how AQE works by providing your own cost evaluato
<td>3.2.0</td>
</tr>
</table>

## Storage Partition Join

Storage Partition Join (SPJ) is an optimization technique in Spark SQL that makes use the existing storage layout to avoid the shuffle phase.

This is a generalization of the concept of Bucket Joins, which is only applicable for [bucketed](sql-data-sources-load-save-functions.html#bucketing-sorting-and-partitioning) tables, to any table partitioned by function registered in FunctionCatalog. Storage Partition Joins are currently supported for compatible V2 DataSources.

The following SQL properties enable Storage Partition Join.

<table class="spark-config">
<thead><tr><th>Property Name</th><th>Default</th><th>Meaning</th><th>Since Version</th></tr></thead>
<tr>
<td><code>spark.sql.sources.v2.bucketing.enabled</code></td>
<td>false</td>
<td>
When true, try to eliminate shuffle by using the partitioning reported by a compatible V2 data source.
</td>
<td>3.3.0</td>
</tr>
<tr>
<td><code>spark.sql.sources.v2.bucketing.pushPartValues.enabled</code></td>
<td>true</td>
<td>
When enabled, try to eliminate shuffle if one side of the join has missing partition values from the other side. This config requires <code>spark.sql.sources.v2.bucketing.enabled</code> to be true.
</td>
<td>3.4.0</td>
</tr>
<tr>
<td><code>spark.sql.requireAllClusterKeysForCoPartition</code></td>
<td>true</td>
<td>
When true, require the join or MERGE keys to be same and in the same order as the partition keys to eliminate shuffle. Hence, set to <b>false</b> in this situation to eliminate shuffle.
</td>
<td>3.4.0</td>
</tr>
<tr>
<td><code>spark.sql.sources.v2.bucketing.partiallyClusteredDistribution.enabled</code></td>
<td>false</td>
<td>
When true, and when the join is not a full outer join, enable skew optimizations to handle partitions with large amounts of data when avoiding shuffle. One side will be chosen as the big table based on table statistics, and the splits on this side will be partially-clustered. The splits of the other side will be grouped and replicated to match. This config requires both <code>spark.sql.sources.v2.bucketing.enabled</code> and <code>spark.sql.sources.v2.bucketing.pushPartValues.enabled</code> to be true.
<td>3.4.0</td>
</tr>
<tr>
<td><code>spark.sql.sources.v2.bucketing.allowJoinKeysSubsetOfPartitionKeys.enabled</code></td>
<td>false</td>
<td>
When enabled, try to avoid shuffle if join or MERGE condition does not include all partition columns. This config requires both <code>spark.sql.sources.v2.bucketing.enabled</code> and <code>spark.sql.sources.v2.bucketing.pushPartValues.enabled</code> to be true, and <code>spark.sql.requireAllClusterKeysForCoPartition</code> to be false.
<td>4.0.0</td>
</tr>
<tr>
<td><code>spark.sql.sources.v2.bucketing.allowCompatibleTransforms.enabled</code></td>
<td>false</td>
<td>
When enabled, try to avoid shuffle if partition transforms are compatible but not identical. This config requires both <code>spark.sql.sources.v2.bucketing.enabled</code> and <code>spark.sql.sources.v2.bucketing.pushPartValues.enabled</code> to be true.</td>
<td>4.0.0</td>
</tr>
<tr>
<td><code>spark.sql.sources.v2.bucketing.shuffle.enabled</code></td>
<td>false</td>
<td>
When enabled, try to avoid shuffle on one side of the join, by recognizing the partitioning reported by a V2 data source on the other side.
<td>4.0.0</td>
</tr>
</table>

If Storage Partition Join is performed, the query plan will not contain Exchange nodes prior to the join.

```sql
-- Plan without Storage Partition Join

== Physical Plan ==
* Project (12)
+- * SortMergeJoin Inner (11)
:- * Sort (5)
: +- Exchange (4) // DATA SHUFFLE
: +- * Filter (3)
: +- * ColumnarToRow (2)
: +- BatchScan (1)
+- * Sort (10)
+- Exchange (9) // DATA SHUFFLE
+- * Filter (8)
+- * ColumnarToRow (7)
+- BatchScan (6)

-- Plan with Storage Partition Join

== Physical Plan ==
* Project (10)
+- * SortMergeJoin Inner (9)
:- * Sort (4)
: +- * Filter (3)
: +- * ColumnarToRow (2)
: +- BatchScan (1)
+- * Sort (8)
+- * Filter (7)
+- * ColumnarToRow (6)
+- BatchScan (5)

0 comments on commit bc9bca2

Please sign in to comment.