Forward skipDanglingRelationships on buffer creation

neo4j · Nov 20, 2023 · 79667c6 · 79667c6
1 parent 412067c
commit 79667c6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/core/src/main/java/org/neo4j/gds/core/loading/SingleTypeRelationshipImporter.java b/core/src/main/java/org/neo4j/gds/core/loading/SingleTypeRelationshipImporter.java
@@ -87,6 +87,10 @@ public int typeId() {
         return this.typeId;
     }
 
+    public boolean skipDanglingRelationships() {
+        return this.importMetaData.skipDanglingRelationships();
+    }
+
     public Collection<AdjacencyBuffer.AdjacencyListBuilderTask> adjacencyListBuilderTasks(Optional<AdjacencyCompressor.ValueMapper> mapper) {
         return adjacencyBuffer.adjacencyListBuilderTasks(mapper, Optional.empty());
     }

diff --git a/native-projection/src/main/java/org/neo4j/gds/projection/RelationshipsScannerTask.java b/native-projection/src/main/java/org/neo4j/gds/projection/RelationshipsScannerTask.java
@@ -148,6 +148,7 @@ public void accept(KernelTransaction transaction) {
                         var buffer = new BufferedRelationshipConsumerBuilder()
                             .idMap(idMap)
                             .type(importer.typeId())
+                            .skipDanglingRelationships(importer.skipDanglingRelationships())
                             .capacity(scanner.bufferSize())
                             .build();