Use ctx.nullSafeExec() to generate nullability checking in complex type creators.

Author: ueshin

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

@@ -67,13 +67,10 @@ case class CreateArray(children: Seq[Expression]) extends Expression {
         ctx.INPUT_ROW,
         children.zipWithIndex.map { case (e, i) =>
           val eval = e.genCode(ctx)
-          eval.code + s"""
-            if (${eval.isNull}) {
-              $values[$i] = null;
-            } else {
+          eval.code +
+            ctx.nullSafeExec(e.nullable, eval.isNull)(s"""
               $values[$i] = ${eval.value};
-            }
-           """
+            """)
         }) +
       s"""
         final ArrayData ${ev.value} = new $arrayClass($values);
@@ -164,14 +161,10 @@ case class CreateMap(children: Seq[Expression]) extends Expression {
         ctx.INPUT_ROW,
         values.zipWithIndex.map { case (value, i) =>
           val eval = value.genCode(ctx)
-          s"""
-            ${eval.code}
-            if (${eval.isNull}) {
-              $valueArray[$i] = null;
-            } else {
+          eval.code +
+            ctx.nullSafeExec(value.nullable, eval.isNull)(s"""
               $valueArray[$i] = ${eval.value};
-            }
-          """
+            """)
         }) +
       s"""
         final MapData ${ev.value} = new $mapClass($keyData, $valueData);
@@ -307,12 +300,10 @@ case class CreateNamedStruct(children: Seq[Expression]) extends CreateNamedStruc
         ctx.INPUT_ROW,
         valExprs.zipWithIndex.map { case (e, i) =>
           val eval = e.genCode(ctx)
-          eval.code + s"""
-          if (${eval.isNull}) {
-            $values[$i] = null;
-          } else {
-            $values[$i] = ${eval.value};
-          }"""
+          eval.code +
+            ctx.nullSafeExec(e.nullable, eval.isNull)(s"""
+              $values[$i] = ${eval.value};
+            """)
         }) +
       s"""
         final InternalRow ${ev.value} = new $rowClass($values);