From f16f50c3000990f13c3ef2a42fba4ba853cbd764 Mon Sep 17 00:00:00 2001 From: Martin Junghanns Date: Wed, 24 Jan 2024 12:20:54 +0100 Subject: [PATCH] Forward nodeCount to NodesBuilder within NodesFilter This will improve memory usage for ArrayIdMap. Before, we only passed highest id and internally allocated HLA with that id used as capacity. --- .../src/main/java/org/neo4j/gds/beta/filter/NodesFilter.java | 1 + 1 file changed, 1 insertion(+) diff --git a/subgraph-filtering/src/main/java/org/neo4j/gds/beta/filter/NodesFilter.java b/subgraph-filtering/src/main/java/org/neo4j/gds/beta/filter/NodesFilter.java index a01c44ee7e..4c32bb8946 100644 --- a/subgraph-filtering/src/main/java/org/neo4j/gds/beta/filter/NodesFilter.java +++ b/subgraph-filtering/src/main/java/org/neo4j/gds/beta/filter/NodesFilter.java @@ -73,6 +73,7 @@ public static FilteredNodes filterNodes( var nodesBuilder = GraphFactory.initNodesBuilder() .idMapBuilderType(inputNodes.typeId()) .deduplicateIds(false) + .nodeCount(inputNodes.nodeCount()) .concurrency(concurrency) .maxOriginalId(inputNodes.highestOriginalId()) .hasLabelInformation(!inputGraphStore.nodeLabels().isEmpty())