add deploy mode

apache · Oct 7, 2023 · 75499a6 · 75499a6
1 parent 52b8f67
commit 75499a6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/server/coordinator/scheduler/rebalanced_shard_scheduler.go b/server/coordinator/scheduler/rebalanced_shard_scheduler.go
@@ -39,6 +39,10 @@ func NewRebalancedShardScheduler(logger *zap.Logger, factory *coordinator.Factor
 	}
 }
 
+func (r *RebalancedShardScheduler) UpdateDeployMode(_ context.Context, enable bool) {
+	r.updateDeployMode(enable)
+}
+
 func (r *RebalancedShardScheduler) Schedule(ctx context.Context, clusterSnapshot metadata.Snapshot) (ScheduleResult, error) {
 	// RebalancedShardScheduler can only be scheduled when the cluster is not empty.
 	if clusterSnapshot.Topology.ClusterView.State == storage.ClusterStateEmpty {