|
32 | 32 | import org.neo4j.gds.conductance.ConductanceResult; |
33 | 33 | import org.neo4j.gds.core.JobId; |
34 | 34 | import org.neo4j.gds.core.loading.GraphStoreCatalogService; |
| 35 | +import org.neo4j.gds.core.loading.validation.NoAlgorithmValidation; |
35 | 36 | import org.neo4j.gds.core.loading.validation.NodePropertyAnyExistsGraphStoreValidation; |
36 | 37 | import org.neo4j.gds.core.loading.validation.UndirectedOnlyGraphStoreValidation; |
37 | 38 | import org.neo4j.gds.hdbscan.HDBScanParameters; |
38 | 39 | import org.neo4j.gds.hdbscan.Labels; |
| 40 | +import org.neo4j.gds.k1coloring.K1ColoringParameters; |
| 41 | +import org.neo4j.gds.k1coloring.K1ColoringResult; |
39 | 42 | import org.neo4j.gds.result.TimedAlgorithmResult; |
40 | 43 | import org.neo4j.gds.results.ResultTransformerBuilder; |
41 | 44 |
|
@@ -184,4 +187,35 @@ public <TR> CompletableFuture<TR> hdbscan( |
184 | 187 | ).thenApply(resultTransformerBuilder.build(graphResources)); |
185 | 188 | } |
186 | 189 |
|
| 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 | + |
187 | 221 | } |
0 commit comments