enhance: add channel num for queryHook optimization (#35104)

At most cases, data in each channel is almost evenly distributed, we could utilize the channel num info to optimize searh param in queryHook Signed-off-by: chasingegg <[email protected]>
milvus-io · Jul 31, 2024 · 6695c6d · 6695c6d
1 parent 03912a8
commit 6695c6d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/internal/querynodev2/optimizers/query_hook.go b/internal/querynodev2/optimizers/query_hook.go
@@ -66,6 +66,9 @@ func OptimizeSearchParams(ctx context.Context, req *querypb.SearchRequest, query
 			common.WithOptimizeKey: paramtable.Get().AutoIndexConfig.EnableOptimize.GetAsBool(),
 			common.CollectionKey:   req.GetReq().GetCollectionID(),
 		}
+		if withFilter && channelNum > 1 {
+			params[common.ChannelNumKey] = channelNum
+		}
 		err := queryHook.Run(params)
 		if err != nil {
 			log.Warn("failed to execute queryHook", zap.Error(err))

diff --git a/pkg/common/common.go b/pkg/common/common.go
@@ -109,6 +109,7 @@ const (
 	SegmentNumKey   = "segment_num"
 	WithFilterKey   = "with_filter"
 	DataTypeKey     = "data_type"
+	ChannelNumKey   = "channel_num"
 	WithOptimizeKey = "with_optimize"
 	CollectionKey   = "collection"