partition_balancer: abort planner tick on state changes

aborts the balancer tick on node state changes.
redpanda-data · Jun 12, 2023 · 45686ac · 45686ac
1 parent ba57559
commit 45686ac
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/v/cluster/partition_balancer_backend.cc b/src/v/cluster/partition_balancer_backend.cc
@@ -125,6 +125,10 @@ void partition_balancer_backend::on_members_update(
     if (
       state == model::membership_state::active
       || state == model::membership_state::draining) {
+        if (_tick_in_progress) {
+            _tick_in_progress->request_abort_ex(balancer_tick_aborted_exception{
+              fmt::format("new membership update: {}", state)});
+        }
         maybe_rearm_timer(/*now = */ true);
     }
 }