From ac82cef04d8e9f591cd5e05333f0d2d4cfcb3d49 Mon Sep 17 00:00:00 2001 From: Chun Han <116052805+MrPresent-Han@users.noreply.github.com> Date: Mon, 29 Apr 2024 19:11:26 +0800 Subject: [PATCH] enhance: disable reload partstats by config (#32702) Signed-off-by: MrPresent-Han --- internal/querynodev2/delegator/delegator.go | 4 +++- internal/querynodev2/delegator/delegator_data.go | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/internal/querynodev2/delegator/delegator.go b/internal/querynodev2/delegator/delegator.go index 78b7258c7a..c4baf7a64c 100644 --- a/internal/querynodev2/delegator/delegator.go +++ b/internal/querynodev2/delegator/delegator.go @@ -873,6 +873,8 @@ func NewShardDelegator(ctx context.Context, collectionID UniqueID, replicaID Uni go sd.watchTSafe() } log.Info("finish build new shardDelegator") - sd.maybeReloadPartitionStats(ctx) + if paramtable.Get().QueryNodeCfg.EnableSegmentPrune.GetAsBool() { + sd.maybeReloadPartitionStats(ctx) + } return sd, nil } diff --git a/internal/querynodev2/delegator/delegator_data.go b/internal/querynodev2/delegator/delegator_data.go index 2c74bd5e4b..a9587ba99b 100644 --- a/internal/querynodev2/delegator/delegator_data.go +++ b/internal/querynodev2/delegator/delegator_data.go @@ -504,7 +504,9 @@ func (sd *shardDelegator) LoadSegments(ctx context.Context, req *querypb.LoadSeg lo.ForEach(req.GetInfos(), func(info *querypb.SegmentLoadInfo, _ int) { partStatsToReload = append(partStatsToReload, info.PartitionID) }) - sd.maybeReloadPartitionStats(ctx, partStatsToReload...) + if paramtable.Get().QueryNodeCfg.EnableSegmentPrune.GetAsBool() { + sd.maybeReloadPartitionStats(ctx, partStatsToReload...) + } return nil } @@ -906,7 +908,9 @@ func (sd *shardDelegator) ReleaseSegments(ctx context.Context, req *querypb.Rele partitionsToReload = append(partitionsToReload, segment.Partition()) } }) - sd.maybeReloadPartitionStats(ctx, partitionsToReload...) + if paramtable.Get().QueryNodeCfg.EnableSegmentPrune.GetAsBool() { + sd.maybeReloadPartitionStats(ctx, partitionsToReload...) + } return nil }