Fixes java version when arrays have different lengths

Signed-off-by: DylanGuedes <[email protected]>

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

@@ -157,22 +157,49 @@ case class ZipLists(left: Expression, right: Expression)
       val i = ctx.freshName("i")
       val values = ctx.freshName("values")
       val len1 = ctx.freshName("len1")
+      val schema = ctx.freshName("schema")
+      val len2 = ctx.freshName("len2")
       val pair = ctx.freshName("pair")
+      val higher = ctx.freshName("higher")
+      val leftType = left.dataType.asInstanceOf[ArrayType].elementType
+      val rightType = right.dataType.asInstanceOf[ArrayType].elementType
       val getValue1 = CodeGenerator.getValue(
         arr1, left.dataType.asInstanceOf[ArrayType].elementType, i)
       val getValue2 = CodeGenerator.getValue(
         arr2, right.dataType.asInstanceOf[ArrayType].elementType, i)
 
       s"""
       int $len1 = $arr1.numElements();
+      int $len2 = $arr2.numElements();
+      int $higher = $len2;
+
       Object[] $values;
-      $values = new Object[$len1];
-      for (int $i = 0; $i < $len1; $i ++) {
-        Object[] $pair;
-        $pair = new Object[2];
-        $pair[0] = $getValue1;
-        $pair[1] = $getValue2;
-        $values[$i] = new $genericInternalRow($pair);
+      if ($len1 > $len2) {
+        $values = new Object[$len1];
+        for (int $i = 0; $i < $len1; $i ++) {
+          Object[] $pair;
+          $pair = new Object[2];
+          $pair[0] = $getValue1;
+          if ($i >= $len2) {
+            $pair[1] = null;
+          } else {
+            $pair[1] = $getValue2;
+          }
+          $values[$i] = new $genericInternalRow($pair);
+        }
+      } else {
+        $values = new Object[$len2];
+        for (int $i = 0; $i < $len2; $i ++) {
+          Object[] $pair;
+          $pair = new Object[2];
+          $pair[1] = $getValue2;
+          if ($i >= $len1) {
+            $pair[0] = null;
+          } else {
+            $pair[0] = $getValue1;
+          }
+          $values[$i] = new $genericInternalRow($pair);
+        }
       }
       ${ev.value} = new $genericArrayData($values);
       """