Fix missing totalChannelNum bug (#26295)

Signed-off-by: chasingegg <chao.gao@zilliz.com>
milvus-io · Aug 11, 2023 · 9ca8f28 · 9ca8f28
1 parent 48422dd
commit 9ca8f28
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/internal/querynodev2/services.go b/internal/querynodev2/services.go
@@ -821,6 +821,7 @@ func (node *QueryNode) Search(ctx context.Context, req *querypb.SearchRequest) (
 			SegmentIDs:      req.SegmentIDs,
 			FromShardLeader: req.FromShardLeader,
 			Scope:           req.Scope,
+			TotalChannelNum: req.TotalChannelNum,
 		}
 
 		runningGp.Go(func() error {

diff --git a/internal/querynodev2/services_test.go b/internal/querynodev2/services_test.go
@@ -1091,6 +1091,7 @@ func (suite *ServiceSuite) TestSearch_Normal() {
 		Req:             creq,
 		FromShardLeader: false,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)
 
@@ -1117,6 +1118,7 @@ func (suite *ServiceSuite) TestSearch_Concurrent() {
 				Req:             creq,
 				FromShardLeader: false,
 				DmlChannels:     []string{suite.vchannel},
+				TotalChannelNum: 2,
 			}
 			suite.NoError(err)
 			return suite.node.Search(ctx, req)
@@ -1142,6 +1144,7 @@ func (suite *ServiceSuite) TestSearch_Failed() {
 		Req:             creq,
 		FromShardLeader: false,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)
 
@@ -1201,6 +1204,7 @@ func (suite *ServiceSuite) TestSearchSegments_Normal() {
 		Req:             creq,
 		FromShardLeader: true,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)