Skip to content

Commit 36eaac0

Browse files
LiuRuoyu01yuhaijun999
authored andcommitted
[fix][coordinator]Fixup disable balance issue
1 parent 9ac5f49 commit 36eaac0

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

src/store/heartbeat.cc

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,9 @@ DEFINE_int64(store_heartbeat_report_region_multiple, 3,
5353
"store heartbeat report region multiple, this defines how many times of heartbeat will report "
5454
"region_metrics once to coordinator");
5555

56+
DECLARE_bool(enable_balance_leader);
57+
DECLARE_bool(enable_balance_region);
58+
5659
std::atomic<uint64_t> HeartbeatTask::heartbeat_counter = 0;
5760

5861
void HeartbeatTask::SendStoreHeartbeat(std::shared_ptr<CoordinatorInteraction> coordinator_interaction,
@@ -687,12 +690,20 @@ void Heartbeat::TriggerScrubVectorIndex(void*) {
687690
}
688691

689692
void Heartbeat::TriggerBalanceLeader(void*) {
693+
if (!FLAGS_enable_balance_leader) {
694+
DINGO_LOG(INFO) << "disable balance leader";
695+
return;
696+
}
690697
// Free at ExecuteRoutine()
691698
auto task = std::make_shared<BalanceLeaderTask>();
692699
Server::GetInstance().GetHeartbeat()->Execute(task);
693700
}
694701

695702
void Heartbeat::TriggerBalanceRegion(void*) {
703+
if (!FLAGS_enable_balance_region) {
704+
DINGO_LOG(INFO) << "disable balance region";
705+
return;
706+
}
696707
auto task = std::make_shared<BalanceRegionTask>();
697708
Server::GetInstance().GetHeartbeat()->Execute(task);
698709
}

0 commit comments

Comments
 (0)