Skip to content

Commit c5b7f23

Browse files
committed
add required child distribution
1 parent 12cf02a commit c5b7f23

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

sql/core/src/main/scala/org/apache/spark/sql/execution/streaming/statefulOperators.scala

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,14 @@ case class StateStoreSaveExec(
366366
override def output: Seq[Attribute] = child.output
367367

368368
override def outputPartitioning: Partitioning = child.outputPartitioning
369+
370+
override def requiredChildDistribution: Seq[Distribution] = {
371+
if (keyExpressions.isEmpty) {
372+
AllTuples :: Nil
373+
} else {
374+
ClusteredDistribution(keyExpressions) :: Nil
375+
}
376+
}
369377
}
370378

371379
/** Physical operator for executing streaming Deduplicate. */

0 commit comments

Comments
 (0)