From 6fb3542f2a5a2f2148d46ebd57edbfa09e4f9c96 Mon Sep 17 00:00:00 2001 From: MrPresent-Han <116052805+MrPresent-Han@users.noreply.github.com> Date: Fri, 6 Jan 2023 14:45:35 +0800 Subject: [PATCH] enable auto balance paramter(#21504) (#21507) Signed-off-by: MrPresent-Han --- internal/querycoordv2/checkers/balance_checker.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/internal/querycoordv2/checkers/balance_checker.go b/internal/querycoordv2/checkers/balance_checker.go index 399e7c03ff..6c6727fabc 100644 --- a/internal/querycoordv2/checkers/balance_checker.go +++ b/internal/querycoordv2/checkers/balance_checker.go @@ -43,6 +43,9 @@ func (b *BalanceChecker) Description() string { func (b *BalanceChecker) Check(ctx context.Context) []task.Task { ret := make([]task.Task, 0) + if !Params.QueryCoordCfg.AutoBalance.GetAsBool() { + return ret + } segmentPlans, channelPlans := b.Balance.Balance() tasks := balance.CreateSegmentTasksFromPlans(ctx, b.ID(), Params.QueryCoordCfg.SegmentTaskTimeout.GetAsDuration(time.Millisecond), segmentPlans)