msamirkhan commented on a change in pull request #29353:
URL: https://github.com/apache/spark/pull/29353#discussion_r465964580
##########
File path:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/SpecificInternalRow.scala
##########
@@ -192,24 +192,41 @@ final class MutableAny extends MutableValue {
*/
final class SpecificInternalRow(val values: Array[MutableValue]) extends
BaseGenericInternalRow {
- def this(dataTypes: Seq[DataType]) =
- this(
- dataTypes.map {
- case BooleanType => new MutableBoolean
- case ByteType => new MutableByte
- case ShortType => new MutableShort
- // We use INT for DATE internally
- case IntegerType | DateType => new MutableInt
- // We use Long for Timestamp internally
- case LongType | TimestampType => new MutableLong
- case FloatType => new MutableFloat
- case DoubleType => new MutableDouble
- case _ => new MutableAny
- }.toArray)
+ private[this] def constructorHelper(dataType: DataType): MutableValue =
dataType match {
+ case BooleanType => new MutableBoolean
+ case ByteType => new MutableByte
+ case ShortType => new MutableShort
+ // We use INT for DATE internally
+ case IntegerType | DateType => new MutableInt
+ // We use Long for Timestamp internally
+ case LongType | TimestampType => new MutableLong
+ case FloatType => new MutableFloat
+ case DoubleType => new MutableDouble
+ case _ => new MutableAny
+ }
+
+ def this(dataTypes: Seq[DataType]) = {
+ this(new Array[MutableValue](dataTypes.length))
+ val length = values.length
+ var i = 0
+ while (i < length) {
+ values(i) = constructorHelper(dataTypes(i))
+ i += 1
+ }
+ }
def this() = this(Seq.empty)
- def this(schema: StructType) = this(schema.fields.map(_.dataType))
+ def this(schema: StructType) = {
+ this(new Array[MutableValue](schema.fields.length))
+ val length = values.length
+ val fields = schema.fields
+ var i = 0
+ while (i < length) {
+ values(i) = constructorHelper(fields(i).dataType)
+ i += 1
+ }
+ }
override def numFields: Int = values.length
Review comment:
Accidentally attached, will remove.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org
For additional commands, e-mail: reviews-h...@spark.apache.org
