From 45686ac106c57a160f6bc27089f3429ef38b34d9 Mon Sep 17 00:00:00 2001
From: Bharath Vissapragada <bharathv@redpanda.com>
Date: Fri, 2 Jun 2023 16:47:32 -0700
Subject: [PATCH] partition_balancer: abort planner tick on state changes

aborts the balancer tick on node state changes.
---
 src/v/cluster/partition_balancer_backend.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/v/cluster/partition_balancer_backend.cc b/src/v/cluster/partition_balancer_backend.cc
index e7f97ca57a238..a6cbcc1909614 100644
--- 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);
     }
 }