From dbae033af9bb69f2f6040f1ad7707121c8af9455 Mon Sep 17 00:00:00 2001
From: qoo332001 <sean0651101@gmail.com>
Date: Mon, 22 May 2023 21:33:36 +0800
Subject: [PATCH] merge main

---
 .../org/astraea/common/balancer/algorithms/GreedyBalancer.java   | 1 +
 .../astraea/common/balancer/algorithms/SingleStepBalancer.java   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/common/src/main/java/org/astraea/common/balancer/algorithms/GreedyBalancer.java b/common/src/main/java/org/astraea/common/balancer/algorithms/GreedyBalancer.java
index 67136915a3..1892040546 100644
--- a/common/src/main/java/org/astraea/common/balancer/algorithms/GreedyBalancer.java
+++ b/common/src/main/java/org/astraea/common/balancer/algorithms/GreedyBalancer.java
@@ -239,6 +239,7 @@ public Optional<Plan> offer(AlgorithmConfig config) {
                   .overflow()) {
             return Optional.of(
                 new Plan(
+                    config.clusterBean(),
                     config.clusterInfo(),
                     config.clusterCostFunction().clusterCost(config.clusterInfo(), clusterBean),
                     currentClusterInfo,
diff --git a/common/src/main/java/org/astraea/common/balancer/algorithms/SingleStepBalancer.java b/common/src/main/java/org/astraea/common/balancer/algorithms/SingleStepBalancer.java
index 378816217e..e28ab7343b 100644
--- a/common/src/main/java/org/astraea/common/balancer/algorithms/SingleStepBalancer.java
+++ b/common/src/main/java/org/astraea/common/balancer/algorithms/SingleStepBalancer.java
@@ -140,6 +140,7 @@ public Optional<Plan> offer(AlgorithmConfig config) {
                       .overflow()) {
                 return Optional.of(
                     new Plan(
+                        config.clusterBean(),
                         config.clusterInfo(),
                         config.clusterCostFunction().clusterCost(config.clusterInfo(), clusterBean),
                         currentClusterInfo,