From be0123863fd0a6a5dcfa3ffe7a67c0fdba54d538 Mon Sep 17 00:00:00 2001
From: Gao <chao.gao@zilliz.com>
Date: Wed, 31 Jul 2024 18:23:51 +0800
Subject: [PATCH] enhance: add channel num for queryHook optimization (#35105)

pr: #35104

Signed-off-by: chasingegg <chao.gao@zilliz.com>
---
 internal/querynodev2/optimizers/query_hook.go | 3 +++
 pkg/common/common.go                          | 1 +
 2 files changed, 4 insertions(+)

diff --git a/internal/querynodev2/optimizers/query_hook.go b/internal/querynodev2/optimizers/query_hook.go
index 3291fd6595..ed06688b48 100644
--- 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
index 7e1975c3b8..cd2c45746a 100644
--- 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"