3333import org .neo4j .gds .applications .algorithms .metadata .RelationshipsWritten ;
3434import org .neo4j .gds .collections .ha .HugeLongArray ;
3535import org .neo4j .gds .collections .haa .HugeAtomicLongArray ;
36+ import org .neo4j .gds .pathfinding .BellmanFordMutateStep ;
37+ import org .neo4j .gds .pathfinding .PrizeCollectingSteinerTreeMutateStep ;
38+ import org .neo4j .gds .pathfinding .RandomWalkCountingNodeVisitsMutateStep ;
39+ import org .neo4j .gds .pathfinding .SearchMutateStep ;
40+ import org .neo4j .gds .pathfinding .ShortestPathMutateStep ;
41+ import org .neo4j .gds .pathfinding .SpanningTreeMutateStep ;
42+ import org .neo4j .gds .pathfinding .SteinerTreeMutateStep ;
3643import org .neo4j .gds .paths .astar .config .ShortestPathAStarMutateConfig ;
3744import org .neo4j .gds .paths .bellmanford .AllShortestPathsBellmanFordMutateConfig ;
3845import org .neo4j .gds .paths .bellmanford .BellmanFordResult ;
@@ -121,7 +128,10 @@ public <RESULT> RESULT breadthFirstSearch(
121128 BfsMutateConfig configuration ,
122129 ResultBuilder <BfsMutateConfig , HugeLongArray , RESULT , RelationshipsWritten > resultBuilder
123130 ) {
124- var mutateStep = new SearchMutateStep (mutateRelationshipService , RelationshipType .of (configuration .mutateRelationshipType ()));
131+ var mutateStep = new SearchMutateStep (
132+ mutateRelationshipService ,
133+ RelationshipType .of (configuration .mutateRelationshipType ())
134+ );
125135
126136 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
127137 graphName ,
@@ -139,7 +149,7 @@ public <RESULT> RESULT deltaStepping(
139149 AllShortestPathsDeltaMutateConfig configuration ,
140150 ResultBuilder <AllShortestPathsDeltaMutateConfig , PathFindingResult , RESULT , RelationshipsWritten > resultBuilder
141151 ) {
142- var mutateStep = new ShortestPathMutateStep (mutateRelationshipService , configuration );
152+ var mutateStep = new ShortestPathMutateStep (configuration . mutateRelationshipType (), mutateRelationshipService );
143153
144154 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
145155 graphName ,
@@ -157,7 +167,10 @@ public <RESULT> RESULT depthFirstSearch(
157167 DfsMutateConfig configuration ,
158168 ResultBuilder <DfsMutateConfig , HugeLongArray , RESULT , RelationshipsWritten > resultBuilder
159169 ) {
160- var mutateStep = new SearchMutateStep (mutateRelationshipService ,RelationshipType .of (configuration .mutateRelationshipType ()));
170+ var mutateStep = new SearchMutateStep (
171+ mutateRelationshipService ,
172+ RelationshipType .of (configuration .mutateRelationshipType ())
173+ );
161174
162175 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
163176 graphName ,
@@ -175,7 +188,11 @@ public <RESULT> RESULT pcst(
175188 PCSTMutateConfig configuration ,
176189 ResultBuilder <PCSTMutateConfig , PrizeSteinerTreeResult , RESULT , RelationshipsWritten > resultBuilder
177190 ) {
178- var mutateStep = new PrizeCollectingSteinerTreeMutateStep (mutateRelationshipService ,configuration );
191+ var mutateStep = new PrizeCollectingSteinerTreeMutateStep (
192+ configuration .mutateRelationshipType (),
193+ configuration .mutateProperty (),
194+ mutateRelationshipService
195+ );
179196
180197 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
181198 graphName ,
@@ -193,7 +210,11 @@ public <RESULT> RESULT randomWalk(
193210 RandomWalkMutateConfig configuration ,
194211 ResultBuilder <RandomWalkMutateConfig , HugeAtomicLongArray , RESULT , NodePropertiesWritten > resultBuilder
195212 ) {
196- var mutateStep = new RandomWalkCountingNodeVisitsMutateStep (mutateNodeProperty , configuration );
213+ var mutateStep = new RandomWalkCountingNodeVisitsMutateStep (
214+ configuration .nodeLabels (),
215+ configuration .mutateProperty (),
216+ mutateNodeProperty
217+ );
197218
198219 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
199220 graphName ,
@@ -212,7 +233,7 @@ public <RESULT> RESULT singlePairShortestPathAStar(
212233 ShortestPathAStarMutateConfig configuration ,
213234 ResultBuilder <ShortestPathAStarMutateConfig , PathFindingResult , RESULT , RelationshipsWritten > resultBuilder
214235 ) {
215- var mutateStep = new ShortestPathMutateStep (mutateRelationshipService , configuration );
236+ var mutateStep = new ShortestPathMutateStep (configuration . mutateRelationshipType (), mutateRelationshipService );
216237
217238 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
218239 graphName ,
@@ -230,7 +251,7 @@ public <RESULT> RESULT singlePairShortestPathDijkstra(
230251 ShortestPathDijkstraMutateConfig configuration ,
231252 ResultBuilder <ShortestPathDijkstraMutateConfig , PathFindingResult , RESULT , RelationshipsWritten > resultBuilder
232253 ) {
233- var mutateStep = new ShortestPathMutateStep (mutateRelationshipService , configuration );
254+ var mutateStep = new ShortestPathMutateStep (configuration . mutateRelationshipType (), mutateRelationshipService );
234255
235256 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
236257 graphName ,
@@ -271,7 +292,7 @@ public <RESULT> RESULT singlePairShortestPathYens(
271292 ShortestPathYensMutateConfig configuration ,
272293 ResultBuilder <ShortestPathYensMutateConfig , PathFindingResult , RESULT , RelationshipsWritten > resultBuilder
273294 ) {
274- var mutateStep = new ShortestPathMutateStep (mutateRelationshipService , configuration );
295+ var mutateStep = new ShortestPathMutateStep (configuration . mutateRelationshipType (), mutateRelationshipService );
275296
276297 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
277298 graphName ,
@@ -289,7 +310,7 @@ public <RESULT> RESULT singleSourceShortestPathDijkstra(
289310 AllShortestPathsDijkstraMutateConfig configuration ,
290311 ResultBuilder <AllShortestPathsDijkstraMutateConfig , PathFindingResult , RESULT , RelationshipsWritten > resultBuilder
291312 ) {
292- var mutateStep = new ShortestPathMutateStep (mutateRelationshipService , configuration );
313+ var mutateStep = new ShortestPathMutateStep (configuration . mutateRelationshipType (), mutateRelationshipService );
293314
294315 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
295316 graphName ,
@@ -355,7 +376,11 @@ public <RESULT> RESULT spanningTree(
355376 SpanningTreeMutateConfig configuration ,
356377 ResultBuilder <SpanningTreeMutateConfig , SpanningTree , RESULT , RelationshipsWritten > resultBuilder
357378 ) {
358- var mutateStep = new SpanningTreeMutateStep (mutateRelationshipService ,configuration );
379+ var mutateStep = new SpanningTreeMutateStep (
380+ configuration .mutateRelationshipType (),
381+ configuration .mutateProperty (),
382+ mutateRelationshipService
383+ );
359384
360385 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
361386 graphName ,
@@ -373,7 +398,12 @@ public <RESULT> RESULT steinerTree(
373398 SteinerTreeMutateConfig configuration ,
374399 ResultBuilder <SteinerTreeMutateConfig , SteinerTreeResult , RESULT , RelationshipsWritten > resultBuilder
375400 ) {
376- var mutateStep = new SteinerTreeMutateStep (mutateRelationshipService ,configuration );
401+ var mutateStep = new SteinerTreeMutateStep (
402+ configuration .mutateRelationshipType (),
403+ configuration .mutateProperty (),
404+ configuration .sourceNode (),
405+ mutateRelationshipService
406+ );
377407
378408 return algorithmProcessingTemplateConvenience .processRegularAlgorithmInMutateMode (
379409 graphName ,
0 commit comments