From 9ca8f28005a63b9fb88dd1a9736baabc335f71a5 Mon Sep 17 00:00:00 2001 From: Gao Date: Fri, 11 Aug 2023 18:35:29 +0800 Subject: [PATCH] Fix missing totalChannelNum bug (#26295) Signed-off-by: chasingegg --- internal/querynodev2/services.go | 1 + internal/querynodev2/services_test.go | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/internal/querynodev2/services.go b/internal/querynodev2/services.go index f2bffd8e7..66a7ee2f7 100644 --- 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 index b6cbf70d1..860e371ba 100644 --- 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) -- GitLab