Move out internal classes

Author: jjaderberg

algo/src/main/java/org/neo4j/gds/similarity/filteredknn/FilteredKnn.java

@@ -270,38 +270,6 @@ public void release() {
         return neighbors;
     }
 
-    static class UniformFilteredKnnSamplerSupplier implements Function<SplittableRandom, FilteredKnnSampler> {
-
-        private final Graph graph;
-
-        UniformFilteredKnnSamplerSupplier(Graph graph) {
-            this.graph = graph;
-        }
-
-        @Override
-        public FilteredKnnSampler apply(SplittableRandom splittableRandom) {
-            return new UniformFilteredKnnSampler(splittableRandom, graph.nodeCount());
-        }
-    }
-
-    static class RandomWalkFilteredKnnSamplerSupplier implements Function<SplittableRandom, FilteredKnnSampler> {
-
-        private final Graph graph;
-        private final Optional<Long> randomSeed;
-        private final int boundedK;
-
-        RandomWalkFilteredKnnSamplerSupplier(Graph graph, Optional<Long> randomSeed, int boundedK) {
-            this.graph = graph;
-            this.randomSeed = randomSeed;
-            this.boundedK = boundedK;
-        }
-
-        @Override
-        public FilteredKnnSampler apply(SplittableRandom splittableRandom) {
-            return new RandomWalkFilteredKnnSampler(graph.concurrentCopy(), splittableRandom, randomSeed, boundedK);
-        }
-    }
-
     private long iteration(HugeObjectArray<FilteredNeighborList> neighbors) {
         // this is a sanity check
         // we check for this before any iteration and return

algo/src/main/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnSamplerSupplier.java

@@ -0,0 +1,23 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.similarity.filteredknn;
+
+public interface FilteredKnnSamplerSupplier {
+}

algo/src/main/java/org/neo4j/gds/similarity/filteredknn/RandomWalkFilteredKnnSamplerSupplier.java

@@ -0,0 +1,44 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.similarity.filteredknn;
+
+import org.neo4j.gds.api.Graph;
+
+import java.util.Optional;
+import java.util.SplittableRandom;
+import java.util.function.Function;
+
+class RandomWalkFilteredKnnSamplerSupplier implements Function<SplittableRandom, FilteredKnnSampler> {
+
+    private final Graph graph;
+    private final Optional<Long> randomSeed;
+    private final int boundedK;
+
+    RandomWalkFilteredKnnSamplerSupplier(Graph graph, Optional<Long> randomSeed, int boundedK) {
+        this.graph = graph;
+        this.randomSeed = randomSeed;
+        this.boundedK = boundedK;
+    }
+
+    @Override
+    public FilteredKnnSampler apply(SplittableRandom splittableRandom) {
+        return new RandomWalkFilteredKnnSampler(graph.concurrentCopy(), splittableRandom, randomSeed, boundedK);
+    }
+}

algo/src/main/java/org/neo4j/gds/similarity/filteredknn/UniformFilteredKnnSamplerSupplier.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.similarity.filteredknn;
+
+import org.neo4j.gds.api.Graph;
+
+import java.util.SplittableRandom;
+import java.util.function.Function;
+
+class UniformFilteredKnnSamplerSupplier implements Function<SplittableRandom, FilteredKnnSampler> {
+
+    private final Graph graph;
+
+    UniformFilteredKnnSamplerSupplier(Graph graph) {
+        this.graph = graph;
+    }
+
+    @Override
+    public FilteredKnnSampler apply(SplittableRandom splittableRandom) {
+        return new UniformFilteredKnnSampler(splittableRandom, graph.nodeCount());
+    }
+}