Skip to content

Commit f1afb92

Browse files
committed
fix stack
1 parent 18ec598 commit f1afb92

File tree

1 file changed

+5
-6
lines changed
  • sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions

1 file changed

+5
-6
lines changed

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/generators.scala

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,11 @@ case class Stack(children: Seq[Expression]) extends Generator {
201201
// Rows - we write these into an array.
202202
val rowData = ctx.addMutableState("InternalRow[]", "rows",
203203
v => s"$v = new InternalRow[$numRows];")
204+
// Create the collection.
205+
val wrapperClass = classOf[mutable.WrappedArray[_]].getName
206+
ev.value = ctx.addMutableState(s"$wrapperClass<InternalRow>", ev.value,
207+
v => s"$v = $wrapperClass$$.MODULE$$.make($rowData);")
208+
204209
val values = children.tail
205210
val dataTypes = values.take(numFields).map(_.dataType)
206211
val code = ctx.splitExpressionsWithCurrentInputs(Seq.tabulate(numRows) { row =>
@@ -212,12 +217,6 @@ case class Stack(children: Seq[Expression]) extends Generator {
212217
s"${eval.code}\n$rowData[$row] = ${eval.value};"
213218
})
214219

215-
// Create the collection.
216-
val wrapperClass = classOf[mutable.WrappedArray[_]].getName
217-
ctx.addMutableState(
218-
s"$wrapperClass<InternalRow>",
219-
ev.value,
220-
v => s"$v = $wrapperClass$$.MODULE$$.make($rowData);", useFreshName = false)
221220
ev.copy(code = code, isNull = "false")
222221
}
223222
}

0 commit comments

Comments
 (0)