Refactor apply method in InsertMapSortInRepartitionExpressions

apache · Jan 7, 2025 · d6592d2 · d6592d2
1 parent 75e323b
commit d6592d2
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/...lyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/InsertMapSortExpression.scala b/...lyst/src/main/scala/org/apache/spark/sql/catalyst/optimizer/InsertMapSortExpression.scala
@@ -90,7 +90,7 @@ object InsertMapSortInRepartitionExpressions extends Rule[LogicalPlan] {
   override def apply(plan: LogicalPlan): LogicalPlan = {
     plan.transformUpWithPruning(_.containsPattern(REPARTITION_OPERATION)) {
       case rep @
-        RepartitionByExpression(partitionExprs, child, optNumPartitions, optAdvisoryPartitionSize)
+        RepartitionByExpression(partitionExprs, _, _, _)
         if rep.partitionExpressions.exists(mapTypeExistsRecursively) =>
         val exprToMapSort = new mutable.HashMap[Expression, Expression]
         val newPartitionExprs = partitionExprs.map { expr =>
@@ -101,9 +101,7 @@ object InsertMapSortInRepartitionExpressions extends Rule[LogicalPlan] {
             expr
           }
         }
-        RepartitionByExpression(
-          newPartitionExprs, child, optNumPartitions, optAdvisoryPartitionSize
-        )
+        rep.copy(partitionExpressions = newPartitionExprs)
     }
   }
 }