Merge pull request #12078 from s1ck/pform-182-arrow-serialization-forgets-idmap-implementation

pform 182 arrow serialization forgets idmap implementation

Author: s1ck

core/src/main/java/org/neo4j/gds/beta/generator/RandomGraphGenerator.java

@@ -82,6 +82,7 @@ public final class RandomGraphGenerator {
     private final Optional<NodeLabelProducer> maybeNodeLabelProducer;
     private final List<PropertyProducer<double[]>> relationshipPropertyProducers;
     private final Map<NodeLabel, Set<PropertyProducer<?>>> nodePropertyProducers;
+    private final Optional<String> idMapBuilderType;
     private final AdjacencyListBehavior.Factory adjacencyCompressorFactory;
     private final boolean forceDag;
     private final HugeLongArray randomDagMapping;
@@ -98,6 +99,7 @@ public final class RandomGraphGenerator {
         Aggregation aggregation,
         Direction direction,
         AllowSelfLoops allowSelfLoops,
+        Optional<String> idMapBuilderType,
         AdjacencyListBehavior.Factory adjacencyCompressorFactory,
         boolean forceDag,
         boolean inverseIndex
@@ -112,6 +114,7 @@ public final class RandomGraphGenerator {
         this.aggregation = aggregation;
         this.direction = direction;
         this.allowSelfLoops = allowSelfLoops;
+        this.idMapBuilderType = idMapBuilderType;
         this.forceDag = forceDag;
         this.random = new Random(seed);
         this.propertyValueRandom = new Random(seed);
@@ -155,6 +158,7 @@ public GraphStore generateGraphstore(DatabaseId databaseId, Concurrency concurre
     private RandomGraphSpecs generateSpecs() {
         var nodesBuilder = GraphFactory.initNodesBuilder()
             .maxOriginalId(nodeCount)
+            .idMapBuilderType(idMapBuilderType)
             .hasLabelInformation(maybeNodeLabelProducer.isPresent())
             .build();
 

core/src/main/java/org/neo4j/gds/beta/generator/RandomGraphGeneratorBuilder.java

@@ -46,6 +46,7 @@ public class RandomGraphGeneratorBuilder {
     private Direction direction = Direction.DIRECTED;
     private RandomGraphGeneratorConfig.AllowSelfLoops allowSelfLoops = RandomGraphGeneratorConfig.AllowSelfLoops.NO;
     private RelationshipType relationshipType = RelationshipType.of("REL");
+    private Optional<String> idMapBuilderType = Optional.empty();
     private AdjacencyListBehavior.Factory adjacencyCompressorFactory = null;
     private boolean forceDag = false;
     private boolean inverseIndex = false;
@@ -122,6 +123,11 @@ public RandomGraphGeneratorBuilder inverseIndex(boolean inverseIndex) {
         return this;
     }
 
+    public RandomGraphGeneratorBuilder idMapBuilderType(String idMapBuilderType) {
+        this.idMapBuilderType = Optional.of(idMapBuilderType);
+        return this;
+    }
+
     public RandomGraphGeneratorBuilder adjacencyCompressorFactory(AdjacencyListBehavior.Factory adjacencyCompressorFactory) {
         this.adjacencyCompressorFactory = adjacencyCompressorFactory;
         return this;
@@ -141,6 +147,7 @@ public RandomGraphGenerator build() {
             aggregation,
             direction,
             allowSelfLoops,
+            idMapBuilderType,
             adjacencyCompressorFactory,
             forceDag,
             inverseIndex