[incubator-hugegraph-toolchain] branch master updated: fix: schema not cached in spark/flink loader (#334)

[jin]

This is an automated email from the ASF dual-hosted git repository.

jin pushed a commit to branch master
in repository 
https://gitbox.apache.org/repos/asf/incubator-hugegraph-toolchain.git


The following commit(s) were added to refs/heads/master by this push:
     new 6d82b155 fix: schema not cached in spark/flink loader (#334)
6d82b155 is described below

commit 6d82b1551ed5b6b21fa9e0f21da460f6c849f4d0
Author: alan.zhao <30570711+haohao0...@users.noreply.github.com>
AuthorDate: Tue Sep 20 18:24:45 2022 +0800

    fix: schema not cached in spark/flink loader (#334)
---
 .../java/com/baidu/hugegraph/loader/flink/HugeGraphOutputFormat.java     | 1 +
 .../main/java/com/baidu/hugegraph/loader/spark/HugeGraphSparkLoader.java | 1 +
 2 files changed, 2 insertions(+)

diff --git 
a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/flink/HugeGraphOutputFormat.java
 
b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/flink/HugeGraphOutputFormat.java
index a37c73c3..edc503fd 100644
--- 
a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/flink/HugeGraphOutputFormat.java
+++ 
b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/flink/HugeGraphOutputFormat.java
@@ -90,6 +90,7 @@ public class HugeGraphOutputFormat<T> extends 
RichOutputFormat<T> {
             builders.put(new EdgeBuilder(loadContext, this.struct, 
edgeMapping),
                          new ArrayList<>());
         }
+        loadContext.updateSchemaCache();
         return builders;
     }
 
diff --git 
a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/spark/HugeGraphSparkLoader.java
 
b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/spark/HugeGraphSparkLoader.java
index ec94cf19..143dd2f0 100644
--- 
a/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/spark/HugeGraphSparkLoader.java
+++ 
b/hugegraph-loader/src/main/java/com/baidu/hugegraph/loader/spark/HugeGraphSparkLoader.java
@@ -117,6 +117,7 @@ public class HugeGraphSparkLoader implements Serializable {
             this.builders.put(new EdgeBuilder(context, struct, edgeMapping),
                               new ArrayList<>());
         }
+        context.updateSchemaCache();
         return context;
     }