Improve code.

RumbleDB · Jul 11, 2024 · 4dbe3b8 · 4dbe3b8
1 parent a4499f0
commit 4dbe3b8
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/org/rumbledb/runtime/flwor/FlworDataFrameUtils.java b/src/main/java/org/rumbledb/runtime/flwor/FlworDataFrameUtils.java
@@ -87,6 +87,7 @@
 import com.esotericsoftware.kryo.io.Output;
 
 import scala.collection.immutable.ArraySeq;
+import scala.collection.Iterator;
 import sparksoniq.spark.SparkSessionManager;
 
 public class FlworDataFrameUtils {
@@ -812,14 +813,15 @@ public static void deserializeWrappedParameters(
             Kryo kryo,
             Input input
     ) {
-        Object[] serializedParams = (Object[]) wrappedParameters.unsafeArray();
-        for (Object serializedParam : serializedParams) {
-            if (serializedParam == null) {
+        Iterator<byte[]> iterator = wrappedParameters.iterator();
+        while (iterator.hasNext()) {
+            byte[] bytes = iterator.next();
+            if (bytes == null) {
                 deserializedParams.add(Collections.emptyList());
                 continue;
             }
             @SuppressWarnings("unchecked")
-            List<Item> deserializedParam = (List<Item>) deserializeByteArray((byte[]) serializedParam, kryo, input);
+            List<Item> deserializedParam = (List<Item>) deserializeByteArray(bytes, kryo, input);
             deserializedParams.add(deserializedParam);
         }
     }