ever more migrating feature toggles 7

neo4j · Aug 22, 2024 · 0a89763 · 0a89763
1 parent 1c369e7
commit 0a89763
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 7 deletions.
diff --git a/...rations/src/main/java/org/neo4j/gds/applications/operations/FeatureTogglesRepository.java b/...rations/src/main/java/org/neo4j/gds/applications/operations/FeatureTogglesRepository.java
@@ -30,6 +30,10 @@
  * Let's encapsulate feature toggles and eventually make them not a global singleton.
  */
 public class FeatureTogglesRepository {
+    void enableAdjacencyCompressionMemoryTracking(boolean value) {
+        GdsFeatureToggles.ENABLE_ADJACENCY_COMPRESSION_MEMORY_TRACKING.toggle(value);
+    }
+
     void enableArrowDatabaseImport(boolean value) {
         GdsFeatureToggles.ENABLE_ARROW_DATABASE_IMPORT.toggle(value);
     }

diff --git a/...perations/src/main/java/org/neo4j/gds/applications/operations/OperationsApplications.java b/...perations/src/main/java/org/neo4j/gds/applications/operations/OperationsApplications.java
@@ -47,6 +47,10 @@ public static OperationsApplications create(
         return new OperationsApplications(featureTogglesRepository, requestScopedDependencies);
     }
 
+    public void enableAdjacencyCompressionMemoryTracking(boolean value) {
+        featureTogglesRepository.enableAdjacencyCompressionMemoryTracking(value);
+    }
+
     public void enableArrowDatabaseImport(boolean value) {
         featureTogglesRepository.enableArrowDatabaseImport(value);
     }

diff --git a/proc/misc/src/main/java/org/neo4j/gds/FeatureToggleProc.java b/proc/misc/src/main/java/org/neo4j/gds/FeatureToggleProc.java
@@ -48,6 +48,13 @@ public void adjacencyPackingStrategy(@Name(value = "adjacencyPackingStrategy") S
         facade.operations().setAdjacencyPackingStrategy(adjacencyPackingStrategy);
     }
 
+    @Internal
+    @Procedure("gds.features.enableAdjacencyCompressionMemoryTracking")
+    @Description("Enables memory tracking during the construction of an adjacency list.")
+    public void enableAdjacencyCompressionMemoryTracking(@Name(value = "enableAdjacencyCompressionMemoryTracking") boolean enableAdjacencyCompressionMemoryTracking) {
+        facade.operations().enableAdjacencyCompressionMemoryTracking(enableAdjacencyCompressionMemoryTracking);
+    }
+
     @Internal
     @Procedure("gds.features.enableArrowDatabaseImport")
     @Description("Enables support for importing Neo4j databases via the GDS Arrow Flight Server.")
@@ -139,13 +146,6 @@ public void useUncompressedAdjacencyList(@Name(value = "useUncompressedAdjacency
         facade.operations().setUseUncompressedAdjacencyList(useUncompressedAdjacencyList);
     }
 
-    @Internal
-    @Procedure("gds.features.enableAdjacencyCompressionMemoryTracking")
-    @Description("Enables memory tracking during the construction of an adjacency list.")
-    public void enableAdjacencyCompressionMemoryTracking(@Name(value = "enableAdjacencyCompressionMemoryTracking") boolean enableAdjacencyCompressionMemoryTracking) {
-        GdsFeatureToggles.ENABLE_ADJACENCY_COMPRESSION_MEMORY_TRACKING.toggle(enableAdjacencyCompressionMemoryTracking);
-    }
-
     @Internal
     @Procedure("gds.features.enableAdjacencyCompressionMemoryTracking.reset")
     @Description("Sets the default behaviour for enabling memory tracking during the construction of an adjacency list. That value is returned.")

diff --git a/...s-facade/src/main/java/org/neo4j/gds/procedures/operations/OperationsProcedureFacade.java b/...s-facade/src/main/java/org/neo4j/gds/procedures/operations/OperationsProcedureFacade.java
@@ -32,6 +32,10 @@ public OperationsProcedureFacade(ApplicationsFacade applicationsFacade) {
         this.applicationsFacade = applicationsFacade;
     }
 
+    public void enableAdjacencyCompressionMemoryTracking(boolean value) {
+        applicationsFacade.operations().enableAdjacencyCompressionMemoryTracking(value);
+    }
+
     public void enableArrowDatabaseImport(boolean value) {
         applicationsFacade.operations().enableArrowDatabaseImport(value);
     }