Merge branch 'main' into develop

alibaba · Jun 19, 2024 · 8fc703f · 8fc703f
2 parents 35f85ee + a3daf9c
commit 8fc703f
Show file tree

Hide file tree

Showing 7 changed files with 525 additions and 288 deletions.
diff --git a/flex/storages/rt_mutable_graph/loader/abstract_arrow_fragment_loader.cc b/flex/storages/rt_mutable_graph/loader/abstract_arrow_fragment_loader.cc
@@ -190,8 +190,8 @@ void check_edge_invariant(
 
 void AbstractArrowFragmentLoader::AddVerticesRecordBatch(
     label_t v_label_id, const std::vector<std::string>& v_files,
-    std::function<std::shared_ptr<IRecordBatchSupplier>(
-        label_t, const std::string&, const LoadingConfig&)>
+    std::function<std::vector<std::shared_ptr<IRecordBatchSupplier>>(
+        label_t, const std::string&, const LoadingConfig&, int)>
         supplier_creator) {
   auto primary_keys = schema_.get_vertex_primary_key(v_label_id);
 
@@ -228,8 +228,9 @@ void AbstractArrowFragmentLoader::AddVerticesRecordBatch(
 void AbstractArrowFragmentLoader::AddEdgesRecordBatch(
     label_t src_label_i, label_t dst_label_i, label_t edge_label_i,
     const std::vector<std::string>& filenames,
-    std::function<std::shared_ptr<IRecordBatchSupplier>(
-        label_t, label_t, label_t, const std::string&, const LoadingConfig&)>
+    std::function<std::vector<std::shared_ptr<IRecordBatchSupplier>>(
+        label_t, label_t, label_t, const std::string&, const LoadingConfig&,
+        int)>
         supplier_creator) {
   auto src_label_name = schema_.get_vertex_label_name(src_label_i);
   auto dst_label_name = schema_.get_vertex_label_name(dst_label_i);