[controller] Set rebalancing strategy to Waged instead of Crush when …

…adding a resource to controller cluster (#1197)
linkedin · Sep 24, 2024 · 1df15f8 · 1df15f8
1 parent 5d92c7d
commit 1df15f8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/...es/venice-controller/src/main/java/com/linkedin/venice/controller/ZkHelixAdminClient.java b/...es/venice-controller/src/main/java/com/linkedin/venice/controller/ZkHelixAdminClient.java
@@ -16,7 +16,7 @@
 import org.apache.helix.cloud.constants.CloudProvider;
 import org.apache.helix.controller.rebalancer.DelayedAutoRebalancer;
 import org.apache.helix.controller.rebalancer.strategy.AutoRebalanceStrategy;
-import org.apache.helix.controller.rebalancer.strategy.CrushRebalanceStrategy;
+import org.apache.helix.controller.rebalancer.waged.WagedRebalancer;
 import org.apache.helix.manager.zk.ZKHelixAdmin;
 import org.apache.helix.manager.zk.ZKHelixManager;
 import org.apache.helix.manager.zk.ZNRecordSerializer;
@@ -167,8 +167,7 @@ public void addVeniceStorageClusterToControllerCluster(String clusterName) {
       VeniceControllerClusterConfig config = multiClusterConfigs.getControllerConfig(clusterName);
       IdealState idealState = helixAdmin.getResourceIdealState(controllerClusterName, clusterName);
       idealState.setMinActiveReplicas(controllerClusterReplicaCount);
-      idealState.setRebalancerClassName(DelayedAutoRebalancer.class.getName());
-      idealState.setRebalanceStrategy(CrushRebalanceStrategy.class.getName());
+      idealState.setRebalancerClassName(WagedRebalancer.class.getName());
 
       String instanceGroupTag = config.getControllerResourceInstanceGroupTag();
       if (!instanceGroupTag.isEmpty()) {