Replace rootIdMap

IoannisPanagiotas · IoannisPanagiotas · commit b74350feb8cf · 2025-05-30T12:46:02.000+02:00
diff --git a/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/BellmanFordWriteStep.java b/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/BellmanFordWriteStep.java
@@ -70,7 +70,7 @@ public RelationshipsWritten execute(
                 keys,
                 types,
                 relationshipStream,
-                graph.rootIdMap(),
+                graph,
                 "Write shortest Paths",
                 configuration.resolveResultStore(resultStore),
                 configuration.jobId()
diff --git a/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/PrizeCollectingSteinerTreeWriteStep.java b/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/PrizeCollectingSteinerTreeWriteStep.java
@@ -70,7 +70,7 @@ public RelationshipsWritten execute(
             configuration.writeRelationshipType(),
             configuration.writeProperty(),
             spanningGraph,
-            spanningGraph.rootIdMap(),
+            spanningGraph,
             "PrizeCollectingSteinerWrite",
             configuration.resolveResultStore(resultStore),
             (a,b,c)-> true,
diff --git a/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/ShortestPathWriteStep.java b/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/ShortestPathWriteStep.java
@@ -88,7 +88,7 @@ public RelationshipsWritten execute(
                 keys,
                 types,
                 relationshipStream,
-                graph.rootIdMap(),
+                graph,
                 "Write shortest Paths",
                 configuration.resolveResultStore(resultStore),
                 configuration.jobId()
diff --git a/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/SpanningTreeWriteStep.java b/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/SpanningTreeWriteStep.java
@@ -58,7 +58,7 @@ public RelationshipsWritten execute(
             configuration.writeRelationshipType(),
             configuration.writeProperty(),
             spanningGraph,
-            spanningGraph.rootIdMap(),
+            graph,
             AlgorithmLabel.SpanningTree.asString(),
             configuration.resolveResultStore(resultStore),
             (a,b,c)-> true,
diff --git a/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/SteinerTreeWriteStep.java b/applications/algorithms/path-finding/src/main/java/org/neo4j/gds/applications/algorithms/pathfinding/SteinerTreeWriteStep.java
@@ -67,7 +67,7 @@ public RelationshipsWritten execute(
              configuration.writeRelationshipType(),
              configuration.writeProperty(),
              spanningGraph,
-             spanningGraph.rootIdMap(),
+             graph,
              "SteinerWrite",
              configuration.resolveResultStore(resultStore),
              (a,b,c)-> true,
