Merge pull request #10293 from vnickolov/remote-centrality-stats

remote centrality stats

Author: vnickolov

procedures/facade-api/centrality-facade-api/src/main/java/org/neo4j/gds/procedures/algorithms/centrality/CELFStatsResult.java

@@ -20,42 +20,17 @@
 package org.neo4j.gds.procedures.algorithms.centrality;
 
 import org.neo4j.gds.applications.algorithms.machinery.AlgorithmProcessingTimings;
-import org.neo4j.gds.result.AbstractResultBuilder;
 
 import java.util.Map;
 
-public final class CELFStatsResult {
-    public final long computeMillis;
-    public final double totalSpread;
-    public final long nodeCount;
-    public final Map<String, Object> configuration;
-
-
-    public CELFStatsResult(long computeMillis, double totalSpread, long nodeCount, Map<String, Object> configuration) {
-        this.computeMillis = computeMillis;
-        this.totalSpread = totalSpread;
-        this.nodeCount = nodeCount;
-        this.configuration = configuration;
-    }
-
-    public static Builder builder() {
-        return new Builder();
-    }
+public record CELFStatsResult(
+    long computeMillis,
+    double totalSpread,
+    long nodeCount,
+    Map<String, Object> configuration
+) {
 
     static CELFStatsResult emptyFrom(AlgorithmProcessingTimings timings, Map<String, Object> configurationMap) {
         return new CELFStatsResult(timings.computeMillis, 0, 0, configurationMap);
     }
-
-    public static class Builder extends AbstractResultBuilder<CELFStatsResult> {
-        private double totalSpread;
-
-        public Builder withTotalSpread(double totalSpread) {
-            this.totalSpread = totalSpread;
-            return this;
-        }
-
-        public CELFStatsResult build() {
-            return new CELFStatsResult(computeMillis, totalSpread, nodeCount, config.toMap());
-        }
-    }
 }

procedures/facade-api/centrality-facade-api/src/main/java/org/neo4j/gds/procedures/algorithms/centrality/PageRankStatsResult.java

@@ -19,16 +19,21 @@
  */
 package org.neo4j.gds.procedures.algorithms.centrality;
 
-import org.neo4j.gds.api.ProcedureReturnColumns;
 import org.neo4j.gds.applications.algorithms.machinery.AlgorithmProcessingTimings;
-import org.neo4j.gds.core.concurrency.Concurrency;
+import org.neo4j.gds.procedures.algorithms.results.StatsResult;
 
 import java.util.Collections;
 import java.util.Map;
 
-public record PageRankStatsResult(long ranIterations, boolean didConverge, Map<String, Object> centralityDistribution,
-                                  long preProcessingMillis, long computeMillis, long postProcessingMillis,
-                                  Map<String, Object> configuration) {
+public record PageRankStatsResult(
+    long ranIterations,
+    boolean didConverge,
+    Map<String, Object> centralityDistribution,
+    long preProcessingMillis,
+    long computeMillis,
+    long postProcessingMillis,
+    Map<String, Object> configuration
+) implements StatsResult {
 
     static PageRankStatsResult emptyFrom(AlgorithmProcessingTimings timings, Map<String, Object> configurationMap) {
         return new PageRankStatsResult(
@@ -41,23 +46,4 @@ static PageRankStatsResult emptyFrom(AlgorithmProcessingTimings timings, Map<Str
             configurationMap
         );
     }
-
-    public static class Builder extends PageRankProcCompanion.PageRankResultBuilder<PageRankStatsResult> {
-        public Builder(ProcedureReturnColumns returnColumns, Concurrency concurrency) {
-            super(returnColumns, concurrency);
-        }
-
-        @Override
-        public PageRankStatsResult buildResult() {
-            return new PageRankStatsResult(
-                ranIterations,
-                didConverge,
-                centralityHistogram,
-                preProcessingMillis,
-                computeMillis,
-                postProcessingMillis,
-                config.toMap()
-            );
-        }
-    }
 }