Skip to content

Commit b73106d

Browse files
committed
Merge master Refactoring changes
1 parent 7df3ac5 commit b73106d

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

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

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -511,19 +511,18 @@ examples = """
511511
""",
512512
since = "2.4.0")
513513
case class TransformValues(
514-
input: Expression,
514+
argument: Expression,
515515
function: Expression)
516516
extends MapBasedSimpleHigherOrderFunction with CodegenFallback {
517517

518-
override def nullable: Boolean = input.nullable
518+
override def nullable: Boolean = argument.nullable
519519

520520
override def dataType: DataType = {
521-
val map = input.dataType.asInstanceOf[MapType]
521+
val map = argument.dataType.asInstanceOf[MapType]
522522
MapType(map.keyType, function.dataType, function.nullable)
523523
}
524524

525-
@transient val (keyType, valueType, valueContainsNull) =
526-
HigherOrderFunction.mapKeyValueArgumentType(input.dataType)
525+
@transient val MapType(keyType, valueType, valueContainsNull) = argument.dataType
527526

528527
override def bind(f: (Expression, Seq[(DataType, Boolean)]) => LambdaFunction)
529528
: TransformValues = {
@@ -536,8 +535,8 @@ case class TransformValues(
536535
(keyVar, valueVar)
537536
}
538537

539-
override def nullSafeEval(inputRow: InternalRow, value: Any): Any = {
540-
val map = value.asInstanceOf[MapData]
538+
override def nullSafeEval(inputRow: InternalRow, argumentValue: Any): Any = {
539+
val map = argumentValue.asInstanceOf[MapData]
541540
val f = functionForEval
542541
val resultValues = new GenericArrayData(new Array[Any](map.numElements))
543542
var i = 0

0 commit comments

Comments
 (0)