Skip to content

Commit a156150

Browse files
committed
Always pre-aggregate
1 parent 6b54060 commit a156150

File tree

4 files changed

+6
-16
lines changed

4 files changed

+6
-16
lines changed

core/src/main/java/org/neo4j/gds/core/loading/AdjacencyBuffer.java

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,6 @@ public final class AdjacencyBuffer {
6767
private final double[] defaultValues;
6868
private final Aggregation[] aggregations;
6969
private final boolean atLeastOnePropertyToLoad;
70-
private final boolean preAggregate;
7170

7271
public static MemoryEstimation memoryEstimation(
7372
RelationshipType relationshipType,
@@ -110,8 +109,7 @@ public static MemoryEstimation memoryEstimation(
110109
public static AdjacencyBuffer of(
111110
SingleTypeRelationshipImporter.ImportMetaData importMetaData,
112111
AdjacencyCompressorFactory adjacencyCompressorFactory,
113-
ImportSizing importSizing,
114-
boolean preAggregate
112+
ImportSizing importSizing
115113
) {
116114
var numPages = importSizing.numberOfPages();
117115
var pageSize = importSizing.pageSize();
@@ -140,8 +138,7 @@ public static AdjacencyBuffer of(
140138
adjacencyCompressorFactory, localBuilders,
141139
compressedAdjacencyLists,
142140
paging,
143-
atLeastOnePropertyToLoad,
144-
preAggregate
141+
atLeastOnePropertyToLoad
145142
);
146143
}
147144

@@ -151,8 +148,7 @@ private AdjacencyBuffer(
151148
ThreadLocalRelationshipsBuilder[] localBuilders,
152149
ChunkedAdjacencyLists[] chunkedAdjacencyLists,
153150
AdjacencyBufferPaging paging,
154-
boolean atLeastOnePropertyToLoad,
155-
boolean preAggregate
151+
boolean atLeastOnePropertyToLoad
156152
) {
157153
this.adjacencyCompressorFactory = adjacencyCompressorFactory;
158154
this.localBuilders = localBuilders;
@@ -163,7 +159,6 @@ private AdjacencyBuffer(
163159
this.defaultValues = importMetaData.defaultValues();
164160
this.aggregations = importMetaData.aggregations();
165161
this.atLeastOnePropertyToLoad = atLeastOnePropertyToLoad;
166-
this.preAggregate = preAggregate;
167162
}
168163

169164
/**
@@ -216,7 +211,7 @@ void addAll(
216211
if (propertyValues == null) {
217212
compressedTargets.add(localId, targets, startOffset, endOffset, targetsToImport);
218213
} else {
219-
if (preAggregate && aggregations[0] != Aggregation.NONE) {
214+
if (aggregations[0] != Aggregation.NONE) {
220215
targetsToImport = preAggregate(targets, propertyValues, startOffset, endOffset, aggregations);
221216
}
222217
compressedTargets.add(localId, targets, propertyValues, startOffset, endOffset, targetsToImport);

core/src/main/java/org/neo4j/gds/core/loading/SingleTypeRelationshipImporter.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,7 @@ public static SingleTypeRelationshipImporter of(
5454
ImportMetaData importMetaData,
5555
LongSupplier nodeCountSupplier,
5656
boolean validateRelationships,
57-
ImportSizing importSizing,
58-
Optional<Boolean> preAggregate
57+
ImportSizing importSizing
5958
) {
6059
var adjacencyCompressorFactory = AdjacencyListBehavior.asConfigured(
6160
nodeCountSupplier,
@@ -67,7 +66,6 @@ public static SingleTypeRelationshipImporter of(
6766
.importMetaData(importMetaData)
6867
.importSizing(importSizing)
6968
.adjacencyCompressorFactory(adjacencyCompressorFactory)
70-
.preAggregate(preAggregate.orElse(false))
7169
.build();
7270

7371
return new SingleTypeRelationshipImporter(

core/src/main/java/org/neo4j/gds/core/loading/construction/GraphFactory.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -206,8 +206,7 @@ static RelationshipsBuilder relationshipsBuilder(
206206
Optional<Aggregation> aggregation,
207207
Optional<Boolean> validateRelationships,
208208
Optional<Integer> concurrency,
209-
Optional<ExecutorService> executorService,
210-
Optional<Boolean> preAggregate
209+
Optional<ExecutorService> executorService
211210
) {
212211
var loadRelationshipProperties = !propertyConfigs.isEmpty();
213212

@@ -267,7 +266,6 @@ static RelationshipsBuilder relationshipsBuilder(
267266
.nodeCountSupplier(() -> nodes.rootNodeCount().orElse(0L))
268267
.importSizing(importSizing)
269268
.validateRelationships(validateRelationships.orElse(false))
270-
.preAggregate(preAggregate.orElse(false))
271269
.build();
272270

273271
return new RelationshipsBuilder(

test-utils/src/main/java/org/neo4j/gds/core/loading/AdjacencyListBuilderBaseTest.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ void adjacencyListTest(Optional<Long> idOffset) {
6363
.adjacencyCompressorFactory(adjacencyCompressorFactory)
6464
.importMetaData(importMetaData)
6565
.importSizing(ImportSizing.of(1, nodeCount))
66-
.preAggregate(false)
6766
.build();
6867

6968
DirectIdMap idMap = new DirectIdMap(nodeCount);

0 commit comments

Comments
 (0)