File tree Expand file tree Collapse file tree 1 file changed +5
-6
lines changed
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions Expand file tree Collapse file tree 1 file changed +5
-6
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments