1
0

Fixed partitions produced by layout optimization in case order-by key is composed of a single column (#4183)

This commit is contained in:
Alexey Kudinkin
2021-12-01 20:56:04 -08:00
committed by GitHub
parent 5284730175
commit 772f5ca24e

View File

@@ -100,7 +100,7 @@ public class OrderingIndexHelper {
}
// only one col to sort, no need to use z-order
if (sortCols.size() == 1) {
return df.repartitionByRange(fieldNum, org.apache.spark.sql.functions.col(sortCols.get(0)));
return df.repartitionByRange(fileNum, org.apache.spark.sql.functions.col(sortCols.get(0)));
}
Map<Integer, StructField> fieldMap = sortCols
.stream().collect(Collectors.toMap(e -> Arrays.asList(df.schema().fields()).indexOf(columnsMap.get(e)), e -> columnsMap.get(e)));