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