[hotfix][table-runtime] Fix ArrayDataSerializer null fields writing

Signed-off-by: slinkydeveloper <[email protected]>
apache · Mar 28, 2022 · c6681b5 · c6681b5
1 parent df0363d
commit c6681b5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/...e-runtime/src/main/java/org/apache/flink/table/runtime/typeutils/ArrayDataSerializer.java b/...e-runtime/src/main/java/org/apache/flink/table/runtime/typeutils/ArrayDataSerializer.java
@@ -125,7 +125,9 @@ private GenericArrayData copyGenericArray(GenericArrayData array) {
                                     LogicalTypeUtils.toInternalConversionClass(eleType),
                                     array.size());
             for (int i = 0; i < array.size(); i++) {
-                newArray[i] = eleSer.copy(objectArray[i]);
+                if (objectArray[i] != null) {
+                    newArray[i] = eleSer.copy(objectArray[i]);
+                }
             }
             return new GenericArrayData(newArray);
         }

diff --git a/...ntime/src/test/java/org/apache/flink/table/runtime/typeutils/ArrayDataSerializerTest.java b/...ntime/src/test/java/org/apache/flink/table/runtime/typeutils/ArrayDataSerializerTest.java
@@ -79,7 +79,10 @@ protected Class<ArrayData> getTypeClass() {
     @Override
     protected ArrayData[] getTestData() {
         return new ArrayData[] {
-            new GenericArrayData(new StringData[] {StringData.fromString("11")}),
+            new GenericArrayData(
+                    new StringData[] {
+                        StringData.fromString("11"), null, StringData.fromString("ke")
+                    }),
             createArray("11", "haa"),
             createArray("11", "haa", "ke"),
             createArray("11", "haa", "ke"),