commit c85c5502c87efdc32f1d65e5633287e908dc1e70
author zhoney <zhangyi89817@126.com> Tue Sep 06 00:02:31 2022 +0800
committer GitHub <noreply@github.com> Tue Sep 06 00:02:31 2022 +0800
tree fea33f76882ee16b91f41027f24bba96894f02d5
parent 9d34f4aaa9debf51f1f8b433d579d6fe8bd9b110

fix: orc data import npe (#321)

Co-authored-by: ss <50285938+ssisme2019@users.noreply.github.com>

hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/reader/file/OrcFileLineFetcher.java

diff --git a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/reader/file/OrcFileLineFetcher.java b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/reader/file/OrcFileLineFetcher.java
index e5f846d..df6e9b9 100644
--- a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/reader/file/OrcFileLineFetcher.java
+++ b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/reader/file/OrcFileLineFetcher.java
@@ -125,9 +125,15 @@
         }
 
         this.row = this.recordReader.next(this.row);
-        Object[] values = this.inspector.getStructFieldsDataAsList(this.row)
-                                        .stream().map(Object::toString)
-                                        .toArray();
+        List<Object> data = this.inspector.getStructFieldsDataAsList(this.row);
+        Object[] values = new Object[data.size()];
+        for (int i = 0; i < data.size(); i++) {
+            if (data.get(i) == null) {
+                values[i] = "";
+            } else {
+                values[i] = data.get(i).toString();
+            }
+        }
         String rawLine = StringUtils.join(values, Constants.COMMA_STR);
 
         this.increaseOffset();