Skip to content

Commit 2f7abf5

Browse files
Add k1Coloring to business facade
1 parent d20f00b commit 2f7abf5

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

algorithms-compute-business-facade/src/main/java/org/neo4j/gds/community/CommunityComputeBusinessFacade.java

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,13 @@
3232
import org.neo4j.gds.conductance.ConductanceResult;
3333
import org.neo4j.gds.core.JobId;
3434
import org.neo4j.gds.core.loading.GraphStoreCatalogService;
35+
import org.neo4j.gds.core.loading.validation.NoAlgorithmValidation;
3536
import org.neo4j.gds.core.loading.validation.NodePropertyAnyExistsGraphStoreValidation;
3637
import org.neo4j.gds.core.loading.validation.UndirectedOnlyGraphStoreValidation;
3738
import org.neo4j.gds.hdbscan.HDBScanParameters;
3839
import org.neo4j.gds.hdbscan.Labels;
40+
import org.neo4j.gds.k1coloring.K1ColoringParameters;
41+
import org.neo4j.gds.k1coloring.K1ColoringResult;
3942
import org.neo4j.gds.result.TimedAlgorithmResult;
4043
import org.neo4j.gds.results.ResultTransformerBuilder;
4144

@@ -184,4 +187,35 @@ public <TR> CompletableFuture<TR> hdbscan(
184187
).thenApply(resultTransformerBuilder.build(graphResources));
185188
}
186189

190+
public <TR> CompletableFuture<TR> k1Coloring(
191+
GraphName graphName,
192+
GraphParameters graphParameters,
193+
Optional<String> relationshipProperty,
194+
K1ColoringParameters parameters,
195+
JobId jobId,
196+
boolean logProgress,
197+
ResultTransformerBuilder<TimedAlgorithmResult<K1ColoringResult>, TR> resultTransformerBuilder
198+
) {
199+
// Fetch the Graph the algorithm will operate on
200+
var graphResources = graphStoreCatalogService.fetchGraphResources(
201+
graphName,
202+
graphParameters,
203+
relationshipProperty,
204+
new NoAlgorithmValidation(),
205+
Optional.empty(),
206+
user,
207+
databaseId
208+
);
209+
var graph = graphResources.graph();
210+
211+
return computeFacade.k1Coloring(
212+
graph,
213+
parameters,
214+
jobId,
215+
logProgress
216+
217+
).thenApply(resultTransformerBuilder.build(graphResources));
218+
}
219+
220+
187221
}

0 commit comments

Comments
 (0)